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

upload asset endpoint

Browse Source
This commit is contained in:
harisreedhar
2026-01-17 11:27:06 +05:30
committed by henryruhs
parent 746b3249ef
commit 71c79f4e96
3 changed files with 31 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/apis/asset_store.py
+1 -1
View File
@@ -80,7 +80,7 @@ def clear_session(session_id : SessionId) -> None:
for asset in ASSET_STORE[session_id].values():
file_path = asset.get('path')
if file_path and is_file(file_path):
if is_file(file_path):
remove_file(file_path)
del ASSET_STORE[session_id]
facefusion/apis/endpoints/assets.py
+7 -7
View File
@@ -25,12 +25,12 @@ async def upload_asset(request : Request) -> JSONResponse:
if asset_type == 'target':
files = files[:1]
prepared_files = await prepare_files(files)
media_files = await prepare_media_files(files)
if prepared_files:
if media_files:
asset_ids : List[str] = []
for file_path, media_type in prepared_files:
for file_path, media_type in media_files:
asset = asset_store.create_asset(session_id, asset_type, media_type, file_path) #type:ignore[arg-type]
if asset:
@@ -48,8 +48,8 @@ async def upload_asset(request : Request) -> JSONResponse:
return JSONResponse({}, status_code = HTTP_400_BAD_REQUEST)
async def prepare_files(files : List[UploadFile]) -> List[Tuple[str, MediaType]]:
prepared_files : List[Tuple[str, MediaType]] = []
async def prepare_media_files(files : List[UploadFile]) -> List[Tuple[str, MediaType]]:
media_files : List[Tuple[str, MediaType]] = []
for file in files:
file_extension = get_file_extension(file.filename)
@@ -69,9 +69,9 @@ async def prepare_files(files : List[UploadFile]) -> List[Tuple[str, MediaType]]
media_type = 'video'
if media_type:
prepared_files.append((file_path, media_type))
media_files.append((file_path, media_type))
if not media_type and is_file(file_path):
remove_file(file_path)
return prepared_files
return media_files