Update pages/trabajos.py

pages/trabajos.py

@@ -1,10 +1,10 @@
 import streamlit as st
-from utils.data_manager import flujo_laboratorio
+from utils.data_manager import flujo_laboratorio, obtener_respuesta
 
 def show():
-    st.title("🦷 Gestión de Trabajos de Laboratorio")
     flujo_laboratorio()
-if 'mostrar_chat' not in st.session_state:
+
+    if 'mostrar_chat' not in st.session_state:
         st.session_state['mostrar_chat'] = False
 
     def mostrar_chat():
@@ -30,14 +30,19 @@ if 'mostrar_chat' not in st.session_state:
             margin-right: 10px;
         }
         </style>
+        <button class="assistant-button" onclick="window.location.href='#assistant_chat'">
+            <img src='https://img2.gratispng.com/20180808/cxq/kisspng-robotics-science-computer-icons-robot-technology-robo-to-logo-svg-png-icon-free-download-45527-5b6baa46a5e322.4713113715337825986795.jpg' alt='icon' width='20' height='20'/>
+            Hablar con Asistente
+        </button>
         """,
         unsafe_allow_html=True
     )
 
-    if st.button("Hablar con Asistente"):
+    if st.button("Mostrar Chat", key="assistant_button"):
         mostrar_chat()
 
     if st.session_state['mostrar_chat']:
+        st.markdown("<div id='assistant_chat'></div>", unsafe_allow_html=True)
         st.markdown("### Chat con Asistente")
         pregunta_usuario = st.text_input("Escribe tu pregunta aquí:")
         if st.button("Enviar Pregunta"):
@@ -46,6 +51,6 @@ if 'mostrar_chat' not in st.session_state:
                 "",  # No se necesita texto preprocesado en este caso
                 st.session_state['modelo'], 
                 st.session_state['temperatura'],
-                st.session_state.get('assistant_id', '')
+                st.session_state.get('assistant_id', 'asst_KngkX6sbRccg5a6fcnDHO06R')
             )
             st.markdown(f"**Asistente**: {respuesta}")