Update app.py

app.py

@@ -2,22 +2,10 @@ import tempfile
 import traceback
 from pathlib import Path
 import gradio as gr
-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
+from pipeline import get_pipeline, to_json, to_csv, to_abab_text
 
 UPLOAD_DIR = Path(tempfile.gettempdir()) / "speech_annotation"
 UPLOAD_DIR.mkdir(parents=True, exist_ok=True)
-
 _last_segments = []
 
 def process_audio(audio_path, num_speakers):
@@ -33,20 +21,17 @@ def process_audio(audio_path, num_speakers):
     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
+    return f"✅ Done — {len(segments)} segments, {unique} speaker(s) detected", to_abab_text(segments)
 
 def export_json():
-    if not _last_segments:
-        return None
+    if not _last_segments: return None
     out = str(UPLOAD_DIR / "annotation.json")
     to_json(_last_segments, out)
     return out
 
 def export_csv():
-    if not _last_segments:
-        return None
+    if not _last_segments: return None
     out = str(UPLOAD_DIR / "annotation.csv")
     to_csv(_last_segments, out)
     return out
@@ -56,21 +41,20 @@ with gr.Blocks(title="Speech Annotation Pipeline") as demo:
     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", value="Ready.", interactive=False)
-    abab_box     = gr.Textbox(label="Transcript (ABAB)", lines=15, value="",
-                              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")
     csv_file  = gr.File(label="CSV")
 
-    run_btn.click(fn=process_audio,
-                  inputs=[audio_input, num_speakers],
+    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, show_error=True)