CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-08-18 06:47:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

Asset related fixes

Browse Source
This commit is contained in:
henryruhs
2026-05-11 16:34:28 +02:00
parent c2676fa471
commit 60def0f30d
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/apis/endpoints/assets.py
+8 -3
View File
@@ -7,11 +7,11 @@ from starlette.requests import Request
from starlette.responses import JSONResponse, Response from starlette.responses import JSONResponse, Response
from starlette.status import HTTP_200_OK, HTTP_201_CREATED, HTTP_400_BAD_REQUEST, HTTP_404_NOT_FOUND from starlette.status import HTTP_200_OK, HTTP_201_CREATED, HTTP_400_BAD_REQUEST, HTTP_404_NOT_FOUND
from facefusion import ffmpeg, process_manager, session_manager, state_manager from facefusion import ffmpeg, process_manager, session_context, session_manager, state_manager
from facefusion.apis import asset_store from facefusion.apis import asset_store
from facefusion.apis.asset_helper import detect_media_type from facefusion.apis.asset_helper import detect_media_type
from facefusion.apis.endpoints.session import extract_access_token from facefusion.apis.endpoints.session import extract_access_token
from facefusion.filesystem import get_file_extension, get_file_name, remove_file from facefusion.filesystem import create_directory, get_file_extension, get_file_name, remove_file
async def upload_asset(request : Request) -> Response: async def upload_asset(request : Request) -> Response:
@@ -20,6 +20,8 @@ async def upload_asset(request : Request) -> Response:
asset_type = request.query_params.get('type') asset_type = request.query_params.get('type')
if session_id and asset_type in [ 'source', 'target' ]: if session_id and asset_type in [ 'source', 'target' ]:
session_context.set_session_id(session_id)
form = await request.form() form = await request.form()
upload_files = form.getlist('file') upload_files = form.getlist('file')
asset_paths = await save_asset_files(upload_files) # type: ignore[arg-type] asset_paths = await save_asset_files(upload_files) # type: ignore[arg-type]
@@ -60,8 +62,11 @@ async def save_asset_files(upload_files : List[UploadFile]) -> List[str]:
media_type = detect_media_type(temp_file.name) media_type = detect_media_type(temp_file.name)
temp_path = state_manager.get_temp_path() temp_path = state_manager.get_temp_path()
create_directory(temp_path)
asset_file_name = get_file_name(temp_file.name) asset_file_name = get_file_name(temp_file.name)
asset_path = os.path.join(temp_path, asset_file_name + '_' + upload_file_extension) asset_path = os.path.join(temp_path, asset_file_name + upload_file_extension)
process_manager.start() process_manager.start()