Update app.py

app.py

@@ -179,50 +179,7 @@ def compress_pil_image_to_2mb(
         quality -= 5
 
     return buffer.getvalue()
-
-# def run_image_generation(
-#     image: Image.Image,
-#     prompt: str
-# ) -> Image.Image:
-#     """
-#     Unified image generation interface.
-#     Returns PIL Image.
-#     """
-
-#     if MODEL == "QWEN":
-#         return hf_client.image_to_image(
-#             image=image,
-#             prompt=prompt,
-#             model="Qwen/Qwen-Image-Edit"
-#         )
-
-#     elif MODEL == "GEMINI":
-#         model = genai.GenerativeModel(GEMINI_IMAGE_MODEL)
-
-#         # Gemini expects bytes
-#         img_buffer = io.BytesIO()
-#         image.save(img_buffer, format="PNG")
-#         img_buffer.seek(0)
-
-#         response = model.generate_content(
-#             [
-#                 prompt,
-#                 {
-#                     "mime_type": "image/png",
-#                     "data": img_buffer.getvalue()
-#                 }
-#             ]
-#         )
-
-#         # Extract image bytes from response
-#         for part in response.candidates[0].content.parts:
-#             if hasattr(part, "inline_data"):
-#                 return Image.open(io.BytesIO(part.inline_data.data)).convert("RGB")
-
-#         raise RuntimeError("Gemini did not return an image")
-
-#     else:
-#         raise RuntimeError(f"Unsupported IMAGE_MODEL: {MODEL}")
+    
 def run_image_generation(
     image1: Image.Image,
     prompt: str,
@@ -373,18 +330,6 @@ async def generate(
         except Exception as e:
             raise HTTPException(400, f"Failed to read second image: {e}")
 
-    # -------------------------
-    # 2. COMBINE IF NEEDED
-    # -------------------------
-    # if pil_img2:
-    #     total_width = pil_img1.width + pil_img2.width
-    #     max_height = max(pil_img1.height, pil_img2.height)
-    #     combined_img = Image.new("RGB", (total_width, max_height))
-    #     combined_img.paste(pil_img1, (0, 0))
-    #     combined_img.paste(pil_img2, (pil_img1.width, 0))
-    # else:
-    #     combined_img = pil_img1
-
     # -------------------------
     # 3. CATEGORY CLICK LOGIC
     # -------------------------
@@ -525,16 +470,6 @@ async def generate(
     # 4. HF INFERENCE
     # -------------------------
     try:
-        # pil_output = hf_client.image_to_image(
-        #     image=combined_img,
-        #     prompt=prompt,
-        #     model="Qwen/Qwen-Image-Edit"
-        # )
-        #-------------------------------------#
-        # pil_output = run_image_generation(
-        # image=combined_img,
-        # prompt=prompt
-        # )
         # --------------------------------------------------
         # MODEL OVERRIDE BASED ON CATEGORY
         # --------------------------------------------------
@@ -629,119 +564,6 @@ async def generate(
             f"https://logicgoinfotechspaces-polaroidimage.hf.space/image/{compressed_id}"
         )
     })
-####################---------------------------------------------------------------------------------------------------###
-###-----OLD CODE--------------####
-# @app.post("/generate")
-# async def generate(
-#     prompt: str = Form(...),
-#     image1: UploadFile = File(...),
-#     image2: Optional[UploadFile] = File(None),
-#     user=Depends(verify_firebase_token)
-# ):
-#     start_time = time.time()
-
-#     # PREPARE IMAGE(S)
-#     try:
-#         img1_bytes = await image1.read()
-#         pil_img1 = prepare_image(img1_bytes)
-#         input1_id = fs.put(
-#             img1_bytes,
-#             filename=image1.filename,
-#             contentType=image1.content_type,
-#             metadata={"role": "input"}
-#         )
-#     except Exception as e:
-#         raise HTTPException(status_code=400, detail=f"Failed to read first image: {e}")
-
-#     img2_bytes = None
-#     input2_id = None
-#     pil_img2 = None
-
-#     if image2:
-#         try:
-#             img2_bytes = await image2.read()
-#             pil_img2 = prepare_image(img2_bytes)
-#             input2_id = fs.put(
-#                 img2_bytes,
-#                 filename=image2.filename,
-#                 contentType=image2.content_type,
-#                 metadata={"role": "input"}
-#             )
-#         except Exception as e:
-#             raise HTTPException(status_code=400, detail=f"Failed to read second image: {e}")
-
-#     # COMBINE IF NEEDED
-#     if pil_img2:
-#         total_width = pil_img1.width + pil_img2.width
-#         max_height = max(pil_img1.height, pil_img2.height)
-#         combined_img = Image.new("RGB", (total_width, max_height))
-#         combined_img.paste(pil_img1, (0, 0))
-#         combined_img.paste(pil_img2, (pil_img1.width, 0))
-#     else:
-#         combined_img = pil_img1
-
-#     # INFERENCE
-#     try:
-#         pil_output = hf_client.image_to_image(
-#             image=combined_img,
-#             prompt=prompt,
-#             model="Qwen/Qwen-Image-Edit"
-#         )
-#     except Exception as e:
-#         # LOG FAILURE
-#         response_time_ms = round((time.time() - start_time) * 1000)
-#         logs_collection.insert_one({
-#             "timestamp": datetime.utcnow(),
-#             "status": "failure",
-#             "input1_id": str(input1_id),
-#             "input2_id": str(input2_id) if input2_id else None,
-#             "prompt": prompt,
-#             "user_email": user.get("email"),
-#             "error": str(e),
-#             "response_time_ms": response_time_ms
-#         })
-#         raise HTTPException(status_code=500, detail=f"Inference failed: {e}")
-
-#     # SAVE OUTPUT IMAGE
-#     out_buf = io.BytesIO()
-#     pil_output.save(out_buf, format="PNG")
-#     out_bytes = out_buf.getvalue()
-
-#     out_id = fs.put(
-#         out_bytes,
-#         filename=f"result_{input1_id}.png",
-#         contentType="image/png",
-#         metadata={
-#             "role": "output",
-#             "prompt": prompt,
-#             "input1_id": str(input1_id),
-#             "input2_id": str(input2_id) if input2_id else None,
-#             "user_email": user.get("email"),
-#         }
-#     )
-
-#     # FINAL RESPONSE TIME
-#     response_time_ms = round((time.time() - start_time) * 1000)
-
-#     # LOG SUCCESS - THIS WAS MISSING THE STATUS AND RESPONSE_TIME_MS FIELDS!
-#     logs_collection.insert_one({
-#         "timestamp": datetime.utcnow(),
-#         "status": "success",  # ✅ FIXED: Added status field
-#         "input1_id": str(input1_id),
-#         "input2_id": str(input2_id) if input2_id else None,
-#         "output_id": str(out_id),
-#         "prompt": prompt,
-#         "user_email": user.get("email"),  # ✅ FIXED: Changed from "user" to "user_email"
-#         "response_time_ms": response_time_ms  # ✅ FIXED: Added response_time_ms field
-#     })
-
-#     return JSONResponse({
-#         "output_image_id": str(out_id),
-#         "user": user.get("email"),
-#         "response_time_ms": response_time_ms
-#     })
-
-
 
 @app.get("/image/{image_id}")
 def get_image(image_id: str, download: Optional[bool] = False):
@@ -765,7 +587,6 @@ def get_image(image_id: str, download: Optional[bool] = False):
         headers=headers
     )
 
-
 # ---------------------------------------------------------------------
 # Run locally
 # ---------------------------------------------------------------------