Update app.py

app.py

@@ -1,9 +1,11 @@
 # app.py
 from fastapi import FastAPI, HTTPException
+from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel
 import subprocess
 import os
-import tempfile
+import shutil
+from datetime import datetime
 
 app = FastAPI(
     title="GAMDL API",
@@ -11,6 +13,13 @@ app = FastAPI(
     version="1.0.0"
 )
 
+# Create downloads directory if it doesn't exist
+DOWNLOADS_DIR = "downloads"
+os.makedirs(DOWNLOADS_DIR, exist_ok=True)
+
+# Mount the downloads directory
+app.mount("/files", StaticFiles(directory=DOWNLOADS_DIR), name="files")
+
 class DownloadRequest(BaseModel):
     url: str
     
@@ -18,39 +27,49 @@ class DownloadResponse(BaseModel):
     success: bool
     message: str
     filename: str = None
+    download_url: str = None
 
 @app.post("/download", response_model=DownloadResponse)
 async def download_file(request: DownloadRequest):
     try:
-        # Create a temporary directory for downloads
-        with tempfile.TemporaryDirectory() as temp_dir:
-            # Change to temp directory
-            original_dir = os.getcwd()
-            os.chdir(temp_dir)
-            
-            # Run gamdl command
-            process = subprocess.run(
-                ["gamdl", request.url],
-                capture_output=True,
-                text=True,
-                check=True
-            )
-            
-            # Get the downloaded filename (last file in temp directory)
-            files = os.listdir()
-            if not files:
-                raise Exception("No file was downloaded")
-                
-            downloaded_file = files[0]
-            
-            # Move back to original directory
-            os.chdir(original_dir)
+        # Create a unique subdirectory for this download
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        download_subdir = os.path.join(DOWNLOADS_DIR, timestamp)
+        os.makedirs(download_subdir, exist_ok=True)
+        
+        # Change to download directory
+        original_dir = os.getcwd()
+        os.chdir(download_subdir)
+        
+        # Run gamdl command
+        process = subprocess.run(
+            ["gamdl", request.url],
+            capture_output=True,
+            text=True,
+            check=True
+        )
+        
+        # Get the downloaded filename
+        files = os.listdir()
+        if not files:
+            raise Exception("No file was downloaded")
             
-            return DownloadResponse(
-                success=True,
-                message="File downloaded successfully",
-                filename=downloaded_file
-            )
+        downloaded_file = files[0]
+        
+        # Generate the download URL
+        # Using the Space's domain - you'll need to replace this with your actual Space URL
+        space_url = os.getenv("SPACE_URL", "http://localhost:7860")
+        download_url = f"{space_url}/files/{timestamp}/{downloaded_file}"
+        
+        # Move back to original directory
+        os.chdir(original_dir)
+        
+        return DownloadResponse(
+            success=True,
+            message="File downloaded successfully",
+            filename=downloaded_file,
+            download_url=download_url
+        )
             
     except subprocess.CalledProcessError as e:
         raise HTTPException(
@@ -66,4 +85,20 @@ async def download_file(request: DownloadRequest):
 # Add basic homepage with API documentation link
 @app.get("/")
 async def root():
-    return {"message": " Visit /docs for API documentation."}
+    return {"message": "Welcome to GAMDL API. Visit /docs for API documentation."}
+
+# Optional: Add cleanup endpoint to manage disk space
+@app.post("/cleanup")
+async def cleanup_old_files():
+    """Remove files older than 24 hours"""
+    try:
+        for item in os.listdir(DOWNLOADS_DIR):
+            item_path = os.path.join(DOWNLOADS_DIR, item)
+            if os.path.isdir(item_path):
+                shutil.rmtree(item_path)
+        return {"message": "Cleanup completed successfully"}
+    except Exception as e:
+        raise HTTPException(
+            status_code=500,
+            detail=f"Cleanup failed: {str(e)}"
+        )