Update app.py

app.py

@@ -83,13 +83,19 @@ def gen_draw(user_query:str)->tuple:
     return "Invalid prompt"
 
 
-
 def vid_tube(user_query:str) -> tuple:
   py_tube_list_of_videos = Search(user_query)
   first_video = py_tube_list_of_videos.results[0]
-  youtube_object = first_video.streams.get_highest_resolution() #if facing problems use get_lowest_resolution
-  file_path = youtube_object.download('/tmp/')
-  return (file_path,)
+  yt_flag = False
+  for vid in py_tube_list_of_videos.results:
+    print(vid.vid_info.keys())
+    if vid.vid_info.get('streamingData'):
+      print(vid.vid_info.keys(),'-')
+      yt_flag = True
+      file_path = vid.streams.get_highest_resolution().download('/content/')
+      break
+  
+  return (file_path,) if yt_flag else "The system cannot fulfill your request currently please try later"
 
 
 def search_internet(user_query:str,*,key_number:int) -> str: