Update app.py

app.py

@@ -5,39 +5,56 @@ import tempfile
 import os
 import re
 
-# Initialize the client
-client = InferenceClient("Qwen/Qwen2-0.5B-Instruct")
+# Initialize the client with a publicly available model that doesn't require API token
+# Models like facebook/opt-350m, EleutherAI/gpt-neo-125m, or bigscience/bloom-560m are good options
+client = InferenceClient(model="facebook/opt-1.3b")
 
 # Function to format the prompt for Rorschach interpretation
 def format_prompt(message, history):
-    prompt = "<s>"
+    # For OPT model, we use a simpler prompt format
+    # Start with context about the task
+    prompt = "Esto es una interpretación del Test de Rorschach. "
+    
+    # Add history if any
     for user_prompt, bot_response in history:
-        prompt += f"[INST] {user_prompt} [/INST]"
-        prompt += f" {bot_response} "
-    prompt += f"[INST] Interpretación Rorschach: {message} [/INST]"
+        prompt += f"Persona: {user_prompt} Respuesta: {bot_response} "
+    
+    # Add current message
+    prompt += f"Persona describe lo que ve en la mancha de Rorschach: {message} Interpretación psicológica detallada: "
+    
     return prompt
 
 # Function to generate response
 def generate(prompt, history, temperature=0.9, max_new_tokens=512, top_p=0.95, repetition_penalty=1.0):
-    temperature = max(float(temperature), 1e-2)
-    top_p = float(top_p)
+    try:
+        temperature = max(float(temperature), 1e-2)
+        top_p = float(top_p)
 
-    generate_kwargs = dict(
-        temperature=temperature,
-        max_new_tokens=max_new_tokens,
-        top_p=top_p,
-        repetition_penalty=repetition_penalty,
-        do_sample=True,
-        seed=42,
-    )
+        generate_kwargs = dict(
+            temperature=temperature,
+            max_new_tokens=max_new_tokens,
+            top_p=top_p,
+            repetition_penalty=repetition_penalty,
+            do_sample=True,
+            seed=42,
+        )
 
-    formatted_prompt = format_prompt(prompt, history)
-
-    stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
-    output = ""
-    for response in stream:
-        output += response.token.text
-    return output
+        formatted_prompt = format_prompt(prompt, history)
+        
+        # Simplified call to text_generation for OPT model
+        response = client.text_generation(
+            formatted_prompt,
+            max_new_tokens=max_new_tokens,
+            temperature=temperature,
+            top_p=top_p,
+            repetition_penalty=repetition_penalty,
+            do_sample=True
+        )
+        
+        return response
+    except Exception as e:
+        st.error(f"Error generando respuesta: {str(e)}")
+        return "Lo siento, hubo un error al generar la interpretación. Por favor, intenta de nuevo."
 
 # Function to replace variables in a Word template
 def replace_variables_word(doc, variables):
@@ -45,6 +62,7 @@ def replace_variables_word(doc, variables):
         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:
@@ -55,15 +73,25 @@ def replace_variables_word(doc, variables):
 
 # 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)
+    try:
+        template_path = os.path.join('PLANTILLAS', 'PLANTILLA_INTERPRETACION.docx')
+        if not os.path.exists(template_path):
+            st.warning(f"La plantilla no se encontró en: {template_path}")
+            # Fallback: Create a simple document without template
+            doc = Document()
+            doc.add_heading('Interpretación del Test de Rorschach', 0)
+            doc.add_paragraph(interpretation)
+        else:
+            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
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.docx') as tmp:
+            doc.save(tmp.name)
+            return tmp.name
+    except Exception as e:
+        st.error(f"Error generando documento: {str(e)}")
+        return None
 
 # Streamlit interface
 st.title("Interpretación del Test de Rorschach")
@@ -73,28 +101,37 @@ if 'history' not in st.session_state:
     st.session_state.history = []
 
 # User input
-entrada_usuario = st.text_input("Que ves en las imágenes ..:", key="entrada_usuario")
+entrada_usuario = st.text_input("Qué ves en las imágenes:", key="entrada_usuario")
 
 # 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))
+        with st.spinner("Generando interpretación..."):
+            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"
-            )
+            # Generate Word document
+            document_path = generate_word_document(bot_response)
+            
+            if document_path:
+                # 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"
+                    )
+                # Clean up the temporary file
+                try:
+                    os.unlink(document_path)
+                except:
+                    pass
 
 # Display conversation
-chat_text = ""
-for user_msg, bot_msg in st.session_state.history:
-    chat_text += f"Tu: {user_msg}\nBuho: {bot_msg}\n\n"
-st.text_area("Respuestas del Buho", value=chat_text, height=300, disabled=False)
+if st.session_state.history:
+    st.subheader("Historial de Conversación")
+    chat_text = ""
+    for user_msg, bot_msg in st.session_state.history:
+        chat_text += f"Tú: {user_msg}\nBuho: {bot_msg}\n\n"
+    st.text_area("Respuestas del Buho", value=chat_text, height=300, disabled=True)