Update app.py

app.py

@@ -16,41 +16,79 @@ language_map = {
     "Japanese": "Helsinki-NLP/opus-mt-en-jap"
 }
 
+# Custom CSS for dark mode and animations
+css = """
+body {
+    transition: background-color 0.5s ease;
+}
+.dark-theme {
+    background-color: #1a1a1a !important;
+    color: #e0e0e0;
+}
+.gradio-container {
+    max-width: 1200px;
+    margin: 0 auto;
+    padding: 20px;
+    border-radius: 10px;
+    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+}
+.file-preview {
+    border: 2px dashed #6c757d;
+    padding: 20px;
+    border-radius: 10px;
+}
+.progress-text {
+    font-size: 16px;
+    color: #28a745;
+    animation: blink 1s infinite;
+}
+@keyframes blink {
+    50% { opacity: 0.5; }
+}
+"""
+
 # Define Gradio Interface
-with gr.Blocks(theme=gr.themes.Monochrome()) as demo:  # Add dark theme
+with gr.Blocks(theme=gr.themes.Monochrome(), css=css) as demo:
     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.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"
-        )
-
+        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):
+            language_dropdown = gr.Dropdown(
+                choices=list(language_map.keys()),
+                label="Select Subtitle Language",
+                value="English"
+            )
     generate_button = gr.Button("Generate Subtitles 🚀")
+    progress_text = gr.Textbox(
+        label="Progress",
+        interactive=False,
+        elem_classes=["progress-text"]
+    )
     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}")
+            progress_text.update(value="🔄 Processing video...")  # Update progress
             srt_path = process_video(video_path, language)
-            return srt_path, "✅ Subtitles generated successfully!"
+            if srt_path:
+                return gr.File(srt_path), "✅ Subtitles generated successfully!"
+            else:
+                return None, "❌ Error during processing. Check logs."
         except Exception as e:
             return None, f"❌ Error: {str(e)}"
 
     generate_button.click(
         generate_subtitles,
-        inputs=[video_file, language_dropdown],
-        outputs=[output_srt, progress]  # Include progress feedback
+        inputs=[video_input, language_dropdown],
+        outputs=[output_srt, progress_text]
     )
 
 demo.launch()