Update main.py

main.py

@@ -58,6 +58,9 @@ from fastapi import Request, Header
 from fastapi import Body, Query
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
+from fastapi import FastAPI, File, UploadFile
+from fastapi.responses import FileResponse, JSONResponse
+import uuid
 
 import g4f
 from g4f.client import Client
@@ -130,6 +133,40 @@ app = FastAPI(docs_url=None, redoc_url="/")
 app.include_router(fluxai_router, prefix="/api/v1")
 app.include_router(whisper_router, prefix="/api/v1")
 
+
+UPLOAD_DIR = "./uploads"
+
+@app.post("/uploadfile/")
+async def upload_file(file: UploadFile = File(...)):
+    try:
+        ext = file.filename.split(".")[-1]
+        unique_filename = f"{uuid.uuid4().hex}.{ext}"
+        file_location = os.path.join(UPLOAD_DIR, unique_filename)
+        
+        with open(file_location, "wb") as f:
+            f.write(await file.read())
+        
+        return JSONResponse(
+            status_code=200,
+            content={"url": f"https://{os.getenv('SPACE_ID')}.hf.space/uploads/{unique_filename}"}
+        )
+    except Exception as e:
+        return JSONResponse(
+            status_code=500,
+            content={"error": str(e)}
+        )
+
+@app.get("/uploads/{filename}")
+async def serve_file(filename: str):
+    file_location = os.path.join(UPLOAD_DIR, filename)
+    if os.path.exists(file_location):
+        return FileResponse(file_location)
+    return JSONResponse(
+        status_code=404,
+        content={"error": "File not found"}
+    )
+
+
 timeout = 100
 
 contact_support = """