Update app.py

app.py

@@ -1,5 +1,9 @@
 import streamlit as st
 from huggingface_hub import InferenceClient
+from docx import Document
+import tempfile
+import os
+import re
 
 # Initialize the client
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
@@ -35,6 +39,32 @@ def generate(prompt, history, temperature=0.9, max_new_tokens=256, top_p=0.95, r
         output += response.token.text
     return output
 
+# Function to replace variables in a Word template
+def replace_variables_word(doc, variables):
+    for paragraph in doc.paragraphs:
+        for key, value in variables.items():
+            if f'<{key}>' in paragraph.text:
+                paragraph.text = re.sub(f'<{key}>', value, paragraph.text)
+    for table in doc.tables:
+        for row in table.rows:
+            for cell in row.cells:
+                for paragraph in cell.paragraphs:
+                    for key, value in variables.items():
+                        if f'<{key}>' in paragraph.text:
+                            paragraph.text = re.sub(f'<{key}>', value, paragraph.text)
+
+# Function to generate a Word document with interpretation
+def generate_word_document(interpretation):
+    template_path = os.path.join('PLANTILLAS', 'PLANTILLA_INTERPRETACION.docx')
+    doc = Document(template_path)
+    
+    variables = {'INTERPRETACION': interpretation}
+    replace_variables_word(doc, variables)
+
+    with tempfile.NamedTemporaryFile(delete=False, suffix='.docx') as tmp:
+        doc.save(tmp.name)
+        return tmp.name
+
 # Streamlit interface
 st.title("Interpretación del Test de Rorschach")
 
@@ -43,14 +73,26 @@ if 'history' not in st.session_state:
     st.session_state.history = []
 
 # User input
-entrada_usuario = st.text_input("Que ves en las imagenes ..:", key="entrada_usuario")
+entrada_usuario = st.text_input("Que ves en las imágenes ..:", key="entrada_usuario")
 
-# Generate response and update history
+# Generate response and create Word document
 if st.button("Enviar"):
     if entrada_usuario:
         bot_response = generate(entrada_usuario, st.session_state.history)
         st.session_state.history.append((entrada_usuario, bot_response))
 
+        # Generate Word document
+        document_path = generate_word_document(bot_response)
+        
+        # Provide download link
+        with open(document_path, "rb") as file:
+            st.download_button(
+                label="Descargar Interpretación",
+                data=file,
+                file_name="Interpretacion_Rorschach.docx",
+                mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+            )
+
 # Display conversation
 chat_text = ""
 for user_msg, bot_msg in st.session_state.history: