Update app.py

app.py

@@ -56,7 +56,6 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=css) as demo:
         with gr.Column(scale=2):
             video_input = gr.File(
                 label="Upload Video File",
-                file_types=["mp4", "mkv", "avi"],
                 elem_classes=["file-preview"]
             )
         with gr.Column(scale=1):
@@ -65,6 +64,7 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=css) as demo:
                 label="Select Subtitle Language",
                 value="English"
             )
+
     generate_button = gr.Button("Generate Subtitles 🚀")
     progress_text = gr.Textbox(
         label="Progress",
@@ -75,9 +75,13 @@ with gr.Blocks(theme=gr.themes.Monochrome(), css=css) as demo:
 
     def generate_subtitles(video_file, language):
         try:
-            video_path = video_file.name  # Get the actual file path
-            progress_text.update(value="🔄 Processing video...")  # Update progress
-            srt_path = process_video(video_path, language)
+            # Validate file type
+            if not video_file.name.lower().endswith(('.mp4', '.mkv', '.avi')):
+                return None, "❌ Invalid file type. Please upload an MP4, MKV, or AVI file."
+
+            # Process video
+            progress_text.update(value="🔄 Processing video...")
+            srt_path = process_video(video_file.name, language)
             if srt_path:
                 return gr.File(srt_path), "✅ Subtitles generated successfully!"
             else: