Update app.py

app.py

@@ -16,33 +16,41 @@ language_map = {
     "Japanese": "Helsinki-NLP/opus-mt-en-jap"
 }
 
-def generate_subtitles(video_file, language):
-    try:
-        srt_path = process_video(video_file, language)
-        return srt_path
-    except Exception as e:
-        return f"Error: {str(e)}"
-
 # Define Gradio Interface
-with gr.Blocks() as demo:
-    gr.Markdown("# AI-Powered Video Subtitling")
-    gr.Markdown("Upload a video and select a language to generate subtitles.")
+with gr.Blocks(theme=gr.themes.Monochrome()) as demo:  # Add dark theme
+    gr.Markdown("# 🎥 AI-Powered Video Subtitling")
+    gr.Markdown("Upload a video (MP4/MKV/AVI) and select a language to generate subtitles.")
 
     with gr.Row():
-        video_input = gr.Video(label="Upload Video File", format="mp4")  # Use gr.Video instead of gr.File
+        video_input = gr.File(
+            label="Upload Video File",
+            file_types=["mp4", "mkv", "avi"],  # Remove dots before extensions
+            interactive=True
+        )
         language_dropdown = gr.Dropdown(
             choices=list(language_map.keys()),
             label="Select Subtitle Language",
             value="English"
         )
 
-    generate_button = gr.Button("Generate Subtitles")
+    generate_button = gr.Button("Generate Subtitles 🚀")
     output_srt = gr.File(label="Download Subtitles")
+    progress = gr.Textbox(label="Progress", interactive=False)  # Add progress updates
+
+    def generate_subtitles(video_file, language):
+        try:
+            # Extract the file path from Gradio's File object
+            video_path = video_file.name  # Get the actual file path
+            print(f"Processing video: {video_path}")
+            srt_path = process_video(video_path, language)
+            return srt_path, "✅ Subtitles generated successfully!"
+        except Exception as e:
+            return None, f"❌ Error: {str(e)}"
 
     generate_button.click(
         generate_subtitles,
-        inputs=[video_input, language_dropdown],
-        outputs=output_srt
+        inputs=[video_file, language_dropdown],
+        outputs=[output_srt, progress]  # Include progress feedback
     )
 
 demo.launch()