Update main.py

main.py

@@ -1,39 +1,32 @@
 from fastapi import FastAPI, HTTPException, BackgroundTasks
 from fastapi.responses import FileResponse
+from pydantic import BaseModel
 import yt_dlp
-import os
-import glob
 import uuid
+import os
 
 app = FastAPI()
-
-# Create a temporary directory for downloads
 DOWNLOAD_DIR = "/app/downloads"
 os.makedirs(DOWNLOAD_DIR, exist_ok=True)
 
+# 1. Define the Request Body
+class VideoRequest(BaseModel):
+    url: str
+
 def cleanup_file(path: str):
-    """Deletes the file after it has been sent."""
     if os.path.exists(path):
         os.remove(path)
 
-@app.get("/download-mp3")
-async def download_mp3(url: str, background_tasks: BackgroundTasks):
-    """
-    Downloads YouTube audio, converts to MP3, and returns the file.
-    """
-    
-    # Generate a unique filename to prevent conflicts between users
+@app.post("/convert")
+async def convert_to_mp3(request: VideoRequest, background_tasks: BackgroundTasks):
     unique_id = str(uuid.uuid4())
     
+    # Configure yt-dlp for Audio
     ydl_opts = {
         'format': 'bestaudio/best',
         'outtmpl': f'{DOWNLOAD_DIR}/{unique_id}.%(ext)s',
         'quiet': True,
-        'no_warnings': True,
-        # COOKIES ARE STILL REQUIRED!
-        'cookiefile': 'cookies.txt', 
-        
-        # POST-PROCESSOR: Convert to MP3
+        'cookiefile': 'cookies.txt', # REQUIRED
         'postprocessors': [{
             'key': 'FFmpegExtractAudio',
             'preferredcodec': 'mp3',
@@ -42,33 +35,26 @@ async def download_mp3(url: str, background_tasks: BackgroundTasks):
     }
 
     try:
-        # Run download
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            ydl.download([url])
+            # Get Info First (for title)
+            info = ydl.extract_info(request.url, download=False)
+            clean_title = "".join(c for c in info['title'] if c.isalnum() or c in " -_").strip()
+            filename = f"{clean_title}.mp3"
             
-            # The file will be named {unique_id}.mp3
-            file_path = f"{DOWNLOAD_DIR}/{unique_id}.mp3"
-            
-            if not os.path.exists(file_path):
-                raise HTTPException(status_code=500, detail="Conversion failed.")
+            # Download & Convert
+            ydl.download([request.url])
             
-            # Get the actual title for the filename the user sees
-            info = ydl.extract_info(url, download=False)
-            display_name = f"{info.get('title', 'audio')}.mp3"
+            file_path = f"{DOWNLOAD_DIR}/{unique_id}.mp3"
             
-            # Clean filename (remove special chars that break headers)
-            display_name = "".join(c for c in display_name if c.isalnum() or c in (' ', '.', '-', '_')).strip()
-
+            # Return the file as a stream
             return FileResponse(
                 path=file_path, 
-                filename=display_name,
+                filename=filename,
                 media_type='audio/mpeg',
-                headers={'Content-Disposition': f'attachment; filename="{display_name}"'},
+                # This header tells Android "Here is the filename"
+                headers={'Content-Disposition': f'attachment; filename="{filename}"'},
                 background=background_tasks.add_task(cleanup_file, file_path)
             )
 
     except Exception as e:
-        # cleanup if something failed
-        if os.path.exists(f"{DOWNLOAD_DIR}/{unique_id}.mp3"):
-            os.remove(f"{DOWNLOAD_DIR}/{unique_id}.mp3")
-        raise HTTPException(status_code=400, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e))