Update modules/ui.py

modules/ui.py

@@ -411,61 +411,38 @@ def display_chatbot_interface(lang_code):
             'initial_message': "Bonjour! Comment pouvons-nous vous aider?"
         }
     }
-
     t = translations[lang_code]
-
-    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)
-
     st.title(t['title'])
-
     if 'chatbot' not in st.session_state:
         st.session_state.chatbot = initialize_chatbot()
-
     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
-
     if 'message_counter' not in st.session_state:
         st.session_state.message_counter = 0
     
     chat_container = st.container()
-
+    
     # Mostrar mensajes existentes
     with chat_container:
-        assistant_message = st.empty()
-        full_response = ""
-        for char in response:
-            full_response += char
-            assistant_message.text_area("Assistant:", value=full_response + "▌", height=50, key=f"assistant_msg_{st.session_state.message_id}_typing_{st.session_state.message_counter}", disabled=True)
-            st.session_state.message_counter += 1
-            time.sleep(0.01)
-        assistant_message.text_area("Assistant:", value=full_response, height=50, key=f"assistant_msg_{st.session_state.message_id}", disabled=True)
+        for message in st.session_state.messages:
+            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)
+            st.session_state.message_id += 1
 
     # Á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:
             st.text_area("You:", value=user_input, height=50, key=f"user_msg_{st.session_state.message_id}", disabled=True)
-
+        
         # Generar respuesta del chatbot
         try: 
             response = get_chatbot_response(st.session_state.chatbot, user_input, lang_code)
@@ -475,26 +452,23 @@ def display_chatbot_interface(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:
             assistant_message = st.empty()
             full_response = ""
             for char in response:
                 full_response += char
-                assistant_message.text_area("Assistant:", value=full_response + "▌", height=50, key=f"assistant_msg_{st.session_state.message_id}_typing", disabled=True)
+                assistant_message.text_area("Assistant:", value=full_response + "▌", height=50, key=f"assistant_msg_{st.session_state.message_id}_typing_{st.session_state.message_counter}", disabled=True)
+                st.session_state.message_counter += 1
                 time.sleep(0.01)
             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)
-
+        
         # Limpiar el input del usuario
         st.session_state.user_input = ""
-
-    # 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()
+    st.experimental_rerun()