Fixed file validation to use original filename

app.py

@@ -72,16 +72,16 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=css) as demo:
 
     def generate_subtitles(video_file, language):
         try:
-            # Validate file type
-            if not video_file.name.lower().endswith(('.mp4', '.mkv', '.avi')):
+            # Validate original filename (not the temporary path)
+            if not video_file.orig_name.lower().endswith(('.mp4', '.mkv', '.avi')):
                 return None, "❌ Invalid file type. Please upload an MP4, MKV, or AVI file."
 
             # Initialize progress tracking
             progress = gr.Progress(track_tqdm=True)
 
-            # Process video
+            # Process video using the temporary file path
             progress(0, desc="Initializing...")
-            srt_path = process_video(video_file.name, language, progress=progress)
+            srt_path = process_video(video_file.name, language, progress=progress)  # Use video_file.name for processing
             if srt_path:
                 progress(1, desc="✅ Subtitles generated successfully!")
                 return gr.File(srt_path), "✅ Subtitles ready for download!"