Update app.py

app.py

@@ -8,7 +8,7 @@ from fastapi import FastAPI, Query
 from huggingface_hub import list_repo_files, hf_hub_download, upload_file
 import io
 import requests
-
+from fastapi import BackgroundTasks
 from fastapi import FastAPI, UploadFile, File
 from fastapi.middleware.cors import CORSMiddleware
 
@@ -131,7 +131,10 @@ import uuid, os
 
 
 @app.post("/upload")
-async def upload_images(files: List[UploadFile] = File(...)):
+async def upload_images(
+    background_tasks: BackgroundTasks,
+    files: List[UploadFile] = File(...)
+):
     # Step 1: Generate dynamic folder name
     timestamp = datetime.utcnow().strftime("%Y%m%d_%H%M%S")
     unique_id = uuid.uuid4().hex[:6]
@@ -168,22 +171,23 @@ async def upload_images(files: List[UploadFile] = File(...)):
                 "status": f"❌ failed: {str(e)}"
             })
 
-        # Clean up
         os.remove(temp_path)
 
-    # Step 3: Call the filter function after upload
-    try:
-        filter_result = filter_and_rename_images(folder=hf_folder_prefix)
-    except Exception as e:
-        filter_result = {
-            "error": f"❌ Filtering failed: {str(e)}"
-        }
+    # Step 3: Add filter job to background
+    def run_filter():
+        try:
+            result = filter_and_rename_images(folder=hf_folder_prefix)
+            print(f"🧼 Filter result: {result}")
+        except Exception as e:
+            print(f"❌ Filter failed: {str(e)}")
+
+    background_tasks.add_task(run_filter)
 
     return {
         "message": f"{len(files)} file(s) uploaded",
         "upload_folder": hf_folder_prefix,
         "results": responses,
-        "filter_result": filter_result
+        "note": "Filtering started in background"
     }