Update app.py

app.py

@@ -27,6 +27,7 @@ from io import BytesIO
 from pathlib import Path
 from fastapi.staticfiles import StaticFiles
 from collections import defaultdict
+import logging
 
 tmp_dir = tempfile.gettempdir()
 BASE_URL = "https://chrunos-multi.hf.space"
@@ -311,6 +312,68 @@ async def download_audio(request: Request):
     encoded_filename = urllib.parse.quote(downloaded_file.name)
     download_url = f"{BASE_URL}/file/{encoded_filename}"
     return {"url": download_url}
+
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+
+@app.post("/search")
+async def search_and_download_song(request: Request):
+    data = await request.json()
+    song_name = data.get('songname')
+    artist_name = data.get('artist')
+    if artist_name:
+        search_query = f"ytsearch:{song_name} {artist_name}"
+    else:
+        search_query = f"ytsearch:{song_name}"
+
+    logging.info(f"Search query: {search_query}")
+    cookiefile = "firefox-cookies.txt"
+    env_to_cookies_from_env("firefox-cookies.txt")
+    
+    timestamp = datetime.now().strftime('%Y%m%d%H%M%S')
+    output_template = str(Path(global_download_dir) / f'%(title)s_{timestamp}.%(ext)s')
+
+    ydl_opts = {
+        'format': 'bestaudio/best',
+        'outtmpl': output_template,
+        'quiet': True,
+        'no_warnings': True,
+        'noprogress': True,
+        'postprocessors': [{
+            'key': 'FFmpegExtractAudio',
+            'preferredcodec': 'mp3',
+            'preferredquality': '192'
+        }],
+        'cookiefile': cookiefile
+    }
+
+    try:
+        logging.info("Starting yt-dlp search and download...")
+        await run_in_threadpool(lambda: yt_dlp.YoutubeDL(ydl_opts).download([search_query]))
+        logging.info("yt-dlp search and download completed")
+    except yt_dlp.utils.DownloadError as e:
+        error_message = str(e)
+        logging.error(f"yt-dlp error: {error_message}")
+        return JSONResponse(content={"error": error_message}, status_code=500)
+    except Exception as e:
+        error_message = str(e)
+        logging.error(f"General error: {error_message}")
+        return JSONResponse(content={"error": error_message}, status_code=500)
+    
+    downloaded_files = list(Path(global_download_dir).glob(f"*_{timestamp}.mp3"))
+    if not downloaded_files:
+        logging.error("Download failed: No MP3 files found")
+        return JSONResponse(content={"error": "Download failed"}, status_code=500)
+
+    downloaded_file = downloaded_files[0]
+    encoded_filename = urllib.parse.quote(downloaded_file.name)
+    download_url = f"{BASE_URL}/file/{encoded_filename}"
+    logging.info(f"Download URL: {download_url}")
+
+    # Log just before returning the response
+    logging.info("Preparing to send response back to the client")
+    return JSONResponse(content={"url": download_url}, status_code=200)
+
 # Mount the static files directory
 app.mount("/file", StaticFiles(directory=global_download_dir), name="downloads")