Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -974,7 +974,7 @@ async def handle_upload(file: UploadFile = File(...)):
|
|
974 |
|
975 |
cookies = await get_cookies()
|
976 |
if 'csrftoken' not in cookies or 'sessionid' not in cookies:
|
977 |
-
return JSONResponse(content={"error": "Failed"}, status_code=500)
|
978 |
|
979 |
original_extension = os.path.splitext(file.filename)[1][1:]
|
980 |
supported_types = ['mp4', 'png', 'jpg', 'jpeg', 'gif', 'mp3', 'pdf', 'txt']
|
@@ -988,12 +988,12 @@ async def handle_upload(file: UploadFile = File(...)):
|
|
988 |
|
989 |
upload_result = await initiate_upload(cookies, temp_filename, content_type)
|
990 |
if not upload_result or 'upload_url' not in upload_result:
|
991 |
-
return JSONResponse(content={"error": "Failed to upload"}, status_code=500)
|
992 |
|
993 |
file_content = await file.read()
|
994 |
upload_success = await retry_upload(upload_result['upload_url'], file_content, content_type)
|
995 |
if not upload_success:
|
996 |
-
return JSONResponse(content={"error": "
|
997 |
|
998 |
original_url = upload_result['serving_url']
|
999 |
mirrored_url = f"/rbxg/{original_url.split('/pbxt/')[1]}"
|
@@ -1114,7 +1114,7 @@ async def upload_file(upload_url: str, file_content: bytes, content_type: str) -
|
|
1114 |
return False
|
1115 |
|
1116 |
async def retry_upload(upload_url: str, file_content: bytes, content_type: str, max_retries: int = 5, delay: int = 1) -> bool:
|
1117 |
-
|
1118 |
try:
|
1119 |
success = await upload_file(upload_url, file_content, content_type)
|
1120 |
if success:
|
|
|
974 |
|
975 |
cookies = await get_cookies()
|
976 |
if 'csrftoken' not in cookies or 'sessionid' not in cookies:
|
977 |
+
return JSONResponse(content={"error": "Failed to get cookies"}, status_code=500)
|
978 |
|
979 |
original_extension = os.path.splitext(file.filename)[1][1:]
|
980 |
supported_types = ['mp4', 'png', 'jpg', 'jpeg', 'gif', 'mp3', 'pdf', 'txt']
|
|
|
988 |
|
989 |
upload_result = await initiate_upload(cookies, temp_filename, content_type)
|
990 |
if not upload_result or 'upload_url' not in upload_result:
|
991 |
+
return JSONResponse(content={"error": "Failed to initiate upload"}, status_code=500)
|
992 |
|
993 |
file_content = await file.read()
|
994 |
upload_success = await retry_upload(upload_result['upload_url'], file_content, content_type)
|
995 |
if not upload_success:
|
996 |
+
return JSONResponse(content={"error": "Failed to upload after multiple attempts"}, status_code=500)
|
997 |
|
998 |
original_url = upload_result['serving_url']
|
999 |
mirrored_url = f"/rbxg/{original_url.split('/pbxt/')[1]}"
|
|
|
1114 |
return False
|
1115 |
|
1116 |
async def retry_upload(upload_url: str, file_content: bytes, content_type: str, max_retries: int = 5, delay: int = 1) -> bool:
|
1117 |
+
for _ in range(max_retries):
|
1118 |
try:
|
1119 |
success = await upload_file(upload_url, file_content, content_type)
|
1120 |
if success:
|