Update utils.py to fix openai-whisper compatibility and improve error handling

utils.py

@@ -1,19 +1,32 @@
-import openai_whisper as whisper
+import openai_whisper as whisper  # Use the correct import for openai-whisper
 from transformers import MarianMTModel, MarianTokenizer, AutoTokenizer, AutoModelForSeq2SeqLM
 import os
 
 # Load Whisper model
-model = whisper.load_model("base")
+try:
+    print("Loading Whisper model...")
+    model = whisper.load_model("base")
+    print("Whisper model loaded successfully!")
+except Exception as e:
+    raise ImportError(f"Failed to load Whisper model: {e}")
 
 def process_video(video_file, language):
+    """
+    Process the uploaded video and generate subtitles in the specified language.
+    """
     # Save uploaded video locally
     video_path = "/tmp/video.mp4"
-    with open(video_path, "wb") as f:
-        f.write(video_file.read())
+    try:
+        with open(video_path, "wb") as f:
+            f.write(video_file.read())
+        print(f"Video saved to {video_path}")
+    except Exception as e:
+        return f"Error saving video file: {str(e)}"
 
     try:
         print("Transcribing video to English...")
         result = model.transcribe(video_path, language="en")
+        print("Transcription completed!")
 
         segments = []
         if language == "English":
@@ -42,7 +55,10 @@ def process_video(video_file, language):
                     "Arabic": "Helsinki-NLP/opus-mt-en-ar",
                     "Japanese": "Helsinki-NLP/opus-mt-en-jap"
                 }
-                model_name = model_map[language]
+                model_name = model_map.get(language)
+                if not model_name:
+                    return f"Unsupported language: {language}"
+
                 tokenizer = MarianTokenizer.from_pretrained(model_name)
                 translation_model = MarianMTModel.from_pretrained(model_name)
                 print(f"Translating to {language}...")
@@ -60,7 +76,8 @@ def process_video(video_file, language):
                 end = f"{segment['end']:.3f}".replace(".", ",")
                 text = segment["text"].strip()
                 f.write(f"{i}\n00:00:{start} --> 00:00:{end}\n{text}\n\n")
+        print(f"SRT file created at {srt_path}")
         return srt_path
 
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"Error processing video: {str(e)}"