Update rag_hf.py

rag_hf.py

@@ -205,8 +205,19 @@ Answer:
         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."
+        if isinstance(out_es, dict) and "generated_text" in out_es:
+            response_es = out_es["generated_text"].strip()
+        elif isinstance(out_es, list) and "generated_text" in out_es[0]:
+            response_es = out_es[0]["generated_text"].replace(prompt_es.strip(), "").strip()
+        else:
+            response_es = "Error en respuesta en español."
+
+        if isinstance(out_en, dict) and "generated_text" in out_en:
+            response_en = out_en["generated_text"].strip()
+        elif isinstance(out_en, list) and "generated_text" in out_en[0]:
+            response_en = out_en[0]["generated_text"].replace(prompt_en.strip(), "").strip()
+        else:
+            response_en = "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
@@ -214,6 +225,7 @@ Answer:
     except Exception as e:
         return f"Error al consultar el modelo: {str(e)}", ids, context, rdf_facts
 
+
 # === MAIN ===
 def main():
     methods, embedder = load_all_components()