Update main.py

main.py

@@ -1066,7 +1066,7 @@ from fastapi.responses import RedirectResponse
 
 @app.get("/batch/download-all/{batch_id}")
 async def download_batch_files(batch_id: str):
-    """Download all files from a specific batch"""
+    """Get JSON list of all downloadable files for a specific batch"""
     if batch_id not in batch_status_store:
         raise HTTPException(status_code=404, detail="Batch not found")
 
@@ -1078,35 +1078,40 @@ async def download_batch_files(batch_id: str):
             detail=f"Batch not completed. Status: {batch_status['status']}"
         )
     
-    # Get first available file from batch
+    files = []
     for result in batch_status["results"]:
         if result.get("success") and result.get("download_path"):
             file_path = result["download_path"]
             if os.path.exists(file_path):
-                filename = os.path.basename(file_path)
-                return FileResponse(
-                    path=file_path,
-                    filename=filename,
-                    media_type='application/octet-stream'
-                )
-    
-    raise HTTPException(status_code=404, detail="No downloadable files found in this batch")
+                files.append({
+                    "filename": os.path.basename(file_path),
+                    "url": f"/video/file/{os.path.basename(file_path)}"
+                })
+
+    if not files:
+        raise HTTPException(status_code=404, detail="No downloadable files found in this batch")
+
+    return {"batch_id": batch_id, "total_files": len(files), "downloads": files}
+
 
 @app.get("/batch/download-all")
 async def download_all_files():
-    """Download all available files (returns first found file)"""
+    """Get JSON list of all available files in the download directory"""
     download_dir = Path(downloader.download_dir)
     
-    # Get first available file in download directory
+    files = []
     for file_path in download_dir.glob("*"):
         if file_path.is_file():
-            return FileResponse(
-                path=str(file_path),
-                filename=file_path.name,
-                media_type='application/octet-stream'
-            )
+            files.append({
+                "filename": file_path.name,
+                "url": f"/video/file/{file_path.name}"
+            })
     
-    raise HTTPException(status_code=404, detail="No files available for download")
+    if not files:
+        raise HTTPException(status_code=404, detail="No files available for download")
+
+    return {"total_files": len(files), "downloads": files}
+
     
 @app.get("/video/file/{filename}")
 async def download_file(filename: str):