102 lines
2.1 KiB
Python
102 lines
2.1 KiB
Python
import asyncio
|
|
import os
|
|
import re
|
|
import time
|
|
import traceback
|
|
|
|
from dotenv import load_dotenv
|
|
from edge_tts import Communicate
|
|
|
|
from agent import Agent
|
|
from util.logger import get_logger
|
|
|
|
load_dotenv()
|
|
|
|
logger = get_logger("NORA")
|
|
|
|
TTS_VOICE = "de-DE-AmalaNeural"
|
|
|
|
|
|
class NoraChat:
|
|
|
|
def __init__(self):
|
|
self.agent = Agent()
|
|
|
|
async def run(self):
|
|
|
|
print("\n" + "=" * 40)
|
|
print(" N.O.R.A START")
|
|
print("=" * 40 + "\n")
|
|
|
|
logger.info("Verbinde MCP...")
|
|
await self.agent.mcp.connect()
|
|
|
|
logger.info("Lade Tools...")
|
|
await self.agent.load_tools()
|
|
|
|
print("N.O.R.A ist bereit. (exit zum Beenden)\n")
|
|
|
|
try:
|
|
while True:
|
|
|
|
user = input("Du: ").strip()
|
|
|
|
if user.lower() in ["exit", "quit", "bye", "stop"]:
|
|
break
|
|
|
|
if not user:
|
|
continue
|
|
|
|
logger.info(f"USER: {user}")
|
|
|
|
response = await self.agent.run(user)
|
|
|
|
logger.info(f"NORA: {response}")
|
|
|
|
print(f"\nN.O.R.A: {response}\n")
|
|
|
|
await self.speak(response)
|
|
|
|
except KeyboardInterrupt:
|
|
pass
|
|
|
|
finally:
|
|
await self.agent.mcp.close()
|
|
logger.info("Shutdown abgeschlossen")
|
|
|
|
async def speak(self, text: str):
|
|
|
|
try:
|
|
|
|
# Nur die ersten zwei Sätze sprechen
|
|
sentences = re.split(r'(?<=[.!?])\s+', text.strip())
|
|
speech_text = " ".join(sentences[:2])
|
|
|
|
if not speech_text:
|
|
return
|
|
|
|
filename = f"response_{int(time.time())}.mp3"
|
|
|
|
tts = Communicate(
|
|
speech_text,
|
|
voice=TTS_VOICE
|
|
)
|
|
|
|
await tts.save(filename)
|
|
|
|
os.system(f"mpg123 -q {filename}")
|
|
|
|
os.remove(filename)
|
|
|
|
except Exception:
|
|
logger.error("TTS Fehler")
|
|
logger.error(traceback.format_exc())
|
|
|
|
|
|
async def main():
|
|
chat = NoraChat()
|
|
await chat.run()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main()) |