Update app.py

app.py

@@ -4,13 +4,13 @@ from huggingface_hub import InferenceClient
 # Initialize the client
 client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
-# Function to format the prompt
+# Function to format the prompt for Rorschach interpretation
 def format_prompt(message, history):
     prompt = "<s>"
     for user_prompt, bot_response in history:
         prompt += f"[INST] {user_prompt} [/INST]"
         prompt += f" {bot_response} "
-    prompt += f"[INST] {message} [/INST]"
+    prompt += f"[INST] Interpretación Rorschach: {message} [/INST]"
     return prompt
 
 # Function to generate response
@@ -36,24 +36,23 @@ def generate(prompt, history, temperature=0.3, max_new_tokens=256, top_p=0.95, r
     return output
 
 # Streamlit interface
-st.title("Preguntale al Buho")
+st.title("Interpretación del Test de Rorschach")
 
 # Chat history
 if 'history' not in st.session_state:
     st.session_state.history = []
 
 # User input
-Entrada_usuario = st.text_input("Tu duda:", key="Entrada_usuario")
+entrada_usuario = st.text_input("Tu interpretación del Rorschach:", key="entrada_usuario")
 
 # Generate response and update history
 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))
-#        st.session_state.user_input = ""
+    if entrada_usuario:
+        bot_response = generate(entrada_usuario, st.session_state.history)
+        st.session_state.history.append((entrada_usuario, bot_response))
 
 # Display conversation
 chat_text = ""
 for user_msg, bot_msg in st.session_state.history:
-    chat_text += f"Tu: {user_msg}\nBuhIA: {bot_msg}\n\n"
-st.text_area("La respuesta", value=chat_text, height=300, disabled=False)
+    chat_text += f"Tu: {user_msg}\nPsicólogo: {bot_msg}\n\n"
+st.text_area("Respuestas del Psicólogo", value=chat_text, height=300, disabled=False)