Update app.py

app.py

@@ -10,6 +10,7 @@ from datetime import datetime
 import tempfile
 from pathlib import Path
 from dotenv import load_dotenv
+from urllib.parse import quote
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -71,18 +72,28 @@ os.makedirs(DOWNLOADS_DIR, exist_ok=True)
 # Mount the downloads directory
 app.mount("/files", StaticFiles(directory=DOWNLOADS_DIR), name="files")
 
+
+
+
+# [Previous cookie handling code remains the same...]
+
+
+
+
+# [Previous cookie handling code remains the same...]
+
 class DownloadRequest(BaseModel):
     url: str
     
+class FileInfo(BaseModel):
+    filename: str
+    download_url: str
+    file_type: str
+
 class DownloadResponse(BaseModel):
     success: bool
     message: str
-    filename: str = None
-    download_url: str = None
-
-
-
-# [Previous cookie handling code remains the same...]
+    files: List[FileInfo]
 
 @app.post("/download", response_model=DownloadResponse)
 async def download_file(request: DownloadRequest):
@@ -124,40 +135,49 @@ async def download_file(request: DownloadRequest):
         for root, dirs, files in os.walk('.'):
             for file in files:
                 if file != "cookies.txt":
-                    all_files.append(os.path.join(root, file))
+                    file_path = os.path.join(root, file)
+                    all_files.append(file_path)
         
         logger.info(f"All files found: {all_files}")
         
         if not all_files:
             raise Exception("No files found after download attempt")
         
-        # Get the first downloaded file (usually there's only one)
-        downloaded_file = all_files[0]
-        if downloaded_file.startswith('./'):
-            downloaded_file = downloaded_file[2:]  # Remove leading ./
+        # Process all downloaded files
+        downloaded_files = []
+        space_url = os.getenv("SPACE_URL", "http://localhost:7860")
+        
+        for file_path in all_files:
+            # Get the filename without the directory structure
+            filename = os.path.basename(file_path)
             
-        # Move the file to the download directory root if it's in a subdirectory
-        if os.path.dirname(downloaded_file):
-            new_path = os.path.basename(downloaded_file)
-            shutil.move(downloaded_file, new_path)
-            downloaded_file = new_path
+            # Move file to download directory root if it's in a subdirectory
+            if os.path.dirname(file_path):
+                new_path = os.path.join(download_subdir, filename)
+                shutil.move(file_path, new_path)
             
-        logger.info(f"Final downloaded file: {downloaded_file}")
-        
-        # Generate the download URL with proper URL encoding
-        space_url = os.getenv("SPACE_URL", "http://tecuts-testing.hf.space")
-        encoded_filename = quote(downloaded_file)
-        download_url = f"{space_url}/files/{timestamp}/{encoded_filename}"
-        logger.info(f"Generated download URL: {download_url}")
+            # Get file extension
+            file_type = os.path.splitext(filename)[1].lstrip('.')
+            
+            # Generate download URL
+            encoded_filename = quote(filename)
+            download_url = f"{space_url}/files/{timestamp}/{encoded_filename}"
+            
+            downloaded_files.append(FileInfo(
+                filename=filename,
+                download_url=download_url,
+                file_type=file_type
+            ))
+            
+            logger.info(f"Processed file: {filename} -> {download_url}")
         
         # Move back to original directory
         os.chdir(original_dir)
         
         return DownloadResponse(
             success=True,
-            message="File downloaded successfully",
-            filename=downloaded_file,
-            download_url=download_url
+            message=f"Successfully downloaded {len(downloaded_files)} files",
+            files=downloaded_files
         )
             
     except subprocess.CalledProcessError as e:
@@ -176,6 +196,10 @@ async def download_file(request: DownloadRequest):
         if 'original_dir' in locals():
             os.chdir(original_dir)
 
+@app.get("/")
+async def root():
+    return {"message": "Welcome to GAMDL API. Visit /docs for API documentation."}
+
 @app.get("/test")
 async def test():
     """Test endpoint to verify setup"""