Update app.py

app.py

@@ -4,7 +4,7 @@ import traceback
 from datetime import datetime
 from typing import Optional
 
-from fastapi import FastAPI, File, UploadFile, Form, HTTPException, Header, Depends
+from fastapi import FastAPI, File, UploadFile, Form, HTTPException
 from fastapi.responses import StreamingResponse, JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
@@ -47,19 +47,6 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# ---------------------------------------------------------------------
-# Authentication
-# ---------------------------------------------------------------------
-BEARER_TOKEN = "logicgo@123"
-
-def verify_token(authorization: str = Header(None)):
-    if not authorization or not authorization.startswith("Bearer "):
-        raise HTTPException(status_code=401, detail="Missing or invalid Authorization header")
-    token = authorization.split(" ")[1]
-    if token != BEARER_TOKEN:
-        raise HTTPException(status_code=403, detail="Invalid bearer token")
-    return True
-
 # ---------------------------------------------------------------------
 # Models
 # ---------------------------------------------------------------------
@@ -79,12 +66,12 @@ def health():
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"DB ping failed: {e}")
 
+
 @app.post("/generate")
 async def generate(
     prompt: str = Form(...),
     image1: UploadFile = File(...),
     image2: Optional[UploadFile] = File(None),
-    authorized: bool = Depends(verify_token)
 ):
     """Upload 1 or 2 images + prompt and get edited image via HF Inference"""
     # Read images
@@ -119,18 +106,13 @@ async def generate(
         if pil_img2:
             images_to_pass.append(pil_img2)
 
-        # The "Qwen/Qwen-Image-Edit" expects a list of PIL images and a prompt
         pil_output = hf_client.image_to_image(
             images=images_to_pass,
             prompt=prompt,
             model="Qwen/Qwen-Image-Edit"
         )
 
-        # Handle list return
-        if isinstance(pil_output, list):
-            output_image = pil_output[0]
-        else:
-            output_image = pil_output
+        output_image = pil_output[0] if isinstance(pil_output, list) else pil_output
 
         out_buf = io.BytesIO()
         output_image.save(out_buf, format="PNG")
@@ -186,6 +168,7 @@ def get_image(image_id: str, download: Optional[bool] = False):
 
     return StreamingResponse(iterfile(), media_type=grid_out.content_type or "application/octet-stream", headers=headers)
 
+
 # ---------------------------------------------------------------------
 # Run locally
 # ---------------------------------------------------------------------