Update utils/data_manager.py

utils/data_manager.py

@@ -353,8 +353,9 @@ def mostrar_datos_como_texto(datos):
 
 def flujo_presupuestos():
     st.title("💰 Asistente de Presupuestos")
-    st.markdown("Hola Dr. cuénteme en que puedo ayudarle?")
+    st.markdown("Hola Dr. cuénteme en qué puedo ayudarle?")
 
+    # Listado de precios
     lista_precios = {
         "Restauraciones en resina de una superficie": 75000,
         "Restauraciones en resina de dos superficies": 95000,
@@ -381,49 +382,92 @@ def flujo_presupuestos():
         "Endodoncia de premolares superiores": 480000,
     }
 
+    if 'presupuesto' not in st.session_state:
+        st.session_state['presupuesto'] = []
+
+    # Mostrar lista de precios en un menú desplegable
+    with st.expander("Mostrar Lista de Precios"):
+        st.markdown("### Lista de Precios")
+        for tratamiento, precio in lista_precios.items():
+            st.markdown(f"**{tratamiento}:** {precio} COP")
+
+    # Selección de tratamientos y cantidad
     tratamiento = st.selectbox("Selecciona el tratamiento", list(lista_precios.keys()))
     cantidad = st.number_input("Cantidad", min_value=1, step=1)
-    
-    if st.button("Calcular Presupuesto"):
-        if tratamiento and cantidad:
-            precio_total = lista_precios[tratamiento] * cantidad
-            st.success(f"El precio total para {cantidad} {tratamiento} es: {precio_total} COP")
-
-def flujo_radiografias():
-    st.title("📸 Registro de Radiografías")
-
-    if 'radiografias' not in st.session_state:
-        st.session_state.radiografias = []
-
-    with st.form("radiografias_form"):
-        nombre_paciente = st.text_input("Nombre del Paciente:")
-        tipo_radiografia = st.selectbox("Tipo de Radiografía:", ["Periapical", "Panorámica", "Cefalométrica"])
-        fecha_realizacion = st.date_input("Fecha de Realización:")
-        observaciones = st.text_area("Observaciones:")
+    agregar = st.button("Agregar al Presupuesto")
+
+    if agregar:
+        precio_total = lista_precios[tratamiento] * cantidad
+        st.session_state['presupuesto'].append({"tratamiento": tratamiento, "cantidad": cantidad, "precio_total": precio_total})
+        st.success(f"Agregado: {cantidad} {tratamiento} - Total: {precio_total} COP")
+
+    # Mostrar servicios seleccionados y total
+    if st.session_state['presupuesto']:
+        st.write("### Servicios Seleccionados")
+        total_presupuesto = sum(item['precio_total'] for item in st.session_state['presupuesto'])
+        for item in st.session_state['presupuesto']:
+            st.write(f"{item['cantidad']} x {item['tratamiento']} - {item['precio_total']} COP")
+        st.write(f"**Total: {total_presupuesto} COP**")
+
+    # Copiar presupuesto al asistente de chat
+    copiar_presupuesto = st.button("Copiar Presupuesto al Asistente")
+    if copiar_presupuesto:
+        servicios = "\n".join([f"{item['cantidad']} x {item['tratamiento']} - {item['precio_total']} COP" for item in st.session_state['presupuesto']])
+        total = f"**Total: {total_presupuesto} COP**"
+        st.session_state['presupuesto_texto'] = f"{servicios}\n{total}"
+        st.success("Presupuesto copiado al asistente de chat.")
+
+    if 'mostrar_chat' not in st.session_state:
+        st.session_state['mostrar_chat'] = False
+
+    def mostrar_chat():
+        st.session_state['mostrar_chat'] = not st.session_state['mostrar_chat']
+
+    st.markdown(
+        """
+        <style>
+        .assistant-button {
+            display: flex;
+            align-items: center;
+            justify-content: center;
+            background-color: #4CAF50;
+            color: white;
+            padding: 10px;
+            border: none;
+            border-radius: 5px;
+            cursor: pointer;
+            font-size: 16px;
+            margin-top: 10px;
+        }
+        .assistant-button img {
+            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("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")
         
-        submitted = st.form_submit_button("Registrar Radiografía")
-
-        if submitted:
-            radiografia = {
-                "nombre_paciente": nombre_paciente,
-                "tipo_radiografia": tipo_radiografia,
-                "fecha_realizacion": str(fecha_realizacion),
-                "observaciones": observaciones
-            }
-            st.session_state.radiografias.append(radiografia)
-            datos_guardados = mostrar_datos_como_texto([radiografia])
-            guardar_en_txt('radiografias.txt', datos_guardados)
-            st.success("Radiografía registrada con éxito.")
-
-    if st.session_state.radiografias:
-        st.write("### Radiografías Registradas")
-        df_radiografias = pd.DataFrame(st.session_state.radiografias)
-        st.write(df_radiografias)
-
-        pdf_file = generar_pdf(df_radiografias, "Registro de Radiografías", "radiografias.pdf")
-        st.download_button(
-            label="📥 Descargar PDF",
-            data=open(pdf_file, 'rb').read(),
-            file_name="radiografias.pdf",
-            mime="application/pdf"
-        )
+        pregunta_usuario = st.text_input("Escribe tu pregunta aquí:", value=st.session_state.get('presupuesto_texto', ''))
+        if st.button("Enviar Pregunta"):
+            respuesta = obtener_respuesta(
+                pregunta_usuario, 
+                "",  # No se necesita texto preprocesado en este caso
+                st.session_state['modelo'], 
+                st.session_state['temperatura'],
+                st.session_state.get('assistant_id', 'asst_4ZYvBvf4IUVQPjnugSZGLdV2')
+            )
+            st.markdown(f"**Asistente**: {respuesta}")
+
+if __name__ == "__main__":
+    flujo_presupuestos()