Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import streamlit as st
 import openai
 from dotenv import load_dotenv
@@ -10,9 +11,9 @@ from nltk.stem import SnowballStemmer
 import PyPDF2
 import time
 from google.cloud import texttospeech
-from streamlit_webrtc import webrtc_streamer, WebRtcMode
+from streamlit_webrtc import webrtc_streamer, WebRtcMode, AudioProcessorBase
 from Historial.historial_chat import cargar_historial, guardar_historial
-from agent_functions import menu_df, tomar_pedido_agente, procesar_orden_agente, generar_pdf_orden, obtener_respuesta
+from agent_functions import menu_df, tomar_pedido_agente, procesar_orden_agente, generar_pdf_orden, obtener_respuesta, generar_mensaje_automatico
 
 # Configuración de NLTK
 nltk.download('punkt')
@@ -47,6 +48,15 @@ def preprocesar_texto(texto):
     tokens = [stemmer.stem(word) for word in tokens]
     return " ".join(tokens)
 
+# Clase para procesar audio
+class AudioProcessor(AudioProcessorBase):
+    def __init__(self):
+        self.audio_bytes = b''
+
+    def recv(self, frame):
+        self.audio_bytes += frame.to_ndarray().tobytes()
+        return frame
+
 # Main App
 def main():
     # --- Diseño general ---
@@ -143,10 +153,22 @@ def main():
                 st.session_state.run_webrtc = True
             if st.session_state.get("run_webrtc", False):
                 webrtc_ctx = webrtc_streamer(
-                    key="example", mode=WebRtcMode.SENDONLY,
-                    in_audio=True, on_audio_frame=on_audio,
+                    key="example",
+                    mode=WebRtcMode.SENDONLY,
+                    audio_receiver_size=256,
+                    rtc_configuration={
+                        "iceServers": [{"urls": ["stun:stun.l.google.com:19302"]}]
+                    },
+                    media_stream_constraints={"audio": True},
+                    audio_processor_factory=AudioProcessor,
                 )
 
+                if webrtc_ctx.audio_receiver:
+                    audio_frames = webrtc_ctx.audio_receiver.get_frames(timeout=1)
+                    for audio_frame in audio_frames:
+                        audio_bytes = audio_frame.to_ndarray().tobytes()
+                        on_audio(audio_bytes)
+
         st.markdown("---")
         st.subheader("📄 Subir PDF")
         st.info("Sube un archivo PDF y el asistente responderá en función de su contenido.")
@@ -242,6 +264,12 @@ def main():
                 pdf_path = generar_pdf_orden(order_details)
                 st.markdown(f"[Descargar PDF de la Orden]({pdf_path})", unsafe_allow_html=True)
 
+                # Generar mensaje automático
+                mensaje_automatico = generar_mensaje_automatico(confirmados)
+                st.session_state.mensajes_agente.append({"role": "assistant", "content": mensaje_automatico, "timestamp": time.time()})
+                with st.chat_message("assistant"):
+                    st.markdown(mensaje_automatico)
+
         # --- Chat para Agentes ---
         st.subheader("Chat con Agentes")
         if 'mensajes_agente' not in st.session_state: