Update main.py

main.py

@@ -1,65 +1,58 @@
 import os
 import uuid
+import subprocess
 from fastapi import FastAPI, BackgroundTasks, HTTPException
 from fastapi.responses import FileResponse
 import yt_dlp
 
 app = FastAPI()
-
-# Cookies ෆයිල් එක තියෙන තැන
 COOKIES_FILE = "cookies.txt"
 
 def cleanup_file(path: str):
-    """ෆයිල් එක යැව්වට පස්සේ මකලා දානවා (Space එක පිරෙන්නේ නැති වෙන්න)"""
     if os.path.exists(path):
         os.remove(path)
 
 @app.get("/")
 def home():
-    return {"status": "Alive", "message": "Music Downloader API is Running! 🚀"}
+    # FFmpeg තියෙනවද කියලා චෙක් කරනවා
+    ffmpeg_check = subprocess.run(["ffmpeg", "-version"], capture_output=True, text=True)
+    return {
+        "status": "Alive",
+        "ffmpeg": "Installed" if ffmpeg_check.returncode == 0 else "Not Found"
+    }
 
 @app.get("/download")
 async def download_song(url: str, background_tasks: BackgroundTasks):
-    """
-    Koyeb එකෙන් URL එක එව්වම සින්දුව බාලා දෙන Function එක
-    """
-    try:
-        # තාවකාලික ෆයිල් නමක් හදාගන්නවා
-        file_id = str(uuid.uuid4())
-        filename_template = f"{file_id}.%(ext)s"
-        
-        ydl_opts = {
-            'format': 'bestaudio/best',
-            'outtmpl': filename_template,
-            'postprocessors': [{
-                'key': 'FFmpegExtractAudio',
-                'preferredcodec': 'mp3',
-                'preferredquality': '192',
-            }],
-            'quiet': True,
-            'no_warnings': True,
-            # Cookies ෆයිල් එක අනිවාර්යයෙන්ම ඕනේ
-            'cookiefile': COOKIES_FILE if os.path.exists(COOKIES_FILE) else None,
-            # YouTube එකට බොරු කියනවා අපි Browser එකක් කියලා
-            'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36',
-        }
+    file_id = str(uuid.uuid4())
+    output_filename = f"{file_id}.mp3"
+    
+    ydl_opts = {
+        'format': 'bestaudio/best',
+        'outtmpl': f"{file_id}.%(ext)s",
+        'postprocessors': [{
+            'key': 'FFmpegExtractAudio',
+            'preferredcodec': 'mp3',
+            'preferredquality': '192',
+        }],
+        'quiet': False, # Error එක බලාගන්න මේක False කළා
+        'cookiefile': COOKIES_FILE if os.path.exists(COOKIES_FILE) else None,
+        'user_agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
+    }
 
+    try:
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
             ydl.download([url])
 
-        # ඩවුන්ලෝඩ් වුන MP3 එකේ නම
-        mp3_file = f"{file_id}.mp3"
-
-        if not os.path.exists(mp3_file):
-            raise HTTPException(status_code=500, detail="Download failed internally.")
-
-        # ෆයිල් එක යවනවා (යැව්වට පස්සේ මකන්න කියලා Task එකක් දෙනවා)
-        return FileResponse(
-            mp3_file, 
-            media_type="audio/mpeg", 
-            filename="song.mp3",
-            background=background_tasks.add_task(cleanup_file, mp3_file)
-        )
+        if os.path.exists(output_filename):
+            return FileResponse(
+                output_filename, 
+                media_type="audio/mpeg", 
+                filename="song.mp3",
+                background=background_tasks.add_task(cleanup_file, output_filename)
+            )
+        else:
+            return {"error": "File not found after download."}
 
     except Exception as e:
+        # මෙතනින් තමයි ඇත්තම ලෙඩේ එළියට එවන්නේ
         return {"error": str(e)}