Update app.py

app.py

@@ -122,28 +122,34 @@ async def upload_zip(file: UploadFile = File(...)):
     
 # upload a single file from UI
 @app.post("/upload")
-async def upload_image(file: UploadFile = File(...)):
-    filename = file.filename
-    contents = await file.read()
+async def upload_images(files: List[UploadFile] = File(...)):
+    responses = []
 
-    # Save temporarily
-    temp_path = f"/tmp/{filename}"
-    with open(temp_path, "wb") as f:
-        f.write(contents)
+    for file in files:
+        filename = file.filename
+        contents = await file.read()
 
-    try:
-        upload_file(
-            path_or_fileobj=temp_path,
-            path_in_repo=f"demo/{filename}",
-            repo_id="rahul7star/ohamlab",
-            repo_type="model",
-            commit_message=f"Upload {filename} to demo folder",
-            token=True,  # uses HF_TOKEN from Space secrets
-        )
-        return {"message": f"✅ {filename} uploaded successfully!"}
-    except Exception as e:
-        return {"error": f"❌ Upload failed: {str(e)}"}
+        temp_path = f"/tmp/{filename}"
+        with open(temp_path, "wb") as f:
+            f.write(contents)
+
+        try:
+            upload_file(
+                path_or_fileobj=temp_path,
+                path_in_repo=f"demo/{filename}",
+                repo_id="rahul7star/ohamlab",
+                repo_type="model",
+                commit_message=f"Upload {filename} to demo folder",
+                token=True,
+            )
+            responses.append({"filename": filename, "status": "✅ uploaded"})
+        except Exception as e:
+            responses.append({"filename": filename, "status": f"❌ failed: {str(e)}"})
+
+        # Optional: clean up local temp file
+        os.remove(temp_path)
 
+    return {"results": responses}