Upload app.py

app.py

@@ -0,0 +1,128 @@
+import gradio as gr
+import json
+import os
+import shutil
+from kalbetojai_analize import analizuoti_kalbetojus
+from kalbos_nustatymas import transcribe_text, transcribe_text_wav2vec
+from ataskaita import sugeneruoti_ataskaita
+
+# ⚖️ Demo failų keliai
+DEMO_KALBETOJAI_PATH = "demo/demo_3kalbetojai.wav"
+DEMO_SLAPTAZODIS_PATH = "demo/slaptazodis.wav"
+
+# 🔐 Funkcija kalbėtojų analizei ir segmentų atvaizdavimui
+def analizuoti_ir_issaugoti(wav_failas, modelis):
+    if wav_failas is None:
+        return "⚠️ Prašome įkelti pagrindinį .wav failą."
+
+    # 🔥 1. Kopijuojame į /tmp
+    laikinas_kelias = "/tmp/ivestis.wav"
+    shutil.copy(wav_failas, laikinas_kelias)
+
+    # 🔥 2. Atliekame analizę
+    tekstas_ataskaitai, _, segmentai = analizuoti_kalbetojus(modelis, failas=laikinas_kelias)
+
+    # 🔥 3. Įrašome segmentus į JSON
+    os.makedirs("rezultatai", exist_ok=True)
+    failas = os.path.join("rezultatai", f"{modelis.lower()}.json")
+
+    try:
+        with open(failas, "r", encoding="utf-8") as f:
+            esami_duomenys = json.load(f)
+    except:
+        esami_duomenys = {}
+
+    esami_duomenys["segmentai"] = segmentai  # Atnaujinam segmentus
+    with open(failas, "w", encoding="utf-8") as f:
+        json.dump(esami_duomenys, f, ensure_ascii=False, indent=2)
+
+    # 🔥 4. Perskaitome analizės trukmę iš JSON
+    try:
+        with open(failas, "r", encoding="utf-8") as f:
+            duomenys = json.load(f)
+            laikas = duomenys.get("apdorojimo_laikas_tekstu", "?")
+    except Exception:
+        laikas = "?"
+
+    # 🔥 5. Formuojame grąžinamą tekstą
+    tekstas = f"⏱️ {modelis} analizės trukmė: {laikas}\n\n"
+    for s in segmentai:
+        tekstas += f"🧑 Kalbėtojas {s['kalbetojas']} – {s['kalba']}\n"
+        tekstas += f"💬 '{s['tekstas']}'\n"
+        tekstas += f"⏱️ Trukmė: {s['trukme']} s\n\n"
+
+    tekstas += "\n" + tekstas_ataskaitai  # Pridedame ataskaitos informaciją
+
+    return tekstas
+
+# 🔐 Funkcija slaptažodžiui tikrinti iš įkelto WAV
+def tikrinti_slaptazodi(audio, slaptazodis, modelis):
+    if audio is None or not slaptazodis:
+        return "⚠️ Prašome įkelti slaptažodžio įrašą ir įvesti tekstą."
+
+    slaptazodis = slaptazodis.strip().lower()
+    if modelis == "Whisper":
+        tekstas = transcribe_text(audio).lower().strip()
+    elif modelis == "Wav2Vec2":
+        tekstas = transcribe_text_wav2vec(audio, kalba="lt").lower().strip()
+    else:
+        return "❌ Nepalaikomas modelis."
+
+    if slaptazodis in tekstas:
+        return f"✅ Slaptažodis atpažintas! ({tekstas})"
+    else:
+        return f"❌ Slaptažodis neatpažintas. Gauta: ({tekstas})"
+
+# 🚀 Mygtukų funkcijos demo failams
+def naudoti_demo_kalbetojai():
+    return DEMO_KALBETOJAI_PATH
+
+def naudoti_demo_slaptazodis():
+    return DEMO_SLAPTAZODIS_PATH
+
+# 📥 Nauja funkcija: grąžinti json failą parsisiuntimui
+def gauti_json_faila(modelis):
+    failo_kelias = os.path.join("rezultatai", f"{modelis.lower()}.json")
+    if os.path.exists(failo_kelias):
+        return failo_kelias
+    else:
+        return None
+
+# 💻 GUI komponentai
+with gr.Blocks() as demo:
+    gr.Markdown("## 🎤 Kalbėtojų analizė + slaptažodžio tikrinimas + ataskaita")
+
+    with gr.Tab("1. 📦 Kalbėtojų analizė"):
+        with gr.Row():
+            kalbetojai_audio = gr.Audio(type="filepath", label="🔊 Įkelkite .wav su 3 kalbėtojais")
+            model_selector1 = gr.Radio(["Whisper", "Wav2Vec2"], value="Whisper", label="📊 Modelis")
+        demo_kalb_btn = gr.Button("🎧 Naudoti pavyzdinį kalbėtojų failą")
+        analizės_output = gr.Textbox(label="📜 Rezultatai", lines=20)
+        analizės_btn = gr.Button("▶️ Analizuoti kalbėtojus")
+        demo_kalb_btn.click(fn=naudoti_demo_kalbetojai, outputs=kalbetojai_audio)
+        analizės_btn.click(fn=analizuoti_ir_issaugoti, inputs=[kalbetojai_audio, model_selector1], outputs=analizės_output)
+
+    with gr.Tab("2. 🔐 Slaptažodžio tikrinimas"):
+        with gr.Row():
+            slapt_audio = gr.Audio(type="filepath", label="🔊 Įkelkite .wav su išsakytu slaptažodžiu")
+            slapt_input = gr.Textbox(label="🔐 Įveskite slaptažodį")
+            model_selector2 = gr.Radio(["Whisper", "Wav2Vec2"], value="Whisper", label="📊 Modelis")
+        demo_slapt_btn = gr.Button("🔹 Naudoti pavyzdinį slaptažodžio failą")
+        slapt_output = gr.Textbox(label="📋 Tikrinimo rezultatas", lines=4)
+        slapt_btn = gr.Button("✅ Tikrinti slaptažodį")
+        demo_slapt_btn.click(fn=naudoti_demo_slaptazodis, outputs=slapt_audio)
+        slapt_btn.click(fn=tikrinti_slaptazodi, inputs=[slapt_audio, slapt_input, model_selector2], outputs=slapt_output)
+
+    with gr.Tab("3. 📊 Ataskaita ir Atsisiuntimas"):
+        with gr.Row():
+            report_output = gr.Textbox(label="📈 Bendra ataskaita", lines=30)
+            report_btn = gr.Button("📊 Peržiūrėti ataskaitą")
+            report_btn.click(fn=sugeneruoti_ataskaita, outputs=report_output)
+        
+        with gr.Row():
+            atsisiusti_modeli = gr.Radio(["Whisper", "Wav2Vec2"], value="Whisper", label="📁 Pasirinkite modelį JSON atsisiuntimui")
+            atsisiuntimo_btn = gr.Button("📥 Atsisiųsti JSON")
+            failas_output = gr.File(label="⬇️ Parsisiųsti JSON failą")
+            atsisiuntimo_btn.click(fn=gauti_json_faila, inputs=[atsisiusti_modeli], outputs=failas_output)
+
+demo.launch()