Upload folder using huggingface_hub

src/app_kuchiyuya.py

@@ -1,12 +1,11 @@
 import gradio as gr
 import csv
 import os
-import difflib
 from pydub import AudioSegment
 from pydub.playback import play
+from rapidfuzz import process
 
 DATASET = "dataset.csv"
-AUDIO_DIR = "voces"
 
 def cargar_dataset():
     with open(DATASET, newline='', encoding='utf-8') as f:
@@ -17,13 +16,20 @@ dataset = cargar_dataset()
 
 def buscar_audio(texto):
     texto = texto.lower().strip()
-    coincidencias = difflib.get_close_matches(texto, [d["text"] for d in dataset], n=1, cutoff=0.6)
-    if not coincidencias:
+    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."
-    for d in dataset:
-        if d["text"] == coincidencias[0]:
-            return d["path"], f"🎧 Clip encontrado: {d['path']}"
-    return None, "⚠️ Algo falló en la búsqueda."
+
+    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)
@@ -38,4 +44,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch()