Update app.py

app.py

@@ -11,6 +11,16 @@ from pydub import AudioSegment  # Import AudioSegment
 API_URL = "http://astarwiz.com:9998"
 rapid_key = os.environ.get("RAPID_API_KEY")
 
+def fetch_youtube_id(youtube_url: str) -> str:
+    if 'v=' in youtube_url:
+        return youtube_url.split("v=")[1].split("&")[0]
+    elif 'youtu.be/' in youtube_url:
+        return youtube_url.split("youtu.be/")[1]
+    elif 'shorts' in youtube_url:
+        return youtube_url.split("/")[-1]
+    else:
+        raise Exception("Unsupported URL format")
+"""
 def fetch_youtube_id(youtube_url: str) -> str:
     if 'v=' in youtube_url:
         return youtube_url.split("v=")[1]
@@ -18,7 +28,7 @@ def fetch_youtube_id(youtube_url: str) -> str:
         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)