Update app.py

app.py

@@ -67,9 +67,13 @@ 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):
@@ -99,23 +103,23 @@ class DownloadResponse(BaseModel):
 @app.post("/download", response_model=DownloadResponse)
 async def download_file(request: DownloadRequest):
     try:
-        # Create a unique subdirectory for this download
+        # Create a unique subdirectory for this download in the temporary directory
         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)
+        temp_subdir = os.path.join(TEMP_DIR, timestamp)
+        os.makedirs(temp_subdir, exist_ok=True)
         
         # Log the current working directory and download directory
         logger.info(f"Current working directory: {os.getcwd()}")
-        logger.info(f"Download directory: {download_subdir}")
+        logger.info(f"Temporary download directory: {temp_subdir}")
         
         # Create cookies file from environment variable
-        cookie_path = os.path.join(download_subdir, "cookies.txt")
+        cookie_path = os.path.join(temp_subdir, "cookies.txt")
         logger.info(f"Creating cookies file at: {cookie_path}")
         env_to_cookies_from_env(cookie_path)
         
-        # Change to download directory
+        # Change to temporary download directory
         original_dir = os.getcwd()
-        os.chdir(download_subdir)
+        os.chdir(temp_subdir)
         
         # Log the command being executed
         cmd = ["gamdl", request.url]
@@ -137,16 +141,27 @@ 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 download directory: {files}")
+        logger.info(f"Files in temporary download directory: {files}")
         
         if not files:
-            raise Exception("No files found in download directory after download attempt")
+            raise Exception("No files found in temporary download directory after download attempt")
             
         downloaded_file = files[0]
         logger.info(f"Downloaded file: {downloaded_file}")
 
         # Get the file size
         file_size = os.path.getsize(downloaded_file)
+
+        # Check for encryption or corruption issues
+        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")
@@ -155,7 +170,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,