Update rag_hf.py

rag_hf.py

@@ -165,33 +165,73 @@ def query_llm(prompt):
     except Exception as e:
         return f"Error al consultar el modelo: {str(e)}"
 
-def generate_response(matrix, id_map, G, rdf, user_question, k, embedder):
-    ids = get_top_k(matrix, id_map, user_question, k, embedder)
+# === PROMPT PARA MODELO MISTRAL ===
+def generate_response(matrix, id_map, G, rdf, user_question, k=3):
+    ids = get_top_k(matrix, id_map, user_question, k)
     context = [get_context(G, i) for i in ids]
     rdf_facts = []
     for i in ids:
         rdf_facts.extend([f"{p}: {v}" for p, v in query_rdf(rdf, i)])
 
-    prompt_es = (
-        "Contexto: " + " ".join(context) + "\n" +
-        "Hechos RDF: " + ", ".join(rdf_facts) + "\n" +
-        f"Pregunta: {user_question} Responde en español:"
-    )
+    prompt_es = f"""<s>[INST]
+Eres un experto en lenguas indígenas sudamericanas. Utiliza estricta y únicamente la información a continuación para responder la pregunta del usuario en **español**.
+- No infieras ni asumas hechos que no estén explícitamente establecidos.
+- Si la respuesta es desconocida o insuficiente, di \"No puedo responder con los datos disponibles.\"
+- Limita tu respuesta a 100 palabras.
 
-    prompt_en = (
-        "Context: " + " ".join(context) + "\n" +
-        "RDF facts: " + ", ".join(rdf_facts) + "\n" +
-        f"Question: {user_question} Answer in English:"
-    )
+### CONTEXTO:
+{chr(10).join(context)}
 
-    response_es = query_llm(prompt_es)
-    response_en = query_llm(prompt_en)
+### RELACIONES RDF:
+{chr(10).join(rdf_facts)}
 
-    full_response = (
-        f"<b>Respuesta en español:</b><br>{response_es}<br><br>"
-        f"<b>Answer in English:</b><br>{response_en}"
-    )
-    return full_response, ids, context, rdf_facts
+### PREGUNTA:
+{user_question}
+
+Respuesta:
+[/INST]"""
+
+    prompt_en = f"""<s>[INST]
+You are an expert in South American indigenous languages.
+Use strictly and only the information below to answer the user question in **English**.
+- Do not infer or assume facts that are not explicitly stated.
+- If the answer is unknown or insufficient, say \"I cannot answer with the available data.\"
+- Limit your answer to 100 words.
+
+### CONTEXT:
+{chr(10).join(context)}
+
+### RDF RELATIONS:
+{chr(10).join(rdf_facts)}
+
+### QUESTION:
+{user_question}
+
+Answer:
+[/INST]"""
+
+    try:
+        res_es = requests.post(
+            ENDPOINT_URL,
+            headers={"Authorization": f"Bearer {HF_API_TOKEN}", "Content-Type": "application/json"},
+            json={"inputs": prompt_es}, timeout=60
+        )
+        res_en = requests.post(
+            ENDPOINT_URL,
+            headers={"Authorization": f"Bearer {HF_API_TOKEN}", "Content-Type": "application/json"},
+            json={"inputs": prompt_en}, timeout=60
+        )
+        out_es = res_es.json()
+        out_en = res_en.json()
+
+        response_es = out_es[0]["generated_text"].replace(prompt_es.strip(), "").strip() if isinstance(out_es, list) and "generated_text" in out_es[0] else "Error en respuesta en español."
+        response_en = out_en[0]["generated_text"].replace(prompt_en.strip(), "").strip() if isinstance(out_en, list) and "generated_text" in out_en[0] else "Error in English response."
+
+        combined = f"<b>Respuesta en español:</b><br>{response_es}<br><br><b>Answer in English:</b><br>{response_en}"
+        return combined, ids, context, rdf_facts
+
+    except Exception as e:
+        return f"Error al consultar el modelo: {str(e)}", ids, context, rdf_facts
 
 # === UI MAIN ===
 def main():