Update app.py

app.py

@@ -20,27 +20,30 @@ languages = {
     "Russian (ru)": "ru"
 }
 
-def generate_subtitles(video_path, language_name):
+def generate_subtitles(video_file, language_name):
     try:
         # Extract the target language code from the selected language name
         target_language = languages[language_name]
         
-        # Log the file path for debugging
-        print(f"Received video file: {video_path}")
+        # Check if video_file is a file object or a file path string
+        if isinstance(video_file, str):
+            video_path = video_file  # It's a file path
+        else:
+            video_path = video_file.name  # It's a file object
         
-        # Extract audio from video
+        print(f"Processing video from path: {video_path}")
+
+        # Extract audio from video using moviepy
         video = mp.VideoFileClip(video_path)
         audio = video.audio
         audio.write_audiofile("temp_audio.wav", codec='pcm_s16le')
 
-        # Log transcription start
         print("Starting speech-to-text transcription")
-        
+
         # Convert speech to text (ASR using Whisper)
         with open("temp_audio.wav", "rb") as audio_file:
             transcription = asr(audio_file)["text"]
 
-        # Log translation start
         print("Starting translation")
 
         # Translate transcription to the target language using M2M100
@@ -62,10 +65,9 @@ def generate_subtitles(video_path, language_name):
 # Define Gradio interface
 def subtitle_video(video_file, language_name):
     try:
-        # video_file is already the file path, no need for .name
+        # Handle both file-like objects and file paths
         return generate_subtitles(video_file, language_name)
     except Exception as e:
-        # Catch and display any high-level errors
         print(f"Error in processing video: {e}")
         return f"Error in processing video: {e}"