Update app.py
Browse files
app.py
CHANGED
|
@@ -46,10 +46,10 @@ async def handle_upload(file: UploadFile = File(...)):
|
|
| 46 |
if not upload_success:
|
| 47 |
return JSONResponse({"error": "Upload failed"}, status_code=500)
|
| 48 |
|
| 49 |
-
|
| 50 |
-
return JSONResponse({"url":
|
| 51 |
|
| 52 |
-
@app.get("/
|
| 53 |
async def handle_stream(path: str, request: Request):
|
| 54 |
original_url = f'https://replicate.delivery/pbxt/{path}'
|
| 55 |
range_header = request.headers.get('Range')
|
|
@@ -97,3 +97,7 @@ async def retry_upload(upload_url, file_content, content_type, max_retries=5):
|
|
| 97 |
await asyncio.sleep(delay)
|
| 98 |
delay *= 2
|
| 99 |
return False
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
if not upload_success:
|
| 47 |
return JSONResponse({"error": "Upload failed"}, status_code=500)
|
| 48 |
|
| 49 |
+
direct_url = f"https://replicate.delivery/pbxt/{upload_result['serving_url'].split('/pbxt/')[1]}"
|
| 50 |
+
return JSONResponse({"url": direct_url})
|
| 51 |
|
| 52 |
+
@app.get("/file/{path:path}")
|
| 53 |
async def handle_stream(path: str, request: Request):
|
| 54 |
original_url = f'https://replicate.delivery/pbxt/{path}'
|
| 55 |
range_header = request.headers.get('Range')
|
|
|
|
| 97 |
await asyncio.sleep(delay)
|
| 98 |
delay *= 2
|
| 99 |
return False
|
| 100 |
+
|
| 101 |
+
if __name__ == "__main__":
|
| 102 |
+
import uvicorn
|
| 103 |
+
uvicorn.run(app, host="0.0.0.0", port=8000)
|