Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import os,secrets,requests
 from basic_pitch.inference import predict_and_save
 from urllib.parse import quote
-import deta
+import traceback
 from deta import Deta  # Import Deta
 
 # Initialize
@@ -36,40 +36,42 @@ def addrecord(orig,pitmp3,mid,gname,ghint):
 
 
 def downloadaud(video_id,gname,ghint):
-    filename=secrets.token_hex()
-    # Step 1: Search for videos with the given query
-
-    if not video_id:
-        return None
-
-    # Step 3: Get the formats for the video and find the URL for the best audio-only format
-    formats_url = f"https://draw-8fj.begin.app/api/info/{video_id}"
-    formats = requests.get(formats_url)
-    if formats.ok!=True:
-        formats_url = f"https://draw-8fj-staging.begin.app/api/info/{video_id}"
-        formats = requests.get(formats_url) 
-    formats_response=formats.json()
-    #print(formats_response["formats"])
-    best_audio_format = None
-    for fmt in formats_response.get("formats", []):
-        if fmt.get("hasVideo") is False and fmt.get("hasAudio") is True and fmt.get("container") == "mp4":
-            if not best_audio_format or fmt.get("audioBitrate") > best_audio_format.get("audioBitrate"):
-                best_audio_format = fmt
-    print(best_audio_format)
-    # If no suitable audio format was found, return None
-    if not best_audio_format:
-        return None
-
-    aurl=best_audio_format["url"]
-    print(aurl)
-    command = f'./pget -o {filename}.mp4 -p 4 "{aurl}" '
-    os.system(command)
-    os.system(f'ffmpeg -i {filename}.mp4 -q:a 0 -map a {filename}.mp3')
-    os.system(f'mv {filename}.mp3 /tmp/{filename}.mp3')
-    o,m,i= predict(f'/tmp/{filename}.mp3')
-    res=addrecord(o,m,i,gname,ghint)
-    return o,m,i,res
-
+    try:
+        filename=secrets.token_hex()
+        # Step 1: Search for videos with the given query
+    
+        if not video_id:
+            return None
+    
+        # Step 3: Get the formats for the video and find the URL for the best audio-only format
+        formats_url = f"https://draw-8fj.begin.app/api/info/{video_id}"
+        formats = requests.get(formats_url)
+        if formats.ok!=True:
+            formats_url = f"https://draw-8fj-staging.begin.app/api/info/{video_id}"
+            formats = requests.get(formats_url) 
+        formats_response=formats.json()
+        #print(formats_response["formats"])
+        best_audio_format = None
+        for fmt in formats_response.get("formats", []):
+            if fmt.get("hasVideo") is False and fmt.get("hasAudio") is True and fmt.get("container") == "mp4":
+                if not best_audio_format or fmt.get("audioBitrate") > best_audio_format.get("audioBitrate"):
+                    best_audio_format = fmt
+        print(best_audio_format)
+        # If no suitable audio format was found, return None
+        if not best_audio_format:
+            return None
+    
+        aurl=best_audio_format["url"]
+        print(aurl)
+        command = f'./pget -o {filename}.mp4 -p 4 "{aurl}" '
+        os.system(command)
+        os.system(f'ffmpeg -i {filename}.mp4 -q:a 0 -map a {filename}.mp3')
+        os.system(f'mv {filename}.mp3 /tmp/{filename}.mp3')
+        o,m,i= predict(f'/tmp/{filename}.mp3')
+        res=addrecord(o,m,i,gname,ghint)
+        return o,m,i,res
+    except Exception as e:
+        print(traceback(e))
 
 def predict(ina):
     fn=ina