136 lines
2.3 KiB
Python
136 lines
2.3 KiB
Python
import asyncio
|
|
import os
|
|
import time
|
|
import traceback
|
|
|
|
from edge_tts import Communicate
|
|
|
|
from agent import Agent
|
|
from util.logger import get_logger
|
|
|
|
|
|
logger = get_logger("NORA CHAT")
|
|
|
|
|
|
# =========================
|
|
# CONFIG
|
|
# =========================
|
|
|
|
TTS_VOICE = "de-DE-AmalaNeural"
|
|
|
|
|
|
# =========================
|
|
# CHAT ASSISTANT
|
|
# =========================
|
|
|
|
class NoraChat:
|
|
|
|
def __init__(self):
|
|
logger.info("Initialisiere N.O.R.A Chat...")
|
|
|
|
self.agent = Agent()
|
|
self.running = True
|
|
|
|
logger.info("N.O.R.A Chat bereit ✔")
|
|
|
|
|
|
# =========================
|
|
# MAIN LOOP
|
|
# =========================
|
|
|
|
async def run(self):
|
|
|
|
logger.info("Starte Chat-Modus")
|
|
|
|
while self.running:
|
|
|
|
try:
|
|
user_input = input("\nDu: ")
|
|
|
|
if user_input.lower() in [
|
|
"exit",
|
|
"quit",
|
|
"stop",
|
|
"beenden"
|
|
]:
|
|
logger.info("Beende NORA")
|
|
break
|
|
|
|
|
|
if not user_input.strip():
|
|
continue
|
|
|
|
|
|
logger.info(f"USER: {user_input}")
|
|
|
|
|
|
# Agent Anfrage
|
|
response = await self.agent.run(user_input)
|
|
|
|
|
|
logger.info(f"NORA: {response}")
|
|
|
|
print(f"\nNORA: {response}")
|
|
|
|
|
|
# Sprach-Ausgabe
|
|
await self.speak(response)
|
|
|
|
|
|
except KeyboardInterrupt:
|
|
break
|
|
|
|
except Exception:
|
|
logger.error(traceback.format_exc())
|
|
|
|
|
|
# =========================
|
|
# TTS
|
|
# =========================
|
|
|
|
async def speak(self, text):
|
|
|
|
try:
|
|
|
|
filename = f"response_{int(time.time())}.mp3"
|
|
|
|
|
|
tts = Communicate(
|
|
text,
|
|
voice=TTS_VOICE
|
|
)
|
|
|
|
|
|
await tts.save(filename)
|
|
|
|
|
|
# Ausgabe über Standard-Audio
|
|
os.system(
|
|
f"mpg123 -q {filename}"
|
|
)
|
|
|
|
|
|
os.remove(filename)
|
|
|
|
|
|
except Exception:
|
|
|
|
logger.error("TTS Fehler")
|
|
logger.error(traceback.format_exc())
|
|
|
|
|
|
|
|
# =========================
|
|
# MAIN
|
|
# =========================
|
|
|
|
async def main():
|
|
|
|
nora = NoraChat()
|
|
|
|
await nora.run()
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main()) |