import gradio as gr
import csv
import os
from pydub import AudioSegment
from pydub.playback import play
from rapidfuzz import process

DATASET = "dataset.csv"

def cargar_dataset():
    with open(DATASET, newline='', encoding='utf-8') as f:
        reader = csv.DictReader(f)
        return [{"path": row["path"], "text": row["text"].lower().strip()} for row in reader]

dataset = cargar_dataset()

def buscar_audio(texto):
    texto = texto.lower().strip()
    frases = [d["text"] for d in dataset]
    
    match = process.extractOne(texto, frases, score_cutoff=30)

    if not match:
        return None, "❌ No encontré coincidencias en tus clips grabados."

    mejor_texto = match[0]
    ruta = next((d["path"] for d in dataset if d["text"] == mejor_texto), None)

    if ruta and os.path.exists(ruta):
        return ruta, f"🎧 Clip encontrado: {ruta}"
    else:
        return None, "⚠️ Algo falló encontrando el archivo de audio."

def demo_func(texto):
    ruta, mensaje = buscar_audio(texto)
    return (ruta if ruta else None), mensaje

demo = gr.Interface(
    fn=demo_func,
    inputs=gr.Textbox(label="Escribe una frase similar a una grabada"),
    outputs=[gr.Audio(label="Clip de voz"), gr.Textbox(label="Resultado")],
    title="🐽 Voz Kuchiyuya Real",
    description="Este demo busca y reproduce tu grabación más parecida al texto ingresado."
)

if __name__ == "__main__":
    demo.launch()