Update app.py

app.py

@@ -87,6 +87,7 @@ class DownloadRequest(BaseModel):
     url: str
     
 
+
 # [Previous cookie handling code remains the same...]
 
 class FileInfo(BaseModel):
@@ -153,18 +154,16 @@ async def download_file(request: DownloadRequest):
         
         for file_path in all_files:
             try:
-                # Convert to absolute path
-                abs_file_path = os.path.abspath(file_path)
+                # Get just the filename from the path
                 filename = os.path.basename(file_path)
                 
-                # Create new path in download directory
-                new_path = os.path.join(download_subdir, filename)
-                abs_new_path = os.path.abspath(new_path)
+                # Create new path in current directory
+                new_path = os.path.join(".", filename)
                 
-                logger.info(f"Moving file from {abs_file_path} to {abs_new_path}")
+                logger.info(f"Moving file from {file_path} to {new_path}")
                 
-                # Copy file instead of moving
-                shutil.copy2(abs_file_path, abs_new_path)
+                # Copy file to current directory
+                shutil.copy2(file_path, new_path)
                 
                 # Get file extension
                 file_type = os.path.splitext(filename)[1].lstrip('.')
@@ -185,10 +184,10 @@ async def download_file(request: DownloadRequest):
                 logger.error(f"Error processing file {file_path}: {str(e)}")
                 continue
         
-        # Clean up original files after successful copy
+        # Clean up original files and directories after successful copy
         for root, dirs, files in os.walk('.'):
             for dir_name in dirs:
-                if dir_name != download_subdir:
+                if dir_name == "Apple Music":  # Only remove the music directory
                     dir_path = os.path.join(root, dir_name)
                     logger.info(f"Removing directory: {dir_path}")
                     shutil.rmtree(dir_path, ignore_errors=True)
@@ -220,6 +219,7 @@ 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."}