Update modules/ui.py

modules/ui.py

@@ -393,21 +393,6 @@ def display_discourse_analysis_interface(nlp_models, lang_code):
                 st.pyplot(graph2)
 
 ##################################################################################################
-st.markdown("""
-    <style>
-    .stTextInput > div > div > input {
-        background-color: #F0F2F6;
-    }
-    .stTextArea > div > div > textarea {
-        background-color: #F0F2F6;
-        border: none;
-    }
-    .stTextArea > div > div > textarea:disabled {
-        color: #000000;
-    }
-    </style>
-    """, unsafe_allow_html=True)
-
 def display_chatbot_interface(lang_code):
     translations = {
         'es': {
@@ -452,24 +437,34 @@ def display_chatbot_interface(lang_code):
     if 'messages' not in st.session_state:
         st.session_state.messages = [{"role": "assistant", "content": t['initial_message']}]
 
+    if 'message_id' not in st.session_state:
+        st.session_state.message_id = 0
+
     chat_container = st.container()
 
+    # Mostrar mensajes existentes
+    with chat_container:
+        for message in st.session_state.messages:
+            st.session_state.message_id += 1
+            if message["role"] == "user":
+                st.text_area("You:", value=message["content"], height=50, key=f"user_msg_{st.session_state.message_id}", disabled=True)
+            else:
+                st.text_area("Assistant:", value=message["content"], height=50, key=f"assistant_msg_{st.session_state.message_id}", disabled=True)
+
     # Área de entrada del usuario
     user_input = st.text_input(t['input_placeholder'], key="user_input")
 
     # Procesamiento del input del usuario
     if user_input:
         st.session_state.messages.append({"role": "user", "content": user_input})
+        st.session_state.message_id += 1
         with chat_container:
-            for message in st.session_state.messages:
-                if message["role"] == "user":
-                    st.text_area("You:", value=message["content"], height=50, key=f"user_msg_{len(st.session_state.messages)}", disabled=True)
-                else:
-                    st.text_area("Assistant:", value=message["content"], height=50, key=f"assistant_msg_{len(st.session_state.messages)}", disabled=True)
+            st.text_area("You:", value=user_input, height=50, key=f"user_msg_{st.session_state.message_id}", disabled=True)
 
         # Generar respuesta del chatbot
         response = get_chatbot_response(st.session_state.chatbot, user_input, lang_code)
         st.session_state.messages.append({"role": "assistant", "content": response})
+        st.session_state.message_id += 1
 
         # Mostrar la respuesta del chatbot con efecto de escritura
         with chat_container:
@@ -477,9 +472,9 @@ def display_chatbot_interface(lang_code):
             full_response = ""
             for char in response:
                 full_response += char
-                assistant_message.text_area("Assistant:", value=full_response + "▌", height=50, key=f"assistant_msg_{len(st.session_state.messages)+1}", disabled=True)
+                assistant_message.text_area("Assistant:", value=full_response + "▌", height=50, key=f"assistant_msg_{st.session_state.message_id}_typing", disabled=True)
                 time.sleep(0.01)
-            assistant_message.text_area("Assistant:", value=full_response, height=50, key=f"assistant_msg_{len(st.session_state.messages)+1}", disabled=True)
+            assistant_message.text_area("Assistant:", value=full_response, height=50, key=f"assistant_msg_{st.session_state.message_id}", disabled=True)
 
         # Guardar la conversación en la base de datos
         store_chat_history(st.session_state.username, st.session_state.messages)
@@ -490,3 +485,6 @@ def display_chatbot_interface(lang_code):
     # Mostrar el número de llamadas a la API restantes
     remaining_calls = 50 - st.session_state.get('api_calls', 0)
     st.sidebar.write(f"Llamadas a la API restantes: {remaining_calls}")
+
+    # Forzar una actualización de la interfaz
+    st.experimental_rerun()