Update app.py

app.py

@@ -1,7 +1,5 @@
-# app.py
 import os
 import tempfile
-import uuid
 from pathlib import Path
 import gradio as gr
 import ffmpeg
@@ -33,14 +31,14 @@ def segments_to_srt(segments: list) -> str:
 
 
 # -------- Config --------
-MODEL_NAME = "Systran/faster-whisper-small"  # good for HF CPU
+MODEL_NAME = "Systran/faster-whisper-small"
 DEVICE = "cpu"
 OUTPUT_DIR = Path("outputs/subtitles")
 OUTPUT_DIR.mkdir(parents=True, exist_ok=True)
 
 print(f"Loading model {MODEL_NAME} on {DEVICE} ...")
 model = WhisperModel(MODEL_NAME, device=DEVICE)
-print("Model loaded.")
+print("✅ Model loaded.")
 
 
 # -------- Core functions --------
@@ -61,28 +59,21 @@ def extract_audio(input_path: str, out_path: str):
 
 
 def transcribe_file_to_srt(file_obj, language: str = "en"):
-    """Transcribe uploaded file to SRT; compatible with HF Spaces"""
+    """Transcribe uploaded file to SRT"""
     tmp_dir = Path(tempfile.mkdtemp(prefix="subgen_"))
-
-    # Handle Hugging Face NamedString / Path
     input_path = Path(file_obj.name)
+
     if not input_path.exists():
         input_path = tmp_dir / Path(file_obj.name).name
-        if hasattr(file_obj, "read_bytes"):
-            with open(input_path, "wb") as f:
-                f.write(file_obj.read_bytes())
-        else:
-            with open(file_obj.name, "rb") as src, open(input_path, "wb") as dst:
-                dst.write(src.read())
+        with open(input_path, "wb") as f:
+            f.write(file_obj.read())
 
-    # Extract audio and transcribe
     audio_path = tmp_dir / "audio.wav"
     extract_audio(str(input_path), str(audio_path))
     segments, _ = model.transcribe(str(audio_path), language=language)
     segs = [{"start": s.start, "end": s.end, "text": s.text} for s in segments]
     srt_text = segments_to_srt(segs)
 
-    # Save .srt file
     output_path = OUTPUT_DIR / f"{Path(file_obj.name).stem}.srt"
     with open(output_path, "w", encoding="utf-8") as f:
         f.write(srt_text)
@@ -97,44 +88,72 @@ with gr.Blocks(title="AI Subtitle Generator") as demo:
         return "dark" if current == "light" else "light"
 
     def apply_theme(theme_mode):
+        """Apply CSS for dark/light theme dynamically"""
         if theme_mode == "dark":
-            bg = "linear-gradient(135deg, #0f2027, #203a43, #2c5364)"
-            color = "#ffffff"
+            gradient = "linear-gradient(135deg, #0f2027, #203a43, #2c5364)"
+            text_color = "#ffffff"
+            btn_color = "#00adb5"
         else:
-            bg = "linear-gradient(135deg, #fdfbfb, #ebedee)"
-            color = "#000000"
-        return gr.update(
-            value=f"<style>body {{ background: {bg}; color: {color}; }}</style>"
-        )
-
+            gradient = "linear-gradient(135deg, #fdfbfb, #ebedee)"
+            text_color = "#000000"
+            btn_color = "#1e90ff"
+
+        style = f"""
+        <style>
+            body {{
+                background: {gradient};
+                color: {text_color};
+                transition: background 0.5s ease, color 0.5s ease;
+            }}
+            .gr-button {{
+                background-color: {btn_color} !important;
+                color: white !important;
+                border-radius: 10px !important;
+                transition: background 0.3s ease;
+            }}
+        </style>
+        """
+        return gr.update(value=style)
+
+    # Header
     gr.HTML("<h1 style='text-align:center;'>🎬 AI Subtitle Generator</h1>")
-    gr.HTML(
-        "<p style='text-align:center;'>Upload a video or audio file to generate English <b>.srt</b> subtitles.</p>"
-    )
+    gr.HTML("<p style='text-align:center;'>Upload a video or audio file to generate English <b>.srt</b> subtitles.</p>")
 
     style_box = gr.HTML("")
     theme_btn = gr.Button("🌙 Toggle Light/Dark Mode")
 
+    # Input / Output
     with gr.Row():
         input_file = gr.File(label="Upload video/audio file")
         output_file = gr.File(label="Download .srt file")
 
     status_box = gr.Textbox(label="Status", interactive=False)
 
+    # Buttons
+    with gr.Row():
+        generate_btn = gr.Button("🚀 Generate Subtitles")
+        clear_btn = gr.Button("🧹 Clear Chat")
+
+    # --- Logic ---
     def on_click(file):
+        if not file:
+            return None, "⚠️ Please upload a file first!"
         srt_path, msg = transcribe_file_to_srt(file)
         return srt_path, msg
 
-    theme_btn.click(
-        toggle_theme, inputs=[theme_state], outputs=[theme_state]
-    ).then(apply_theme, inputs=[theme_state], outputs=[style_box])
+    def clear_fields():
+        """Reset all UI components"""
+        return None, None, ""
+
+    # Theme button actions
+    theme_btn.click(toggle_theme, inputs=[theme_state], outputs=[theme_state]) \
+              .then(apply_theme, inputs=[theme_state], outputs=[style_box])
 
-    generate_btn = gr.Button("Generate Subtitles")
+    # Main buttons
     generate_btn.click(on_click, inputs=[input_file], outputs=[output_file, status_box])
+    clear_btn.click(clear_fields, outputs=[input_file, output_file, status_box])
 
-    gr.HTML(
-        "<p style='text-align:center;font-size:14px;opacity:0.7;'>Powered by Faster-Whisper + Gradio UI</p>"
-    )
+    gr.HTML("<p style='text-align:center;font-size:14px;opacity:0.7;'>Powered by Faster-Whisper + Gradio</p>")
 
 if __name__ == "__main__":
     demo.launch()