Update utils/data_manager.py

utils/data_manager.py

@@ -1,4 +1,4 @@
-import os
+mport os
 import tempfile
 import openai
 from dotenv import load_dotenv
@@ -9,6 +9,7 @@ from nltk.corpus import stopwords
 from nltk.stem import SnowballStemmer
 import pandas as pd
 from fpdf import FPDF
+import streamlit as st
 import requests
 from google.cloud import texttospeech
 
@@ -23,12 +24,12 @@ os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "botidinamix-g.json"
 
 # Verifica que las claves API están configuradas
 if not openai_api_key:
-    raise ValueError("No API key provided for OpenAI. Please set your API key in the .env file.")
+    st.error("No API key provided for OpenAI. Please set your API key in the .env file.")
 else:
     openai.api_key = openai_api_key
 
 if not brevo_api_key:
-    raise ValueError("No API key provided for Brevo. Please set your API key in the .env file.")
+    st.error("No API key provided for Brevo. Please set your API key in the .env file.")
 
 def extraer_texto_pdf(archivo):
     texto = ""
@@ -42,7 +43,7 @@ def extraer_texto_pdf(archivo):
                 for page in range(len(reader.pages)):
                     texto += reader.pages[page].extract_text()
         except Exception as e:
-            raise ValueError(f"Error al extraer texto del PDF: {e}")
+            st.error(f"Error al extraer texto del PDF: {e}")
         finally:
             os.unlink(temp_file_path)
     return texto
@@ -88,10 +89,12 @@ def obtener_respuesta(pregunta, texto_preprocesado, modelo, temperatura=0.5, ass
         return respuesta
 
     except openai.OpenAIError as e:
-        raise ValueError(f"Error al comunicarse con OpenAI: {e}")
+        st.error(f"Error al comunicarse con OpenAI: {e}")
+        return "Lo siento, no puedo procesar tu solicitud en este momento."
 
     except Exception as e:
-        raise ValueError(f"Error al generar la respuesta y el audio: {e}")
+        st.error(f"Error al generar la respuesta y el audio: {e}")
+        return "Lo siento, ocurrió un error al procesar tu solicitud."
 
 def guardar_en_txt(nombre_archivo, datos):
     carpeta = "datos_guardados"
@@ -101,7 +104,7 @@ def guardar_en_txt(nombre_archivo, datos):
         with open(ruta_archivo, 'a', encoding='utf-8') as archivo:  # Append mode
             archivo.write(datos + "\n")
     except Exception as e:
-        raise ValueError(f"Error al guardar datos en el archivo: {e}")
+        st.error(f"Error al guardar datos en el archivo: {e}")
     return ruta_archivo
 
 def cargar_desde_txt(nombre_archivo):
@@ -112,9 +115,11 @@ def cargar_desde_txt(nombre_archivo):
             with open(ruta_archivo, 'r', encoding='utf-8') as archivo:
                 return archivo.read()
         else:
-            raise ValueError("Archivo no encontrado.")
+            st.warning("Archivo no encontrado.")
+            return ""
     except Exception as e:
-        raise ValueError(f"Error al cargar datos desde el archivo: {e}")
+        st.error(f"Error al cargar datos desde el archivo: {e}")
+        return ""
 
 def listar_archivos_txt():
     carpeta = "datos_guardados"
@@ -125,7 +130,8 @@ def listar_archivos_txt():
         archivos_ordenados = sorted(archivos, key=lambda x: os.path.getctime(os.path.join(carpeta, x)), reverse=True)
         return archivos_ordenados
     except Exception as e:
-        raise ValueError(f"Error al listar archivos: {e}")
+        st.error(f"Error al listar archivos: {e}")
+        return []
 
 def generar_pdf(dataframe, titulo, filename):
     pdf = FPDF()
@@ -142,7 +148,8 @@ def generar_pdf(dataframe, titulo, filename):
             pdf.output(tmp_file.name)
             return tmp_file.name
     except Exception as e:
-        raise ValueError(f"Error al generar PDF: {e}")
+        st.error(f"Error al generar PDF: {e}")
+        return None
 
 def enviar_correo(destinatario, asunto, contenido):
     url = "https://api.brevo.com/v3/smtp/email"
@@ -162,9 +169,9 @@ def enviar_correo(destinatario, asunto, contenido):
         if response.status_code == 201:
             st.success(f"Correo enviado a {destinatario}")
         else:
-            raise ValueError(f"Error al enviar el correo: {response.text}")
+            st.error(f"Error al enviar el correo: {response.text}")
     except Exception as e:
-        raise ValueError(f"Error al enviar el correo: {e}")
+        st.error(f"Error al enviar el correo: {e}")
 
 def enviar_whatsapp(numero, mensaje):
     url = "https://api.brevo.com/v3/whatsapp/send"
@@ -183,9 +190,9 @@ def enviar_whatsapp(numero, mensaje):
         if response.status_code == 201:
             st.success(f"Mensaje de WhatsApp enviado a {numero}")
         else:
-            raise ValueError(f"Error al enviar el mensaje de WhatsApp: {response.text}")
+            st.error(f"Error al enviar el mensaje de WhatsApp: {response.text}")
     except Exception as e:
-        raise ValueError(f"Error al enviar el mensaje de WhatsApp: {e}")
+        st.error(f"Error al enviar el mensaje de WhatsApp: {e}")
 
 def flujo_laboratorio():
     st.title("🦷 Gestión de Trabajos de Laboratorio")
@@ -346,7 +353,7 @@ def mostrar_datos_como_texto(datos):
 
 def flujo_presupuestos():
     st.title("💰 Asistente de Presupuestos")
-    st.markdown("Hola Dr. cuénteme en qué puedo ayudarle?")
+    st.markdown("Hola Dr. cuénteme en que puedo ayudarle?")
 
     lista_precios = {
         "Restauraciones en resina de una superficie": 75000,
@@ -374,64 +381,13 @@ 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)
-    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**")
-
-    # Transferir presupuesto al chat del asistente
-    if st.button("Transferir Presupuesto al Chat del Asistente"):
-        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 transferido al chat del asistente.")
-
-    # Mostrar chat del asistente
-    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']
-
-    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í:", 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_KngkX6sbRccg5a6fcnDHO06R')
-            )
-            st.markdown(f"**Asistente**: {respuesta}")
+    
+    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")
@@ -470,4 +426,4 @@ def flujo_radiografias():
             data=open(pdf_file, 'rb').read(),
             file_name="radiografias.pdf",
             mime="application/pdf"
-        )
+        )