Create main.py

main.py

@@ -0,0 +1,65 @@
+import os
+import uuid
+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! 🚀"}
+
+@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',
+        }
+
+        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)
+        )
+
+    except Exception as e:
+        return {"error": str(e)}