Update app.py

app.py

@@ -950,27 +950,40 @@ async def handle_upload(file: UploadFile = File(...)):
         return JSONResponse(content={"error": "Failed"}, status_code=500)
 
     original_extension = os.path.splitext(file.filename)[1][1:]
-    temp_filename = f"{file.filename}.png"
-
-    upload_result = await initiate_upload(cookies, temp_filename, "image/png")
+    supported_types = ['mp4', 'png', 'jpg', 'jpeg', 'gif', 'mp3', 'pdf', 'txt']
+    
+    if original_extension.lower() in supported_types:
+        temp_filename = file.filename
+        content_type = file.content_type
+    else:
+        temp_filename = f"{file.filename}.png"
+        content_type = "image/png"
+
+    upload_result = await initiate_upload(cookies, temp_filename, content_type)
     if not upload_result or 'upload_url' not in upload_result:
         return JSONResponse(content={"error": "Failed to upload"}, status_code=500)
 
     file_content = await file.read()
-    upload_success = await retry_upload(upload_result['upload_url'], file_content, "image/png")
+    upload_success = await retry_upload(upload_result['upload_url'], file_content, content_type)
     if not upload_success:
         return JSONResponse(content={"error": "FAILED GOD MAN AFTER alot of attempts"}, status_code=500)
 
     original_url = upload_result['serving_url']
     mirrored_url = f"/rbxg/{original_url.split('/pbxt/')[1]}"
+    
+    if original_extension.lower() not in supported_types:
+        mirrored_url = mirrored_url.replace('.png', '')
 
     return JSONResponse(content={"url": mirrored_url, "originalExtension": original_extension})
 
 @app.get("/rbxg/{path:path}")
 async def handle_file_stream(path: str, request: Request):
     original_url = f'https://replicate.delivery/pbxt/{path}'
+    
+    if not path.lower().endswith(('.png', '.jpg', '.jpeg', '.gif', '.mp4', '.mp3', '.pdf', '.txt')):
+        original_url += '.png'
+    
     range_header = request.headers.get('Range')
-
     headers = {'Range': range_header} if range_header else {}
     response = requests.get(original_url, headers=headers, stream=True)
 
@@ -985,7 +998,6 @@ async def handle_file_stream(path: str, request: Request):
     if response.status_code == 206:
         headers['Content-Range'] = response.headers.get('Content-Range')
 
-    # Set the correct Content-Type based on the original file extension
     original_extension = os.path.splitext(path)[1][1:]
     content_type, _ = mimetypes.guess_type(f"file.{original_extension}")
     if content_type: