Update app.py

app.py

@@ -12,18 +12,18 @@ token = os.getenv("HF_TOKEN")      # Set in Space secrets
 
 # --- FastAPI Endpoint ---
 @app.post("/upload/")
-async def upload_file(file: UploadFile = File(...)):
+async def upload_file(folder_name: str, file: UploadFile = File(...)):
     file_path = f"/tmp/temp_{file.filename}"
     with open(file_path, "wb") as f:
         f.write(await file.read())
     
-    # Upload to Hugging Face using the original filename
+    # Upload to Hugging Face with dynamic folder name provided by the user
     api.upload_file(
         path_or_fileobj=file_path,
-        path_in_repo=f"files/{file.filename}",  # Uses the original filename
+        path_in_repo=f"{folder_name}/{file.filename}",  # Dynamic folder name
         repo_id=repo_id,
         repo_type="model",
         token=token
     )
     os.remove(file_path)
-    return {"message": f"File uploaded as {file.filename}"}
+    return {"message": f"File uploaded as {file.filename} to {folder_name}/{file.filename}"}