Upload app (5).py

app (5).py

@@ -0,0 +1,256 @@
+import streamlit as st
+import os
+import openai
+from utils.data_manager import (
+    extraer_texto_pdf, 
+    preprocesar_texto, 
+    obtener_respuesta,
+    flujo_laboratorio,
+    flujo_insumos,
+    buscar_datos_guardados,
+    generar_notificaciones_pendientes,
+    flujo_presupuestos,
+    flujo_radiografias
+)
+import tempfile
+from dotenv import load_dotenv
+from flask import Flask, request, jsonify
+import threading
+
+# Cargar las claves API desde el archivo .env
+load_dotenv()
+openai_api_key = os.getenv("OPENAI_API_KEY")
+
+# Verifica que las claves API están configuradas
+if not openai_api_key:
+    st.error("No API key provided for OpenAI. Please set your API key in the .env file.")
+else:
+    openai.api_key = openai_api_key
+
+# Crear la aplicación Flask
+app = Flask(__name__)
+
+@app.route('/process_audio', methods=['POST'])
+def process_audio():
+    data = request.json
+    transcription = data['transcription']
+    return jsonify({'transcription': transcription})
+
+def run_flask_app():
+    app.run(port=5000, debug=True)
+
+# Ejecutar Flask en un hilo separado
+flask_thread = threading.Thread(target=run_flask_app)
+flask_thread.start()
+
+def inicializar_estado():
+    """Inicializa el estado de la sesión si no está ya inicializado."""
+    if 'modelo' not in st.session_state:
+        st.session_state['modelo'] = "gpt-3.5-turbo-16k"
+    if 'temperatura' not in st.session_state:
+        st.session_state['temperatura'] = 0.5
+    if 'mensajes_chat' not in st.session_state:
+        st.session_state['mensajes_chat'] = []
+    if 'transcripcion_voz' not in st.session_state:
+        st.session_state['transcripcion_voz'] = ""
+
+def barra_lateral():
+    ruta_logo = os.path.join("assets", "Logo Omardent.png")
+    if os.path.exists(ruta_logo):
+        st.sidebar.image(ruta_logo, use_column_width=True)
+    else:
+        st.sidebar.warning(f"Error: No se pudo encontrar la imagen en la ruta: {ruta_logo}")
+
+    st.sidebar.title("🤖 Galatea OMARDENT")
+    st.sidebar.markdown("---")
+    st.sidebar.subheader("🧠 Configuración del Modelo")
+    st.session_state['modelo'] = st.sidebar.selectbox(
+        "Selecciona el modelo:",
+        ["gpt-3.5-turbo", "gpt-4", "gpt-4-32k", "gpt-4o"],
+        index=0,
+        key='modelo_selectbox',  # Clave única
+        help="Elige el modelo de lenguaje de OpenAI que prefieras."
+    )
+    st.sidebar.markdown("---")
+    st.session_state['temperatura'] = st.sidebar.slider(
+        "🌡️ Temperatura",
+        min_value=0.0, max_value=1.0,
+        value=st.session_state['temperatura'],
+        step=0.1,
+        key='temperatura_slider'  # Clave única
+    )
+    assistant_id = st.sidebar.text_input("Assistant ID", key="assistant_id", help="Introduce el Assistant ID del playground de OpenAI")
+    if assistant_id:
+        st.session_state['assistant_id'] = assistant_id
+
+def mostrar_mensajes_chat():
+    for mensaje in st.session_state['mensajes_chat']:
+        with st.chat_message(mensaje["role"]):
+            st.markdown(mensaje["content"])
+
+def manejar_pregunta_usuario(pregunta_usuario, archivo_pdf):
+    st.session_state['mensajes_chat'].append({"role": "user", "content": pregunta_usuario})
+    with st.chat_message("user"):
+        st.markdown(pregunta_usuario)
+
+    texto_preprocesado = ""
+    if archivo_pdf:
+        texto_pdf = extraer_texto_pdf(archivo_pdf)
+        texto_preprocesado = preprocesar_texto(texto_pdf)
+
+    respuesta = obtener_respuesta(
+        pregunta_usuario, 
+        texto_preprocesado, 
+        st.session_state['modelo'], 
+        st.session_state['temperatura'],
+        st.session_state.get('assistant_id', '')
+    )
+    st.session_state['mensajes_chat'].append({"role": "assistant", "content": respuesta})
+    with st.chat_message("assistant"):
+        st.markdown(respuesta)
+
+def capturar_voz():
+    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="startRecording()">
+            <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'/>
+            Capturar Voz
+        </button>
+        <script>
+            function startRecording() {
+                const recognition = new (window.SpeechRecognition || window.webkitSpeechRecognition)();
+                recognition.lang = 'es-ES';
+                recognition.interimResults = false;
+                recognition.maxAlternatives = 1;
+
+                recognition.start();
+
+                recognition.onresult = (event) => {
+                    const lastResult = event.results.length - 1;
+                    const text = event.results[lastResult][0].transcript;
+                    const customEvent = new CustomEvent('audioTranscription', { detail: text });
+                    document.dispatchEvent(customEvent);
+                };
+
+                recognition.onspeechend = () => {
+                    recognition.stop();
+                };
+
+                recognition.onerror = (event) => {
+                    console.error(event.error);
+                };
+            }
+
+            document.addEventListener('audioTranscription', (event) => {
+                fetch('/process_audio', {
+                    method: 'POST',
+                    headers: {
+                        'Content-Type': 'application/json'
+                    },
+                    body: JSON.stringify({ transcription: event.detail })
+                }).then(response => response.json())
+                  .then(data => {
+                      document.querySelector("input[name='unique_chat_input_key']").value = data.transcription;
+                  });
+            });
+        </script>
+        """,
+        unsafe_allow_html=True
+    )
+
+def mostrar_paginas():
+    st.sidebar.title("Navegación")
+    page = st.sidebar.radio("Ir a", ["Página Principal", "Gestión de Trabajos", "Gestión de Insumos", "Registro de Radiografías", "Buscar Datos", "Notificaciones", "Recomendaciones", "Asistente de Presupuestos", "Comunicación"])
+
+    if page == "Página Principal":
+        from pages import home
+        home.show()
+    elif page == "Gestión de Trabajos":
+        from pages import trabajos
+        trabajos.show()
+    elif page == "Gestión de Insumos":
+        from pages import insumos
+        insumos.show()
+    elif page == "Registro de Radiografías":
+        from pages import radiografias
+        radiografias.show()
+    elif page == "Buscar Datos":
+        from pages import buscar_datos
+        buscar_datos.show()
+    elif page == "Notificaciones":
+        from pages import notificaciones
+        notificaciones.show()
+    elif page == "Recomendaciones":
+        from pages import recomendaciones
+        recomendaciones.show()
+    elif page == "Asistente de Presupuestos":
+        from pages import presupuesto
+        presupuesto.show()
+    elif page == "Comunicación":
+        from pages import comunicacion
+        comunicacion.show()
+
+def main():
+    inicializar_estado()
+    barra_lateral()
+    mostrar_paginas()
+
+    st.title("VIRTUAL OMARDENT AI-BOTIDINAMIX")
+    st.markdown(
+        f"""
+        <style>
+        #video-container {{
+            position: relative;
+            width: 100%;
+            padding-bottom: 56.25%;
+            background-color: lightblue;
+            overflow: hidden;
+        }}
+        #background-video {{
+            position: absolute;
+            top: 0;
+            left: 0;
+            width: 100%;
+            height: 100%;
+        }}
+        </style>
+        <div id="video-container">
+            <video id="background-video" autoplay loop muted playsinline>
+                <source src="https://cdn.leonardo.ai/users/645c3d5c-ca1b-4ce8-aefa-a091494e0d09/generations/0c4f0fe7-5937-4644-b984-bdbd95018990/0c4f0fe7-5937-4644-b984-bdbd95018990.mp4" type="video/mp4">
+            </video>
+        </div>
+        """,
+        unsafe_allow_html=True
+    )
+
+    archivo_pdf = st.file_uploader("📂 Cargar PDF", type='pdf', key='chat_pdf')
+
+    col1, col2 = st.columns([3, 1])
+    with col1:
+        pregunta_usuario = st.text_input("Pregunta:", key='unique_chat_input_key', value=st.session_state['transcripcion_voz'])
+    with col2:
+        capturar_voz()
+    
+    if pregunta_usuario:
+        manejar_pregunta_usuario(pregunta_usuario, archivo_pdf)
+
+if __name__ == "__main__":
+    main()