Update utils/data_manager.py

utils/data_manager.py

@@ -1,6 +1,5 @@
 import os
 import tempfile
-import openai
 import PyPDF2
 import nltk
 from nltk.tokenize import word_tokenize
@@ -8,29 +7,13 @@ 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
-from dotenv import load_dotenv
-import streamlit as st
 
 nltk.download('punkt', quiet=True)
 nltk.download('stopwords', quiet=True)
 
-# Cargar las claves API desde el archivo .env
-load_dotenv()
-openai_api_key = os.getenv("OPENAI_API_KEY")
-brevo_api_key = os.getenv("BREVO_API_KEY")
-os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "service_account.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.")
-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.")
-
 def extraer_texto_pdf(archivo):
     texto = ""
     if archivo:
@@ -43,7 +26,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
@@ -84,23 +67,27 @@ def obtener_respuesta(pregunta, texto_preprocesado, modelo, temperatura=0.5, ass
             input=input_text, voice=voice, audio_config=audio_config
         )
 
+        # Reproduce el audio en Streamlit
+        st.audio(response.audio_content, format="audio/mp3")
         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"
-    os.makedirs(carpeta, exist_ok=True)
+    os.makedirs(carpeta, existo_ok=True)
     ruta_archivo = os.path.join(carpeta, nombre_archivo)
     try:
         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):
@@ -111,9 +98,11 @@ def cargar_desde_txt(nombre_archivo):
             with open(ruta_archivo, 'r', encoding='utf-8') as archivo:
                 return archivo.read()
         else:
+            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"
@@ -124,7 +113,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()
@@ -141,7 +131,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"
@@ -159,11 +150,11 @@ def enviar_correo(destinatario, asunto, contenido):
     try:
         response = requests.post(url, json=payload, headers=headers)
         if response.status_code == 201:
-            return f"Correo enviado a {destinatario}"
+            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"
@@ -180,26 +171,11 @@ def enviar_whatsapp(numero, mensaje):
     try:
         response = requests.post(url, json=payload, headers=headers)
         if response.status_code == 201:
-            return f"Mensaje de WhatsApp enviado a {numero}"
+            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}")
-
-def mostrar_datos_como_texto(datos):
-    texto = ""
-    if isinstance(datos, dict):
-        for key, value in datos.items():
-            texto += f"{key}: {value}\n"
-    elif isinstance(datos, list):
-        for item in datos:
-            if isinstance(item, dict):
-                for key, value in item.items():
-                    texto += f"{key}: {value}\n"
-                texto += "\n"
-            else:
-                texto += f"{item}\n"
-    return texto
+        st.error(f"Error al enviar el mensaje de WhatsApp: {e}")
 
 def flujo_laboratorio():
     st.title("🦷 Gestión de Trabajos de Laboratorio")
@@ -286,6 +262,78 @@ def flujo_insumos():
             mime="application/pdf"
         )
 
+def buscar_datos_guardados():
+    st.title("🔍 Buscar Datos Guardados")
+
+    carpeta = "datos_guardados"
+    if not os.path.exists(carpeta):
+        st.info("No se encontraron archivos de datos guardados.")
+        return
+
+    archivos = listar_archivos_txt()
+
+    if archivos:
+        archivo_seleccionado = st.selectbox("Selecciona un archivo para ver:", archivos)
+        
+        if archivo_seleccionado:
+            datos = cargar_desde_txt(os.path.join(carpeta, archivo_seleccionado))
+            if datos:
+                st.write(f"### Datos del archivo {archivo_seleccionado}")
+                st.text_area("Datos", datos, height=300)
+
+                # Link to download the file
+                try:
+                    with open(os.path.join(carpeta, archivo_seleccionado), 'rb') as file:
+                        st.download_button(
+                            label="📥 Descargar Archivo TXT",
+                            data=file,
+                            file_name=archivo_seleccionado,
+                            mime="text/plain"
+                        )
+                except Exception as e:
+                    st.error(f"Error al preparar la descarga: {e}")
+
+                # Enviar el archivo seleccionado por correo
+                if st.button("Enviar por correo"):
+                    contenido = f"Datos del archivo {archivo_seleccionado}:\n\n{datos}"
+                    enviar_correo("josedcape@gmail.com", f"Datos del archivo {archivo_seleccionado}", contenido)
+
+                # Enviar el archivo seleccionado por WhatsApp
+                if st.button("Enviar por WhatsApp"):
+                    mensaje = f"Datos del archivo {archivo_seleccionado}:\n\n{datos}"
+                    enviar_whatsapp("3114329322", mensaje)
+
+            else:
+                st.warning(f"No se encontraron datos en el archivo {archivo_seleccionado}")
+    else:
+        st.info("No se encontraron archivos de datos guardados.")
+
+def generar_notificaciones_pendientes():
+    if 'laboratorio' not in st.session_state or not st.session_state.laboratorio:
+        st.info("No hay trabajos pendientes.")
+        return
+    
+    pendientes = [trabajo for trabajo in st.session_state.laboratorio if trabajo["estado"] == "pendiente"]
+    if pendientes:
+        st.write("### Notificaciones de Trabajos Pendientes")
+        for trabajo in pendientes:
+            st.info(f"Pendiente: {trabajo['tipo_trabajo']} - {trabajo['numero_orden']} para {trabajo['doctor']}. Enviado a {trabajo['laboratorio']} el {trabajo['fecha_envio']}.")
+
+def mostrar_datos_como_texto(datos):
+    texto = ""
+    if isinstance(datos, dict):
+        for key, value in datos.items():
+            texto += f"{key}: {value}\n"
+    elif isinstance(datos, list):
+        for item in datos:
+            if isinstance(item, dict):
+                for key, value in item.items():
+                    texto += f"{key}: {value}\n"
+                texto += "\n"
+            else:
+                texto += f"{item}\n"
+    return texto
+
 def flujo_presupuestos():
     st.title("💰 Asistente de Presupuestos")
     st.markdown("Hola Dr. cuénteme en qué puedo ayudarle?")