Update main.py

main.py

@@ -1060,81 +1060,53 @@ async def get_batch_files(batch_id: str):
         files=files
     )
 
-@app.get("/download-all")
-async def download_all_files():
-    """Get list of all downloadable files on the server"""
-    try:
-        # Get all files from the download directory
-        download_dir = Path(downloader.download_dir)
-        all_files = [f for f in download_dir.glob("*") if f.is_file()]
-        
-        if not all_files:
-            raise HTTPException(status_code=404, detail="No files found on server")
-        
-        # Return list of files with download URLs
-        file_list = []
-        for file_path in all_files:
-            filename = file_path.name
-            file_list.append({
-                "filename": filename,
-                "url": f"/video/file/{filename}",
-                "size": os.path.getsize(file_path)
-            })
-        
-        return {
-            "success": True,
-            "total_files": len(file_list),
-            "files": file_list
-        }
-        
-    except HTTPException:
-        raise
-    except Exception as e:
-        logger.error(f"Error listing all files: {e}")
-        raise HTTPException(status_code=500, detail=str(e))
+
+
+from fastapi.responses import RedirectResponse
 
 @app.get("/batch/download-all/{batch_id}")
 async def download_batch_files(batch_id: str):
-    """Get list of all downloadable files for a specific batch"""
-    try:
-        if batch_id not in batch_status_store:
-            raise HTTPException(status_code=404, detail="Batch not found")
+    """Download all files from a specific batch"""
+    if batch_id not in batch_status_store:
+        raise HTTPException(status_code=404, detail="Batch not found")
 
-        batch_status = batch_status_store[batch_id]
-        
-        if batch_status["status"] != "completed":
-            raise HTTPException(
-                status_code=400, 
-                detail=f"Batch is not completed yet. Current status: {batch_status['status']}"
+    batch_status = batch_status_store[batch_id]
+    
+    if batch_status["status"] != "completed":
+        raise HTTPException(
+            status_code=400, 
+            detail=f"Batch not completed. Status: {batch_status['status']}"
+        )
+    
+    # Get first available file from batch
+    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")
+
+@app.get("/batch/download-all")
+async def download_all_files():
+    """Download all available files (returns first found file)"""
+    download_dir = Path(downloader.download_dir)
+    
+    # Get first available file in download directory
+    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'
             )
-        
-        # Get all successfully downloaded files
-        file_list = []
-        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):
-                    file_list.append({
-                        "filename": os.path.basename(file_path),
-                        "url": f"/video/file/{os.path.basename(file_path)}",
-                        "size": os.path.getsize(file_path)
-                    })
-        
-        if not file_list:
-            raise HTTPException(status_code=404, detail="No files found for this batch")
-        
-        return {
-            "success": True,
-            "batch_id": batch_id,
-            "total_files": len(file_list),
-            "files": file_list
-        }
-        
-    except HTTPException:
-        raise
-    except Exception as e:
-        logger.error(f"Error getting batch files: {e}")
-        raise HTTPException(status_code=500, detail=str(e))
+    
+    raise HTTPException(status_code=404, detail="No files available for download")
     
 @app.get("/video/file/{filename}")
 async def download_file(filename: str):