Update app.py

app.py

@@ -2,7 +2,18 @@ import tempfile
 import traceback
 from pathlib import Path
 import gradio as gr
-from pipeline import get_pipeline, to_json, to_csv, to_abab_text
+import importlib.util, sys
+
+# Load pipeline module explicitly to avoid circular import
+spec = importlib.util.spec_from_file_location("speech_pipeline", "/app/pipeline.py")
+mod = importlib.util.module_from_spec(spec)
+sys.modules["speech_pipeline"] = mod
+spec.loader.exec_module(mod)
+
+get_pipeline = mod.get_pipeline
+to_json      = mod.to_json
+to_csv       = mod.to_csv
+to_abab_text = mod.to_abab_text
 
 UPLOAD_DIR = Path(tempfile.gettempdir()) / "speech_annotation"
 UPLOAD_DIR.mkdir(parents=True, exist_ok=True)
@@ -19,16 +30,13 @@ def process_audio(audio_path, num_speakers):
         segments = pipeline.process(audio_path, num_speakers=n)
     except Exception as e:
         return f"❌ Error: {e}\n{traceback.format_exc()}", ""
-
     if not segments:
         return "⚠️ No speech detected.", ""
-
     _last_segments = segments
     abab = to_abab_text(segments)
     unique = len(set(s.speaker for s in segments))
     return f"✅ Done — {len(segments)} segments, {unique} speaker(s) detected", abab
 
-
 def export_json():
     if not _last_segments:
         return None
@@ -43,31 +51,26 @@ def export_csv():
     to_csv(_last_segments, out)
     return out
 
-
 with gr.Blocks(title="Speech Annotation Pipeline") as demo:
     gr.Markdown("# 🎙️ Speech Annotation Pipeline")
-    gr.Markdown("Upload audio → get speaker-labeled transcripts automatically")
-
-    audio_input = gr.Audio(label="Upload Audio (.wav/.mp3/.flac)", type="filepath")
+    audio_input  = gr.Audio(label="Upload Audio (.wav/.mp3/.flac)", type="filepath")
     num_speakers = gr.Slider(minimum=0, maximum=10, step=1, value=0,
                              label="Number of speakers (0 = auto-detect)")
-    run_btn = gr.Button("▶ Run Annotation", variant="primary")
-    status_box = gr.Textbox(label="Status", interactive=False)
-    abab_box = gr.Textbox(label="Editable Transcript (ABAB)", lines=20,
-                          placeholder="Annotations will appear here...")
+    run_btn      = gr.Button("▶ Run Annotation", variant="primary")
+    status_box   = gr.Textbox(label="Status", value="Ready.", interactive=False)
+    abab_box     = gr.Textbox(label="Transcript (ABAB)", lines=15, value="",
+                              placeholder="Annotations will appear here...")
     with gr.Row():
         json_btn = gr.Button("Export JSON")
         csv_btn  = gr.Button("Export CSV")
-    json_file = gr.File(label="JSON Download")
-    csv_file  = gr.File(label="CSV Download")
+    json_file = gr.File(label="JSON")
+    csv_file  = gr.File(label="CSV")
 
-    run_btn.click(
-        fn=process_audio,
-        inputs=[audio_input, num_speakers],
-        outputs=[status_box, abab_box]
-    )
+    run_btn.click(fn=process_audio,
+                  inputs=[audio_input, num_speakers],
+                  outputs=[status_box, abab_box])
     json_btn.click(fn=export_json, inputs=[], outputs=[json_file])
-    csv_btn.click(fn=export_csv, inputs=[], outputs=[csv_file])
+    csv_btn.click(fn=export_csv,  inputs=[], outputs=[csv_file])
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=7860, show_error=True)