Update app.py

app.py

@@ -11,7 +11,7 @@ import tempfile
 from pathlib import Path
 from dotenv import load_dotenv
 from urllib.parse import quote
-import time
+import time # Import for sleep to ensure file is fully written
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -68,15 +68,9 @@ app = FastAPI(
 
 
 
-
-
-
-
 # Create downloads directory if it doesn't exist
 DOWNLOADS_DIR = "downloads"
-TEMP_DIR = "temp"
 os.makedirs(DOWNLOADS_DIR, exist_ok=True)
-os.makedirs(TEMP_DIR, exist_ok=True)
 
 # Mount the downloads directory with cache control headers
 class CacheControlStaticFiles(StaticFiles):
@@ -106,23 +100,23 @@ class DownloadResponse(BaseModel):
 @app.post("/download", response_model=DownloadResponse)
 async def download_file(request: DownloadRequest):
     try:
-        # Create a unique subdirectory for this download in the temporary directory
+        # Create a unique subdirectory for this download
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-        temp_subdir = os.path.join(TEMP_DIR, timestamp)
-        os.makedirs(temp_subdir, exist_ok=True)
+        download_subdir = os.path.join(DOWNLOADS_DIR, timestamp)
+        os.makedirs(download_subdir, exist_ok=True)
         
         # Log the current working directory and download directory
         logger.info(f"Current working directory: {os.getcwd()}")
-        logger.info(f"Temporary download directory: {temp_subdir}")
+        logger.info(f"Download directory: {download_subdir}")
         
         # Create cookies file from environment variable
-        cookie_path = os.path.join(temp_subdir, "cookies.txt")
+        cookie_path = os.path.join(download_subdir, "cookies.txt")
         logger.info(f"Creating cookies file at: {cookie_path}")
         env_to_cookies_from_env(cookie_path)
         
-        # Change to temporary download directory
+        # Change to download directory
         original_dir = os.getcwd()
-        os.chdir(temp_subdir)
+        os.chdir(download_subdir)
         
         # Log the command being executed
         cmd = ["gamdl", request.url]
@@ -144,30 +138,18 @@ async def download_file(request: DownloadRequest):
         
         # Get the downloaded filename
         files = [f for f in os.listdir() if f != "cookies.txt"]
-        logger.info(f"Files in temporary download directory: {files}")
+        logger.info(f"Files in download directory: {files}")
         
         if not files:
-            raise Exception("No files found in temporary download directory after download attempt")
+            raise Exception("No files found in download directory after download attempt")
             
         downloaded_file = files[0]
         logger.info(f"Downloaded file: {downloaded_file}")
-
-        # Ensure the file is fully written
-        time.sleep(1)  # Sleep for 1 second
+        # Ensure the file is fully written 
+        time.sleep(2) # Sleep for 2 second
 
         # Get the file size
         file_size = os.path.getsize(downloaded_file)
-
-        # Check for potential issues in file handling
-        if process.stderr:
-            raise Exception(f"Error in downloaded file: {process.stderr}")
-
-        # Move the completed download to the final directory
-        final_download_subdir = os.path.join(DOWNLOADS_DIR, timestamp)
-        os.makedirs(final_download_subdir, exist_ok=True)
-        shutil.move(downloaded_file, final_download_subdir)
-        downloaded_file_path = os.path.join(final_download_subdir, downloaded_file)
-        logger.info(f"Moved downloaded file to: {downloaded_file_path}")
         
         # Generate the download URL and URL encode the filename
         space_url = os.getenv("SPACE_URL", "https://tecuts-vob.hf.space")
@@ -176,7 +158,7 @@ async def download_file(request: DownloadRequest):
         logger.info(f"Generated download URL: {download_url}")
         
         # Move back to original directory
-        os.chdir(original_dir)
+       
         
         return DownloadResponse(
             success=True,