Update app.py

app.py

@@ -16,7 +16,6 @@ def _format_timestamp(seconds: float) -> str:
     millis = ms_rem % 1000
     return f"{hours:02d}:{minutes:02d}:{secs:02d},{millis:03d}"
 
-
 def segments_to_srt(segments: list) -> str:
     lines = []
     for i, seg in enumerate(segments, start=1):
@@ -29,21 +28,19 @@ def segments_to_srt(segments: list) -> str:
         lines.append(block)
     return "\n".join(lines)
 
-
 # -------- Config --------
-MODEL_NAME = "Systran/faster-whisper-small"  # best for HF CPU
+MODEL_NAME = "Systran/faster-whisper-small"  # optimized for HF CPU
 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 successfully.")
-
+print("Model loaded.")
 
 # -------- Core functions --------
 def extract_audio(input_path: str, out_path: str):
-    """Extract mono 16kHz WAV using ffmpeg"""
+    """Extracts mono 16 kHz WAV using ffmpeg"""
     try:
         (
             ffmpeg
@@ -55,34 +52,37 @@ def extract_audio(input_path: str, out_path: str):
     except ffmpeg.Error as e:
         stderr = getattr(e, "stderr", None)
         msg = stderr.decode() if stderr else str(e)
-        raise RuntimeError(f"FFmpeg error: {msg}")
-
+        raise RuntimeError(f"ffmpeg error: {msg}")
 
 def transcribe_file_to_srt(file_obj, language: str = "en"):
-    """Transcribe uploaded file to SRT (compatible with HF Spaces)."""
+    """Transcribe uploaded file to SRT; compatible with HF Spaces"""
     tmp_dir = Path(tempfile.mkdtemp(prefix="subgen_"))
+    input_path = Path(file_obj.name)
+
+    if not input_path.exists():
+        input_path = tmp_dir / Path(file_obj.name).name
 
-    # Ensure we can read uploaded file correctly
-    input_path = tmp_dir / Path(file_obj.name).name
-    with open(input_path, "wb") as f:
-        f.write(file_obj.read())
+    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())
 
     # 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)
 
+    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)
 
     return str(output_path), "✅ Subtitles generated successfully!"
 
-
 # -------- Gradio UI --------
 with gr.Blocks(title="AI Subtitle Generator") as demo:
     theme_state = gr.State("light")
@@ -97,58 +97,37 @@ with gr.Blocks(title="AI Subtitle Generator") as demo:
         else:
             bg = "linear-gradient(135deg, #fdfbfb, #ebedee)"
             color = "#000000"
-        return gr.update(
-            value=f"<style>body {{ background: {bg}; color: {color}; }}</style>"
-        )
+        return gr.update(value=f"<style>body {{ background: {bg}; color: {color}; }}</style>")
 
     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")
 
     with gr.Row():
-        input_file = gr.File(
-            label="Upload video/audio file",
-            file_types=None,  # accept all, Gradio will still preview audio/video
-        )
-
+        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)
+        status_box = gr.Textbox(label="Status", interactive=False)
 
     def on_click(file):
-        if file is None:
-            return None, "⚠️ Please upload a file first!"
         srt_path, msg = transcribe_file_to_srt(file)
         return srt_path, msg
 
-    def clear_all():
+    def clear_fields():
         return None, None, ""
 
-    theme_btn.click(
-        toggle_theme, inputs=[theme_state], outputs=[theme_state]
-    ).then(apply_theme, inputs=[theme_state], outputs=[style_box])
+    theme_btn.click(toggle_theme, inputs=[theme_state], outputs=[theme_state]).then(
+        apply_theme, inputs=[theme_state], outputs=[style_box]
+    )
 
     with gr.Row():
-        generate_btn = gr.Button("🎧 Generate Subtitles")
+        generate_btn = gr.Button("Generate Subtitles")
         clear_btn = gr.Button("🧹 Clear")
 
-    generate_btn.click(
-        on_click, inputs=[input_file], outputs=[output_file, status_box]
-    )
-
-    clear_btn.click(
-        clear_all,
-        inputs=[],
-        outputs=[input_file, output_file, status_box],
-    )
+    generate_btn.click(on_click, inputs=[input_file], outputs=[output_file, status_box])
+    clear_btn.click(fn=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 UI</p>")
 
 if __name__ == "__main__":
-    demo.queue().launch()
+    demo.launch()