Update api/main.py

api/main.py

@@ -423,6 +423,24 @@ def _load_rgba_image(path: str) -> Image.Image:
     return img.convert("RGBA")
 
 
+def _compress_image(image_path: str, output_path: str, quality: int = 85) -> None:
+    """
+    Compress an image to reduce file size.
+    Converts to JPEG format with specified quality to achieve smaller file size.
+    """
+    img = Image.open(image_path)
+    # Convert RGBA to RGB if needed (JPEG doesn't support alpha)
+    if img.mode == "RGBA":
+        rgb_img = Image.new("RGB", img.size, (255, 255, 255))
+        rgb_img.paste(img, mask=img.split()[3])  # Use alpha channel as mask
+        img = rgb_img
+    elif img.mode != "RGB":
+        img = img.convert("RGB")
+    
+    # Save as JPEG with quality setting for compression
+    img.save(output_path, "JPEG", quality=quality, optimize=True)
+
+
 def _load_rgba_mask_from_image(img: Image.Image) -> np.ndarray:
     """
     Convert mask image to RGBA format (black/white mask).
@@ -477,11 +495,12 @@ def _load_rgba_mask_from_image(img: Image.Image) -> np.ndarray:
     return rgba
 
 @app.post("/inpaint")
-def inpaint(req: InpaintRequest, _: None = Depends(bearer_auth)) -> Dict[str, str]:
+def inpaint(req: InpaintRequest, request: Request, _: None = Depends(bearer_auth)) -> Dict[str, str]:
     start_time = time.time()
     status = "success"
     error_msg = None
     output_name = None
+    compressed_url = None
 
     try:
         if req.image_id not in file_store or file_store[req.image_id]["type"] != "image":
@@ -510,8 +529,21 @@ def inpaint(req: InpaintRequest, _: None = Depends(bearer_auth)) -> Dict[str, st
             output_path, "PNG", optimize=False, compress_level=1
         )
 
+        # Create compressed version
+        compressed_name = f"compressed_{output_name.replace('.png', '.jpg')}"
+        compressed_path = os.path.join(OUTPUT_DIR, compressed_name)
+        try:
+            _compress_image(output_path, compressed_path, quality=85)
+            compressed_url = str(request.url_for("download_file", filename=compressed_name))
+        except Exception as compress_err:
+            log.warning("Failed to create compressed image: %s", compress_err)
+            compressed_url = None
+
         log_media_click(req.user_id, req.category_id)
-        return {"result": output_name}
+        response = {"result": output_name}
+        if compressed_url:
+            response["Compressed_Image_URL"] = compressed_url
+        return response
 
     except Exception as e:
         status = "fail"
@@ -949,7 +981,6 @@ def view_result(filename: str):
 @app.get("/logs")
 def get_logs(_: None = Depends(bearer_auth)) -> JSONResponse:
     return JSONResponse(content=logs)
-
 # import os
 # import uuid
 # import shutil
@@ -1145,6 +1176,14 @@ def get_logs(_: None = Depends(bearer_auth)) -> JSONResponse:
 #         }
 #     )
 
+#     # Sort by date ascending (older dates first) to guarantee stable ordering:
+#     # [oldest, ..., newest]
+#     try:
+#         result.sort(key=lambda entry: _to_date_only(entry.get("date")))
+#     except Exception:
+#         # If anything goes wrong during sort, fall back to current ordering
+#         pass
+
 #     # Enforce 32-entry limit (keep the most recent 32 days)
 #     if len(result) > 32:
 #         result = result[-32:]
@@ -1269,7 +1308,6 @@ def get_logs(_: None = Depends(bearer_auth)) -> JSONResponse:
 #                         ],
 #                         "createdAt": now,
 #                         "updatedAt": now,
-#                         "ai_edit_complete": 0,  # Default for new users
 #                         "ai_edit_daily_count": daily_for_new,
 #                     },
 #                     "$inc": {"ai_edit_complete": 1},  # Increment to 1 on first use