first commit

agi-scripts/voicebot.py

@@ -0,0 +1,445 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+import os
+import uuid
+import requests
+from asterisk.agi import AGI
+
+
+TMP_DIR = "/tmp/voicebot"
+
+STT_URL = "http://127.0.0.1:5001/transcribe"
+TTS_URL = "http://127.0.0.1:5002/synthesize"
+
+OLLAMA_URL = "http://127.0.0.1:11434/api/generate"
+OLLAMA_MODEL = "qwen3:14b"
+
+
+
+def log(agi, msg):
+    agi.verbose(
+        "VOICEBOT: " + str(msg),
+        1
+    )
+
+
+
+def tts_generate(agi, text, name):
+
+    try:
+
+        log(
+            agi,
+            "TTS request: " + text
+        )
+
+
+        r = requests.post(
+            TTS_URL,
+            json={
+                "text": text
+            },
+            timeout=60
+        )
+
+        r.raise_for_status()
+
+
+        path = TMP_DIR + "/" + name + ".wav"
+
+
+        with open(path, "wb") as f:
+            f.write(r.content)
+
+
+        log(
+            agi,
+            "TTS created " + path
+        )
+
+
+        return path
+
+
+    except Exception as e:
+
+        log(
+            agi,
+            "TTS ERROR " + str(e)
+        )
+
+        return None
+
+
+
+def play_file(agi, path):
+
+    if not path:
+        return
+
+
+    try:
+
+        file = path.replace(
+            ".wav",
+            ""
+        )
+
+
+        agi.appexec(
+            "Playback",
+            file
+        )
+
+
+    except Exception as e:
+
+        log(
+            agi,
+            "PLAY ERROR " + str(e)
+        )
+
+
+
+
+def record_audio(agi, callid):
+
+
+    filename = TMP_DIR + "/" + callid + "_record"
+
+
+    log(
+        agi,
+        "Recording user voice"
+    )
+
+
+    try:
+
+
+        result = agi.send_command(
+
+            f'RECORD FILE "{filename}" wav "#" 10000 0 s=5'
+
+        )
+
+
+        log(
+            agi,
+            "Record result: " + str(result)
+        )
+
+
+    except Exception as e:
+
+
+        log(
+            agi,
+            "Record ERROR " + str(e)
+        )
+
+        return None
+
+
+
+    wav = filename + ".wav"
+
+
+    if os.path.exists(wav):
+
+
+        size=os.path.getsize(wav)
+
+
+        log(
+            agi,
+            f"Recording saved {wav} size={size}"
+        )
+
+
+        return wav
+
+
+
+    log(
+        agi,
+        "Recording file missing"
+    )
+
+
+    return None
+
+
+
+
+
+def stt_process(agi,wav):
+
+
+    try:
+
+
+        log(
+            agi,
+            "Sending audio to STT"
+        )
+
+
+        with open(wav,"rb") as f:
+
+
+            r=requests.post(
+
+                STT_URL,
+
+                files={
+                    "audio":f
+                },
+
+                timeout=120
+
+            )
+
+
+        data=r.json()
+
+
+        text=data.get(
+            "text",
+            ""
+        )
+
+
+        log(
+            agi,
+            "STT text: " + text
+        )
+
+
+        return text
+
+
+
+    except Exception as e:
+
+
+        log(
+            agi,
+            "STT ERROR " + str(e)
+        )
+
+
+        return ""
+
+
+
+
+
+def ask_ai(agi,text):
+
+
+    try:
+
+
+        log(
+            agi,
+            "AI request: " + text
+        )
+
+
+        prompt = """
+تو دستیار تلفنی هستی.
+کوتاه و طبیعی جواب بده.
+حداکثر دو جمله.
+
+کاربر:
+""" + text
+
+
+
+        r=requests.post(
+
+            OLLAMA_URL,
+
+            json={
+
+                "model":OLLAMA_MODEL,
+
+                "prompt":prompt,
+
+                "stream":False,
+
+                "options":{
+                    "num_predict":80,
+                    "temperature":0.3
+                }
+
+            },
+
+            timeout=180
+
+        )
+
+
+        data=r.json()
+
+
+        answer=data.get(
+            "response",
+            ""
+        )
+
+
+        log(
+            agi,
+            "AI response: " + answer
+        )
+
+
+        return answer
+
+
+
+    except Exception as e:
+
+
+        log(
+            agi,
+            "AI ERROR " + str(e)
+        )
+
+
+        return "ارتباط با هوش مصنوعی برقرار نشد."
+
+
+
+
+
+def main():
+
+
+    os.makedirs(
+        TMP_DIR,
+        exist_ok=True
+    )
+
+
+    agi=AGI()
+
+
+    callid=str(uuid.uuid4())[:8]
+
+
+    log(
+        agi,
+        "Call started " + callid
+    )
+
+
+    try:
+
+
+        agi.answer()
+
+
+
+        welcome=tts_generate(
+
+            agi,
+
+            "سلام، من دستیار صوتی هوشمند هستم. بفرمایید.",
+
+            callid+"_welcome"
+
+        )
+
+
+        play_file(
+            agi,
+            welcome
+        )
+
+
+
+        wav=record_audio(
+            agi,
+            callid
+        )
+
+
+        if not wav:
+
+            log(
+                agi,
+                "No recording"
+            )
+
+            return
+
+
+
+        text=stt_process(
+            agi,
+            wav
+        )
+
+
+
+        if text:
+
+            answer=ask_ai(
+                agi,
+                text
+            )
+
+        else:
+
+            answer="صدای شما را دریافت نکردم."
+
+
+
+        reply=tts_generate(
+
+            agi,
+
+            answer,
+
+            callid+"_reply"
+
+        )
+
+
+        play_file(
+            agi,
+            reply
+        )
+
+
+        log(
+            agi,
+            "Call completed"
+        )
+
+
+
+    except Exception as e:
+
+
+        log(
+            agi,
+            "MAIN ERROR " + str(e)
+        )
+
+
+    finally:
+
+        agi.hangup()
+
+
+
+
+
+if __name__ == "__main__":
+
+    main()