Update app.py

app.py

@@ -1,10 +1,12 @@
 import os
+import uuid
 from typing import List
 from fastapi import FastAPI, UploadFile, File, HTTPException
 from fastapi.responses import JSONResponse
 import dagshub
 from dagshub import upload_files
 from dagshub.data_engine import datasources
+
 # =====================================================
 # CONFIG
 # =====================================================
@@ -12,11 +14,12 @@ from dagshub.data_engine import datasources
 REPO_OWNER = "Rizwanali324"
 REPO_NAME = "my-first-repo"
 FULL_REPO = f"{REPO_OWNER}/{REPO_NAME}"
+
 REMOTE_FOLDER = "data"
-BRANCH = "main"  # change if your repo uses another branch
+BRANCH = "main"
 
 # =====================================================
-# AUTHENTICATION (CI SAFE)
+# AUTHENTICATION
 # =====================================================
 
 if "DAGSHUB_USER_TOKEN" not in os.environ:
@@ -24,7 +27,6 @@ if "DAGSHUB_USER_TOKEN" not in os.environ:
         "DAGSHUB_USER_TOKEN secret not found. Add it in Space Settings → Secrets."
     )
 
-# Initialize DagsHub client
 dagshub.init(
     repo_owner=REPO_OWNER,
     repo_name=REPO_NAME,
@@ -35,107 +37,109 @@ dagshub.init(
 # FASTAPI INIT
 # =====================================================
 
-app = FastAPI(title="DagsHub Image Uploader API")
+app = FastAPI(title="Wildlife Detection Image API")
+
+# local temp folder
+os.makedirs("temp_uploads", exist_ok=True)
+
 
 # =====================================================
-# UPLOAD ENDPOINT
+# IMAGE UPLOAD ENDPOINT
 # =====================================================
 
 @app.post("/upload")
 async def upload_images(
-    files: List[UploadFile] = File(
-        ..., description="Upload one or more image files"
-    )
+    files: List[UploadFile] = File(..., description="Upload image files")
 ):
-    """
-    Upload one or more images and push to DagsHub.
-    Returns list of public URLs for uploaded files.
-    """
+
     if not files:
         raise HTTPException(status_code=400, detail="No files uploaded")
 
-    os.makedirs(REMOTE_FOLDER, exist_ok=True)
-    uploaded_urls = []
+    uploaded_files = []
 
     for file in files:
+
         if not file.content_type.startswith("image/"):
             raise HTTPException(
                 status_code=400,
-                detail=f"{file.filename} is not a valid image",
+                detail=f"{file.filename} is not a valid image"
             )
 
         original_name = file.filename
-        local_path = os.path.join(REMOTE_FOLDER, original_name)
 
-        # Save locally
+        # generate unique filename
+        unique_id = uuid.uuid4().hex
+        stored_name = f"{unique_id}_{original_name}"
+
+        local_path = os.path.join("temp_uploads", stored_name)
+
         try:
+            content = await file.read()
+
+            # save locally
             with open(local_path, "wb") as f:
-                f.write(await file.read())
+                f.write(content)
+
         except Exception as e:
             raise HTTPException(
-                status_code=500, detail=f"Failed to save {original_name}: {str(e)}"
+                status_code=500,
+                detail=f"Failed saving {original_name}: {str(e)}"
             )
 
-        # Upload to DagsHub
+        # upload to dagshub
         try:
             upload_files(
                 repo=FULL_REPO,
                 local_path=local_path,
-                remote_path=f"{REMOTE_FOLDER}/{original_name}"
+                remote_path=f"{REMOTE_FOLDER}/{stored_name}"
             )
+
         except Exception as e:
             raise HTTPException(
                 status_code=500,
-                detail=f"Failed to upload {original_name} to DagsHub: {str(e)}"
+                detail=f"Upload failed for {original_name}: {str(e)}"
             )
 
-        # Generate public URL
+        # public url
         file_url = (
-            f"https://dagshub.com/{REPO_OWNER}/{REPO_NAME}/raw/{BRANCH}/"
-            f"{REMOTE_FOLDER}/{original_name}"
+            f"https://dagshub.com/{REPO_OWNER}/{REPO_NAME}/raw/"
+            f"{BRANCH}/{REMOTE_FOLDER}/{stored_name}"
         )
-        uploaded_urls.append(file_url)
+
+        uploaded_files.append({
+            "original_filename": original_name,
+            "stored_filename": stored_name,
+            "url": file_url
+        })
 
     return JSONResponse(
         content={
-            "message": "Upload successful",
-            "count": len(uploaded_urls),
-            "files": uploaded_urls,
+            "message": "Images uploaded successfully",
+            "count": len(uploaded_files),
+            "files": uploaded_files
         }
     )
 
 
-
 # =====================================================
-# DATA FETCH ENDPOINT
+# FETCH DATA FROM DAGSHUB
 # =====================================================
 
 @app.get("/data")
 def get_uploaded_data(min_size: int = None):
-    """
-    Fetch uploaded data from DagsHub Data Engine.
-    Optional query param:
-    - min_size: return files larger than this size (in bytes)
-    """
 
     try:
-        # Connect to datasource
         ds = datasources.get_datasource(FULL_REPO)
 
-        # Base query
         query = ds
 
-        # Apply size filter if provided
         if min_size:
             query = ds["size"] > min_size
 
-        # Select only useful columns
         query = query.select("path", "size")
 
-        # Execute query
         df = query.all().dataframe
 
-        # Convert to list of dicts
         records = df.to_dict(orient="records")
 
         return {
@@ -147,13 +151,15 @@ def get_uploaded_data(min_size: int = None):
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
+
 # =====================================================
 # HEALTH CHECK
 # =====================================================
 
 @app.get("/")
 def root():
-    return {"status": "API is running"}
+    return {"status": "API running successfully"}
+
 
 # =====================================================
 # RUN SERVER