parent
ef7917befc
commit
b49542b2a3
@ -1,173 +0,0 @@
|
||||
#!/usr/bin/env python3
|
||||
import speech_recognition as sr
|
||||
import re
|
||||
import tts
|
||||
import subprocess
|
||||
import os
|
||||
from asterisk.agi import AGI
|
||||
import anthropic
|
||||
import json
|
||||
|
||||
agi = AGI()
|
||||
|
||||
class CallHandler:
|
||||
def __init__(self):
|
||||
self.audio_path = "/var/lib/asterisk/sounds/grabacion_usuario.wav"
|
||||
self.tab_id = self.get_tab_id()
|
||||
|
||||
def get_tab_id(self):
|
||||
# Obtener el tab_id desde una variable de entorno AGI
|
||||
# Asegúrate de pasar esta variable desde el Dialplan
|
||||
tab_id = agi.env.get('TAB_ID', None)
|
||||
if not tab_id:
|
||||
agi.verbose("TAB_ID no fue proporcionado. Utilizando 'default_tab_id'.", 1)
|
||||
return 'default_tab_id'
|
||||
return tab_id
|
||||
|
||||
def grabar_llamada(self):
|
||||
callerid = agi.env.get('agi_callerid', '')
|
||||
uniqueid = agi.env.get('agi_uniqueid', '')
|
||||
|
||||
# Registrar la información de la llamada
|
||||
agi.verbose(f"CallerID: {callerid}", 1)
|
||||
agi.verbose(f"UniqueID: {uniqueid}", 1)
|
||||
|
||||
# Iniciar la grabación
|
||||
agi.verbose("Iniciando grabación...", 1)
|
||||
try:
|
||||
result = agi.record_file(
|
||||
filename='/var/lib/asterisk/sounds/grabacion_usuario',
|
||||
format='wav',
|
||||
timeout=4000, # Puedes ajustar el tiempo de espera según tus necesidades
|
||||
offset=0,
|
||||
beep=True,
|
||||
silence=0,
|
||||
escape_digits='#'
|
||||
)
|
||||
except Exception as e:
|
||||
agi.verbose(f"Error al grabar: {e}", 1)
|
||||
else:
|
||||
agi.verbose(f"Grabación completada. Resultado: {result}", 1)
|
||||
self.detectar_audio()
|
||||
|
||||
def detectar_audio(self):
|
||||
recognizer = sr.Recognizer()
|
||||
|
||||
with sr.AudioFile(self.audio_path) as source:
|
||||
audio_data = recognizer.record(source)
|
||||
try:
|
||||
text = recognizer.recognize_google(audio_data, language="es-ES")
|
||||
agi.verbose(f"La pregunta es: {text}", 1)
|
||||
|
||||
# Llamar a la función consultar_claude con el texto reconocido
|
||||
respuesta = self.consultar_claude(text)
|
||||
ext = self.determine_extension(respuesta)
|
||||
|
||||
cleaned_text = re.sub(r'^(?:\*+|\d+)', '', respuesta).strip()
|
||||
cleaned_text = cleaned_text.replace('%', '')
|
||||
|
||||
# Reproducir la respuesta generada por Claude
|
||||
try:
|
||||
tts.convert_text_to_speech(cleaned_text)
|
||||
agi.stream_file('resp_claude')
|
||||
except Exception as e:
|
||||
agi.verbose(f"Error al convertir texto a voz: {e}", 1)
|
||||
|
||||
if ext is not None and ext != '':
|
||||
try:
|
||||
resultado_dial = agi.execute("EXEC Dial", f"Local/{str(ext)}@from-internal,20")
|
||||
agi.verbose(f"Resultado del Dial: {resultado_dial}", 1)
|
||||
except Exception as e:
|
||||
agi.verbose(f"Error ejecutando Dial en AGI: {e}", 1)
|
||||
else:
|
||||
agi.verbose("La variable ext no tiene un valor válido.", 1)
|
||||
agi.stream_file('algomas')
|
||||
# Volver a grabar si es necesario
|
||||
self.grabar_llamada()
|
||||
except sr.RequestError as e:
|
||||
agi.verbose(f"Error al conectarse con el servicio de reconocimiento de voz: {e}", 1)
|
||||
except sr.UnknownValueError:
|
||||
agi.verbose("No se pudo entender el audio", 1)
|
||||
agi.stream_file('no_entendi')
|
||||
self.grabar_llamada()
|
||||
|
||||
def determine_extension(self, input_str):
|
||||
# Definir la ruta al archivo de condiciones basado en tab_id
|
||||
conditions_file = f"/home/Flow_IA/IVR/tab_{self.tab_id}_conditions.json"
|
||||
|
||||
if not os.path.exists(conditions_file):
|
||||
agi.verbose(f"No se encontró el archivo de condiciones para la pestaña {self.tab_id}", 1)
|
||||
return None
|
||||
|
||||
try:
|
||||
with open(conditions_file, 'r', encoding='utf-8') as f:
|
||||
conditions_data = json.load(f)
|
||||
|
||||
# Iterar sobre todas las condiciones de node3
|
||||
for node_cond in conditions_data.get('node3_conditions', []):
|
||||
for cond in node_cond.get('conditions', []):
|
||||
condition_str = cond.get('condition_string', '').strip().lower()
|
||||
target_id = cond.get('target_node_id', '').strip()
|
||||
if input_str.lower() == condition_str:
|
||||
agi.verbose(f"Condición encontrada: {condition_str} -> Extensión: {target_id}", 1)
|
||||
return target_id
|
||||
|
||||
except Exception as e:
|
||||
agi.verbose(f"Error al leer o procesar el archivo de condiciones: {e}", 1)
|
||||
|
||||
# Si no se encuentra ninguna condición que coincida
|
||||
agi.verbose("No se encontró una condición que coincida con la entrada del usuario.", 1)
|
||||
return None
|
||||
|
||||
def consultar_claude(self, text):
|
||||
# Leer extensiones (este comando parece específico para tu configuración)
|
||||
command = """grep -E '^\[|^callerid=' /etc/asterisk/pjsip.endpoint.conf | awk '{
|
||||
if ($0 ~ /^\[/) {
|
||||
gsub(/[\[\]]/, "", $1);
|
||||
ext = $1;
|
||||
} else if ($0 ~ /^callerid=/) {
|
||||
split($0, arr, "=");
|
||||
callerid = arr[2];
|
||||
printf "Extension: " ext ", CallerID: " callerid "; ";
|
||||
}
|
||||
}'"""
|
||||
|
||||
result = subprocess.run(command, shell=True, capture_output=True, text=True)
|
||||
output = result.stdout.strip()
|
||||
|
||||
# Crear el cliente Anthropic pasando la API key
|
||||
client = anthropic.Anthropic(api_key='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')
|
||||
|
||||
message = client.messages.create(
|
||||
model="claude-3-5-haiku-20241022",
|
||||
#model = "claude-3-5-sonnet-20241022",
|
||||
max_tokens=200,
|
||||
temperature=0,
|
||||
system=(
|
||||
"Vas a ser un agente de telefonía con las siguientes condiciones: "
|
||||
f'"{output}"'
|
||||
),
|
||||
messages=[
|
||||
{
|
||||
"role": "user",
|
||||
"content": [
|
||||
{
|
||||
"type": "text",
|
||||
"text": text
|
||||
}
|
||||
]
|
||||
}
|
||||
]
|
||||
)
|
||||
|
||||
# Asumiendo que `message.content` es una lista de mensajes
|
||||
respuesta_claude = "".join([msg.get('text', '') for msg in message.content])
|
||||
respuesta_claude = respuesta_claude.replace('%', '')
|
||||
agi.verbose(f"Respuesta de Claude: {respuesta_claude}", 1)
|
||||
return respuesta_claude
|
||||
|
||||
if __name__ == "__main__":
|
||||
agi.answer()
|
||||
agi.stream_file('saludo_claude')
|
||||
call_handler = CallHandler()
|
||||
call_handler.grabar_llamada()
|
@ -1,21 +0,0 @@
|
||||
import socket
|
||||
|
||||
def obtener_ip_privada():
|
||||
try:
|
||||
# Crear un socket UDP temporal
|
||||
with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
|
||||
# Conectarse a un host arbitrario en Internet
|
||||
s.connect(("8.8.8.8", 80)) # Google Public DNS
|
||||
# Obtener la dirección IP del socket
|
||||
ip_privada = s.getsockname()[0]
|
||||
return ip_privada
|
||||
except Exception as e:
|
||||
print(f"Error al obtener la IP privada: {e}")
|
||||
return None
|
||||
|
||||
# Llamar a la función
|
||||
mi_ip_privada = obtener_ip_privada()
|
||||
if mi_ip_privada:
|
||||
print(f"Tu IP privada es: {mi_ip_privada}")
|
||||
else:
|
||||
print("No se pudo determinar la IP privada.")
|
Loading…
Reference in new issue