Update app.py

app.py

@@ -129,9 +129,10 @@ from typing import List
 from datetime import datetime
 import uuid, os
 
+
 @app.post("/upload")
 async def upload_images(files: List[UploadFile] = File(...)):
-    # Step 1: Create unique folder name
+    # Step 1: Generate dynamic folder name
     timestamp = datetime.utcnow().strftime("%Y%m%d_%H%M%S")
     unique_id = uuid.uuid4().hex[:6]
     folder_name = f"upload_{timestamp}_{unique_id}"
@@ -139,20 +140,19 @@ async def upload_images(files: List[UploadFile] = File(...)):
 
     responses = []
 
+    # Step 2: Save and upload each image
     for file in files:
         filename = file.filename
         contents = await file.read()
-
         temp_path = f"/tmp/{filename}"
         with open(temp_path, "wb") as f:
             f.write(contents)
 
         try:
-            # Upload to Hugging Face with dynamic folder path
             upload_file(
                 path_or_fileobj=temp_path,
                 path_in_repo=f"{hf_folder_prefix}/{filename}",
-                repo_id="rahul7star/ohamlab",
+                repo_id=T_REPO_ID,
                 repo_type="model",
                 commit_message=f"Upload {filename} to {hf_folder_prefix}",
                 token=True,
@@ -168,13 +168,22 @@ async def upload_images(files: List[UploadFile] = File(...)):
                 "status": f"❌ failed: {str(e)}"
             })
 
-        # Clean up local temp file
+        # 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)}"
+        }
+
     return {
-        "message": f"{len(files)} file(s) processed",
+        "message": f"{len(files)} file(s) uploaded",
         "upload_folder": hf_folder_prefix,
-        "results": responses
+        "results": responses,
+        "filter_result": filter_result
     }