Update agent_functions.py

agent_functions.py

@@ -1,7 +1,15 @@
+
 # agent_functions.py
 import pandas as pd
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
+import openai
+import tempfile
+from google.cloud import texttospeech
+import time
+
+# Instancia el cliente de Text-to-Speech
+client = texttospeech.TextToSpeechClient()
 
 class PedidoTool:
     def __init__(self, menu_df):
@@ -45,6 +53,42 @@ def generar_pdf_orden(order_details, filepath='orden_compra.pdf'):
     c.save()
     return filepath
 
+def obtener_respuesta(pregunta, texto_preprocesado, modelo, temperatura=0.5):
+    try:
+        response = openai.ChatCompletion.create(
+            model=modelo,
+            messages=[
+                {"role": "system", "content": "Actua como Ana una asesora de ventas del restaurante Sazon Burguer, tienes un tono muy amable y cordial"},
+                {"role": "user", "content": f"{pregunta}\n\nContexto: {texto_preprocesado}"}
+            ],
+            temperature=temperatura
+        )
+        respuesta = response.choices[0].message['content'].strip()
+
+        # Configura la solicitud de síntesis de voz
+        input_text = texttospeech.SynthesisInput(text=respuesta)
+        voice = texttospeech.VoiceSelectionParams(
+            language_code="es-ES", ssml_gender=texttospeech.SsmlVoiceGender.MALE,
+        )
+        audio_config = texttospeech.AudioConfig(
+            audio_encoding=texttospeech.AudioEncoding.MP3
+        )
+
+        # Realiza la solicitud de síntesis de voz
+        response = client.synthesize_speech(
+            input=input_text, voice=voice, audio_config=audio_config
+        )
+
+        # Guarda el audio en un archivo temporal
+        audio_path = tempfile.mktemp(suffix=".mp3")
+        with open(audio_path, "wb") as audio_file:
+            audio_file.write(response.audio_content)
+
+        return respuesta, audio_path
+
+    except openai.OpenAIError as e:
+        return f"Error al comunicarse con OpenAI: {e}", None
+
 def tomar_pedido_agente(pedido, menu_df):
     pedido_tool = PedidoTool(menu_df)
     confirmados = pedido_tool.tomar_pedido(pedido)