Update main.py
This commit is contained in:
1 parent
4f9dc8edaa
commit
f19f516f25
1 file changed
+8
-15
@@ -2,8 +2,8 @@ import asyncio
|
|||||||
import threading
|
import threading
|
||||||
from voice_player import VoiceImagePlayer, edge_speak
|
from voice_player import VoiceImagePlayer, edge_speak
|
||||||
from speech_to_text import record_voice
|
from speech_to_text import record_voice
|
||||||
#from recognition import record_voice
|
|
||||||
from groq_ai import get_response
|
from groq_ai import get_response
|
||||||
|
from aircraft_module import handle_aircraft_command
|
||||||
|
|
||||||
async def get_voice_input():
|
async def get_voice_input():
|
||||||
return await asyncio.to_thread(record_voice)
|
return await asyncio.to_thread(record_voice)
|
||||||
@@ -15,26 +15,18 @@ async def ai_loop(player):
|
|||||||
if not user_text:
|
if not user_text:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
print("User:", user_text)
|
handled = handle_aircraft_command(user_text, player)
|
||||||
lowered = user_text.lower().strip()
|
if handled:
|
||||||
|
continue
|
||||||
if lowered in ["exit", "quit", "goodbye", "bye", "stop","mute"]:
|
|
||||||
edge_speak("Goodbye sir.", player)
|
|
||||||
player.root.after(300, player.root.destroy)
|
|
||||||
break
|
|
||||||
|
|
||||||
response = get_response(user_text)
|
response = get_response(user_text)
|
||||||
print("AI:", response)
|
print("AI:", response)
|
||||||
|
|
||||||
edge_speak(response, player)
|
edge_speak(response, player)
|
||||||
|
|
||||||
player.write_log(f"You: {user_text}")
|
player.write_log(f"You: {user_text}")
|
||||||
player.write_log(f"AI: {response}")
|
player.write_log(f"AI: {response}")
|
||||||
|
|
||||||
|
|
||||||
def main_loop():
|
def main_loop():
|
||||||
print("")
|
|
||||||
|
|
||||||
player = VoiceImagePlayer("face.png", size=(900, 900))
|
player = VoiceImagePlayer("face.png", size=(900, 900))
|
||||||
|
|
||||||
def thread_target():
|
def thread_target():
|
||||||
@@ -43,5 +35,6 @@ def main_loop():
|
|||||||
threading.Thread(target=thread_target, daemon=True).start()
|
threading.Thread(target=thread_target, daemon=True).start()
|
||||||
player.root.mainloop()
|
player.root.mainloop()
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main_loop()
|
main_loop()
|
||||||
Reference in new issue
Block a user