Update app.py

app.py

@@ -179,12 +179,20 @@ def write_ppt(transcription, output_file, tokenizer=None, translation_model=None
 
 # Transcribing video and generating output
 def transcribe_video(video_file, video_url, language, target_language, output_format):
-    video_file_path = video_file.name
+    if video_url:
+        video_file_path = download_youtube_video(video_url)
+    elif video_file is not None:  # Ensure the video_file is not None
+        video_file_path = video_file.name
+    else:
+        raise ValueError("No video file provided. Please upload a video file or provide a YouTube link.")
+
     result = model.transcribe(video_file_path, language=language)
     video_name = os.path.splitext(video_file_path)[0]
-
     if target_language != "en":
-        tokenizer, translation_model = load_translation_model(target_language)
+        try:
+            tokenizer, translation_model = load_translation_model(target_language)
+        except Exception as e:
+            raise RuntimeError(f"Error loading translation model: {e}")
     else:
         tokenizer, translation_model = None, None
 
@@ -195,8 +203,11 @@ def transcribe_video(video_file, video_url, language, target_language, output_fo
         return srt_file
     elif output_format == "Video with Hardsub":
         output_video = f"{video_name}_with_subtitles.mp4"
-        embed_hardsub_in_video(video_file_path, srt_file, output_video)
-        return output_video
+        try:
+            embed_hardsub_in_video(video_file_path, srt_file, output_video)
+            return output_video
+        except Exception as e:
+            raise RuntimeError(f"Error embedding subtitles in video: {e}")
     elif output_format == "Word":
         word_file = f"{video_name}.docx"
         write_word(result, word_file, tokenizer, translation_model, target_language)
@@ -210,6 +221,7 @@ def transcribe_video(video_file, video_url, language, target_language, output_fo
         write_ppt(result, ppt_file, tokenizer, translation_model)
         return ppt_file
 
+
 # Gradio interface with YouTube URL
 iface = gr.Interface(
     fn=transcribe_video,