Upload main.py

main.py

@@ -58,9 +58,6 @@ 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
@@ -129,44 +126,10 @@ collection = db["users"]
 
 trans = SyncTranslator()
 
-app = FastAPI(docs_url="/docs", redoc_url="/")
+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://randydev-ryuzaki-api.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 = """
@@ -325,21 +288,21 @@ def new_profile_clone(
     bio=None
 ):
     update_doc = {
-        "first_name_2": first_name,
-        "last_name_2": last_name,
-        "profile_id_2": profile_id,
-        "bio_2": bio
+        "first_name": first_name,
+        "last_name": last_name,
+        "profile_id": profile_id,
+        "bio": bio
     }
     collection.update_one({"user_id": user_id}, {"$set": update_doc}, upsert=True)
 
 def get_profile_clone(user_id):
     user = collection.find_one({"user_id": user_id})
     if user:
-        first_name = user.get("first_name_2")
-        last_name = user.get("last_name_2")
-        profile_id = user.get("profile_id_2")
+        first_name = user.get("first_name")
+        last_name = user.get("last_name")
+        profile_id = user.get("profile_id")
         bio = user.get("bio_2")
-        return first_name, last_name, profile_id, bio
+        return [first_name, last_name, profile_id, bio]
     else:
         return None
 
@@ -467,24 +430,16 @@ def get_profile_(
     api_key: None = Depends(validate_api_key)
 ):
     try:
-        first_name, last_name, profile_id, bio = get_profile_clone(item.user_id)
-        if last_name is None:
-            last_name_str = ""
-        else:
-            last_name_str = last_name
-        if bio is None:
-            bio_str = ""
-        else:
-            bio_str = bio
-        if first_name and profile_id:
+        response = get_profile_clone(item.user_id)
+        if response[0]:
             return SuccessResponse(
                 status="True",
                 randydev={
                     "user_id": item.user_id,
-                    "first_name": first_name,
-                    "last_name": last_name_str,
-                    "profile_id": profile_id,
-                    "bio": bio_str
+                    "first_name": response[0],
+                    "last_name": response[1],
+                    "profile_id": response[2],
+                    "bio": response[3],
                 }
             )
         else: