Update app.py

app.py

@@ -86,6 +86,9 @@ app.mount("/files", StaticFiles(directory=DOWNLOADS_DIR), name="files")
 class DownloadRequest(BaseModel):
     url: str
     
+
+# [Previous cookie handling code remains the same...]
+
 class FileInfo(BaseModel):
     filename: str
     download_url: str
@@ -149,32 +152,53 @@ async def download_file(request: DownloadRequest):
         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 file to download directory root if it's in a subdirectory
-            if os.path.dirname(file_path):
+            try:
+                # Convert to absolute path
+                abs_file_path = os.path.abspath(file_path)
+                filename = os.path.basename(file_path)
+                
+                # Create new path in download directory
                 new_path = os.path.join(download_subdir, filename)
-                shutil.move(file_path, new_path)
-            
-            # 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}")
+                abs_new_path = os.path.abspath(new_path)
+                
+                logger.info(f"Moving file from {abs_file_path} to {abs_new_path}")
+                
+                # Copy file instead of moving
+                shutil.copy2(abs_file_path, abs_new_path)
+                
+                # 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}")
+                
+            except Exception as e:
+                logger.error(f"Error processing file {file_path}: {str(e)}")
+                continue
+        
+        # Clean up original files after successful copy
+        for root, dirs, files in os.walk('.'):
+            for dir_name in dirs:
+                if dir_name != download_subdir:
+                    dir_path = os.path.join(root, dir_name)
+                    logger.info(f"Removing directory: {dir_path}")
+                    shutil.rmtree(dir_path, ignore_errors=True)
         
         # Move back to original directory
         os.chdir(original_dir)
         
+        if not downloaded_files:
+            raise Exception("Failed to process any files")
+        
         return DownloadResponse(
             success=True,
             message=f"Successfully downloaded {len(downloaded_files)} files",
@@ -196,7 +220,6 @@ async def download_file(request: DownloadRequest):
     finally:
         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."}