Update app.py

app.py

@@ -1099,15 +1099,38 @@ 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:
-            if word_file_buffer:
+            with col2:
+            # Inicializar el estado si no existe
+            if 'report_generated' not in st.session_state:
+                st.session_state.report_generated = False
+                st.session_state.word_file_buffer = None
+
+            # Botón para iniciar la generación del informe
+            if st.button("📄 Preparar Informe (Word)", key="prepare_report"):
+                with st.spinner("Generando su informe personalizado, por favor espere..."):
+                    # Generar todo el contenido del informe
+                    ai_analysis_text = st.session_state.analysis_cache.get('best_match_analysis', "Análisis no disponible.")
+                    base_report_text = generate_report_text(st.session_state.user_query, results)
+                    neuro_report_text = generate_neuro_report_text(st.session_state.entities, FOOD_TO_COMPOUND_MAP, INTEGRATED_NEURO_FOOD_MAP)
+                    molecular_report_text = generate_molecular_report_text(best_match, st.session_state.entities, foodb_index, FOOD_NAME_TO_FOODB_KEY, COMPOUND_SYNONYM_MAP, KNOWN_TRIGGERS_MAP)
+                    
+                    complete_report_string = f"{base_report_text}\n\n{ai_analysis_text}\n{neuro_report_text}\n{molecular_report_text}"
+                    
+                    # Guardar el buffer del archivo de Word en el estado de la sesión
+                    st.session_state.word_file_buffer = generate_word_report(complete_report_string)
+                    st.session_state.report_generated = True
+            
+            # Si el informe ya se generó, mostrar el botón de descarga
+            if st.session_state.report_generated and st.session_state.word_file_buffer:
                 st.download_button(
-                    label="📄 Descargar Informe Completo (Word)", 
-                    data=word_file_buffer, 
+                    label="✅ Descargar Informe Ahora",
+                    data=st.session_state.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"
                 )
+                # Opcional: Mostrar un mensaje de éxito
+                st.success("¡Su informe está listo!")
         
         st.subheader("Análisis de Relevancia de las Coincidencias")
         st.altair_chart(create_relevance_chart(results), use_container_width=True)