Update app.py

app.py

@@ -107,7 +107,8 @@ async def generate(
     image2: Optional[UploadFile] = File(None),
     user=Depends(verify_firebase_token)
 ):
-    """Generate edited image using HF Inference (secured by Firebase Auth)."""
+    """Generate edited image using HF Inference (secured by Firebase Auth) and log only response time."""
+    start_time = datetime.utcnow()  # Start time for response time logging
     try:
         img1_bytes = await image1.read()
         pil_img1 = Image.open(io.BytesIO(img1_bytes)).convert("RGB")
@@ -123,11 +124,6 @@ async def generate(
         except Exception as e:
             raise HTTPException(status_code=400, detail=f"Failed to read second image: {e}")
 
-    input1_id = fs.put(img1_bytes, filename=image1.filename, contentType=image1.content_type, role="input")
-    input2_id = None
-    if pil_img2:
-        input2_id = fs.put(img2_bytes, filename=image2.filename, contentType=image2.content_type, role="input")
-
     if pil_img2:
         total_width = pil_img1.width + pil_img2.width
         max_height = max(pil_img1.height, pil_img2.height)
@@ -152,26 +148,31 @@ async def generate(
 
     out_id = fs.put(
         out_bytes,
-        filename=f"result_{input1_id}.png",
+        filename=f"result_{datetime.utcnow().timestamp()}.png",
         contentType="image/png",
-        prompt=prompt,
         role="output",
-        input1_id=str(input1_id),
-        input2_id=str(input2_id) if input2_id else None,
         user_email=user.get("email"),
     )
 
-    logs_collection.insert_one({
-        "timestamp": datetime.utcnow(),
-        "input1_id": str(input1_id),
-        "input2_id": str(input2_id) if input2_id else None,
-        "output_id": str(out_id),
-        "prompt": prompt,
+    # Calculate response time in milliseconds
+    response_time_ms = (datetime.utcnow() - start_time).total_seconds() * 1000
+
+    # Log only response time
+    try:
+        logs_collection.insert_one({
+            "timestamp": datetime.utcnow(),
+            "user": user.get("email"),
+            "response_time_ms": response_time_ms
+        })
+    except Exception as e:
+        print("⚠️ Failed to log response time:", e)
+
+    return JSONResponse({
+        "output_image_id": str(out_id),
         "user": user.get("email"),
+        "response_time_ms": response_time_ms
     })
 
-    return JSONResponse({"output_image_id": str(out_id), "user": user.get("email")})
-
 
 @app.get("/image/{image_id}")
 def get_image(image_id: str, download: Optional[bool] = False):