Update app.py

app.py

@@ -12,6 +12,8 @@ import pandas as pd
 import altair as alt
 from datetime import datetime
 from tenacity import retry, stop_after_attempt, wait_random_exponential
+from io import BytesIO
+import docx
 
 st.set_page_config(page_title="El Detective de Alimentos", page_icon="🍎", layout="wide")
 
@@ -461,6 +463,33 @@ INTEGRATED_NEURO_FOOD_MAP = {
     }
 }
 
+def generate_word_report(report_text):
+    """
+    Carga una plantilla de Word, reemplaza un marcador de posición y la devuelve como un objeto binario en memoria.
+    """
+    try:
+        # Carga el documento de plantilla
+        template_path = os.path.join("PLANTILLAS", "PLANTILLA_INTERPRETACION.docx")
+        doc = docx.Document(template_path)
+
+        # Busca y reemplaza el marcador de posición
+        for paragraph in doc.paragraphs:
+            if '<INTERPRETACION>' in paragraph.text:
+                # Limpiamos el párrafo del marcador para empezar de cero
+                paragraph.clear()
+                # Añadimos el nuevo texto. add_run permite añadir más formato si se quiere en el futuro
+                paragraph.add_run(report_text)
+
+        # Guarda el documento modificado en un buffer de memoria
+        doc_io = BytesIO()
+        doc.save(doc_io)
+        doc_io.seek(0) # Rebobina el buffer al principio para que st.download_button pueda leerlo
+        
+        return doc_io
+    
+    except Exception as e:
+        logger.error(f"Error al generar el informe de Word: {e}")
+        return None
 
 def sanitize_text(text):
     if not text: return ""
@@ -753,15 +782,21 @@ if st.session_state.search_results is not None:
         col1, col2 = st.columns([3,1])
         with col1:
             st.success(f"Hemos encontrado {len(results)} posible(s) causa(s) relacionada(s) con tu caso.")
-        with col2:
-            report_data = generate_report_text(st.session_state.user_query, results)
-            st.download_button(
-                label="📄 Descargar Informe", 
-                data=report_data, 
-                file_name=f"informe_detective_{datetime.now().strftime('%Y%m%d')}.txt", 
-                mime="text/plain",
-                key="download_main_report"
-            )
+with col2:
+            # 1. Genera el contenido de texto como siempre
+            report_data_text = generate_report_text(st.session_state.user_query, results)
+            
+            # 2. Procesa la plantilla de Word con ese texto
+            word_file_buffer = generate_word_report(report_data_text)
+            
+            if word_file_buffer:
+                st.download_button(
+                    label="📄 Descargar Informe (Word)", 
+                    data=word_file_buffer, 
+                    file_name=f"Informe_Detective_Alimentos_{datetime.now().strftime('%Y%m%d')}.docx",
+                    mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document",
+                    key="download_word_report"
+                )
         
         st.subheader("Análisis de Relevancia de las Coincidencias")
         st.altair_chart(create_relevance_chart(results), use_container_width=True)