Update main.py

main.py

@@ -29,6 +29,7 @@ import tempfile
 import io
 import string
 import openai
+import uuid
 from io import BytesIO
 from datetime import datetime as dt
 from dotenv import load_dotenv
@@ -147,6 +148,36 @@ Please modify your search terms and/or try again later thank you for your unders
 ~ @xtdevs Team
 """
 
+UPLOAD_DIRECTORY = "./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_DIRECTORY, unique_filename)
+        with open(file_location, "wb") as f:
+            f.write(await file.read())
+        return JSONResponse(
+            status_code=200,
+            content={"url": f"https://randydev-ryuzaki-api.hf.space/api/v1/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_DIRECTORY, filename)
+    if os.path.exists(file_location):
+        return FileResponse(file_location)
+    return JSONResponse(
+        status_code=404,
+        content={"error": "File not found"}
+    )
+
 class FedBans(BaseModel):
     user_id: int
     hashtag: str