Update app.py

app.py

@@ -30,21 +30,28 @@ ENRUTADOR_MODELOS = {
 }
 
 # ==========================================
-# 2. LÓGICA DEL ORÁCULO Y RAG AVANZADO
+# 2. LÓGICA DEL ORÁCULO Y BÚSQUEDA HÍBRIDA
 # ==========================================
-def realizar_exegesis_web(consulta, canon_seleccionado):
+def realizar_exegesis_web(consulta, canon_seleccionado, filtro_obra):
     modelo_asignado = ENRUTADOR_MODELOS[canon_seleccionado]
     vector_query = modelo_vectorial.encode(consulta, convert_to_tensor=False).tolist()
 
+    # Si el usuario deja el filtro vacío, lo convertimos en None para la base de datos
+    if not filtro_obra or filtro_obra.strip() == "":
+        filtro_obra = None
+    else:
+        filtro_obra = filtro_obra.strip()
+
     try:
-        # MEJORA RAG: Extraemos 15 pasajes con un umbral más permisivo para no perder autores específicos
+        # BÚSQUEDA HÍBRIDA: Mandamos tanto el vector matemático como el texto del filtro
         respuesta_db = supabase.rpc(
             "buscar_pasajes_clasicos",
             {
                 "query_embedding": vector_query,
-                "match_threshold": 0.15,
+                "match_threshold": 0.15, 
                 "match_count": 15,
-                "tabla_destino": canon_seleccionado
+                "tabla_destino": canon_seleccionado,
+                "filtro_autor_obra": filtro_obra
             }
         ).execute()
 
@@ -53,20 +60,18 @@ def realizar_exegesis_web(consulta, canon_seleccionado):
         if not pasajes:
             return "❌ No database matches found. / No se hallaron testimonios. / 未找到匹配的文本。", ""
 
-        # Construimos un bloque con los 15 pasajes recuperados
         contexto_documentos = ""
         for i, frag in enumerate(pasajes):
             obra = frag.get("obra", "Desconocida")
             texto = frag.get("texto_original", "")
             contexto_documentos += f"\n[Documento {i+1} - Obra: {obra}]\n{texto}\n"
 
-        # PROMPT MASTER: Razonamiento, Reranking y Detección de Idioma
         prompt_sistema = """
         Eres un riguroso filólogo clásico e historiador.
         
         INSTRUCCIONES VITALES:
         1. IDIOMA: Detecta el idioma de la "Consulta del Usuario". Toda tu respuesta (razonamiento y exégesis) DEBE estar exclusivamente en ese idioma. Si es chino, usa obligatoriamente Chino Tradicional (繁體中文).
-        2. SELECCIÓN: Se te proporcionan hasta 15 [Documentos] antiguos. Evalúa atentamente cuál o cuáles son los más pertinentes para responder a la consulta (especialmente si se pide un autor concreto) y descarta el resto. Si no hay ninguno útil del autor pedido, indícalo.
+        2. SELECCIÓN: Se te proporcionan varios [Documentos] antiguos. Evalúa atentamente cuál o cuáles son los más pertinentes para responder a la consulta y descarta el resto.
         3. TRANSPARENCIA: Antes de dar tu respuesta final, debes escribir tu proceso lógico de selección y análisis filológico dentro de etiquetas <razonamiento> y </razonamiento>.
         
         FORMATO DE SALIDA ESPERADO:
@@ -92,7 +97,6 @@ def realizar_exegesis_web(consulta, canon_seleccionado):
 
         respuesta_completa = response.choices[0].message.content
         
-        # Parseo con expresiones regulares para separar el razonamiento de la exégesis
         match = re.search(r'<razonamiento>(.*?)</razonamiento>', respuesta_completa, re.DOTALL | re.IGNORECASE)
         
         if match:
@@ -121,6 +125,12 @@ with gr.Blocks() as demo:
                 value="textos_clasicos_latin_v3",
                 label="📚 Canon / 經典"
             )
+            # NUEVO CAMPO: Filtro opcional
+            filtro_input = gr.Textbox(
+                label="🔍 Author or Work (Optional) / Autor u Obra (Opcional) / 作者或作品（可選）",
+                placeholder="Ej: Salustio, Ab urbe condita, 忠經...",
+                lines=1
+            )
             consulta_input = gr.Textbox(
                 label="✍️ Query / Consulta / 查詢",
                 placeholder="Ask in English, Spanish or Traditional Chinese...",
@@ -134,9 +144,10 @@ with gr.Blocks() as demo:
                 
             salida_markdown = gr.Markdown(label="Exegesis / 詮釋")
 
+    # Actualizamos el click para enviar las 3 variables (consulta, canon, y filtro)
     boton_consultar.click(
         fn=realizar_exegesis_web,
-        inputs=[consulta_input, canon_dropdown],
+        inputs=[consulta_input, canon_dropdown, filtro_input],
         outputs=[salida_markdown, salida_razonamiento]
     )