nora_chat.py gelöscht
This commit is contained in:
-136
@@ -1,136 +0,0 @@
|
|||||||
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())
|
|
||||||
Reference in New Issue
Block a user