Update app.py

app.py

@@ -11,6 +11,7 @@ import ast
 from IPython.display import Audio, display
 import requests
 nltk.download('punkt_tab')
+from io import BytesIO
 
 model = whisper.load_model("base")
 
@@ -27,10 +28,12 @@ def extract_yt_audio(it, video_url, video_file):
       a = yt.streams.filter(only_audio=True).first()
       audio_file = a.download()
       sample = AudioSegment.from_file(audio_file, format="mp4")
-    elif it == 'URL':
+    elif it == 'URL' and ("https://www" in video_url or "https://" in video_url or "www." in video_url):
       response = requests.get(video_url)
       video_data = BytesIO(response.content)
-      sample = AudioSegment.from_file(video_data, format="mp4")                                
+      sample = AudioSegment.from_file(video_data, format="mp4")
+    elif it == 'URL'
+      sample = AudioSegment.from_file(video_url)  
     else:
       sample = AudioSegment.from_file(video_file)
     audio_path = 'audio.wav'