update downloading

app.py

@@ -11,8 +11,16 @@ from pydub import AudioSegment  # Import AudioSegment
 API_URL = "http://astarwiz.com:9998"
 rapid_key = os.environ.get("RAPID_API_KEY")
 
-def download_youtube_audio(metadata, output_dir: Optional[str] = None) -> str:
-    video_id = metadata['id']
+def fetch_youtube_id(youtube_url: str) -> str:
+    if 'v=' in youtube_url:
+        return youtube_url.split("v=")[1]
+    elif 'shorts' in youtube_url:
+        return youtube_url.split("/")[-1]
+    else:
+        raise Exception("Unsupported URL format")
+    
+def download_youtube_audio(youtube_url: str, output_dir: Optional[str] = None) -> str:
+    video_id = fetch_youtube_id(youtube_url)
     
     if output_dir is None:
         output_dir = tempfile.gettempdir()
@@ -29,7 +37,7 @@ def download_youtube_audio(metadata, output_dir: Optional[str] = None) -> str:
         'x-rapidapi-key': rapid_key  # Replace <key> with your actual API key
     }
     data = {
-        "url": metadata['url']
+        "url": youtube_url
     }
     
     response = requests.post(url, headers=headers, json=data)
@@ -89,16 +97,7 @@ def run_asr(audio_file, youtube_url):
 def embed_youtube(youtube_url):
     if youtube_url:
         try:
-            # video_id = YoutubeDL().extract_info(youtube_url, download=False)['id']\
-            print('short' in youtube_url)
-            print('v=' in youtube_url)
-            if 'v=' in youtube_url:
-                video_id = youtube_url.split("v=")[1]
-            elif 'shorts' in youtube_url:
-                video_id = youtube_url.split("/")[-1]
-            else:
-                raise Exception("Unsupported URL format")
-            print(video_id)
+            video_id = fetch_youtube_id(youtube_url)
             embed_html = f'<iframe width="560" height="315" src="https://www.youtube.com/embed/{video_id}" frameborder="0" allow="autoplay; encrypted-media" allowfullscreen></iframe>'
             return gr.update(value=embed_html, visible=True), "", None
         except Exception as e: