try improved life management

app.py

@@ -15,6 +15,7 @@ import torch
 from sample_utils.turn import get_ice_servers
 import json
 from typing import List
+import subprocess
 
 st.set_page_config(layout="wide")
 
@@ -28,8 +29,7 @@ webrtc_ctx = None
 
 # Initialize Ray
 import ray
-if not ray.is_initialized():
-    import subprocess
+def init_ray():
     try:
         subprocess.check_output(["ray", "start", "--head"])
     except Exception as e:
@@ -40,12 +40,11 @@ if not ray.is_initialized():
         ray.init(ray_address, namespace="project_charles")
     else:
         ray.init(namespace="project_charles")
-    import platform
-    print (f"platform:{platform.system()}")
-    if platform.system() == 'Linux':
-        charles_actor_proc = subprocess.Popen(["python3", "charles_actor.py"])
 
 
+if not ray.is_initialized():
+    init_ray()
+
 async def main():
     st.title("Project Charles")
 
@@ -60,9 +59,13 @@ async def main():
         charles_actor_debug_output = st.empty()
         environment_state_ouput = st.empty()
         with nested_col3:
-            if st.button('Reboot Server'):
-                st.write('Rebooting...')
-                subprocess.run(["sudo", "reboot"])
+            if st.button('Reboot Charles'):
+                st.write('Killing RAY...')
+                subprocess.check_output(["ray", "start", "--head"])
+                st.write('Restarting RAY...')
+                init_ray()
+                charles_actor = None
+                st.write('Reboot Charles')
 
     with col2:
         if "streamlit_av_queue" not in st.session_state:
@@ -103,7 +106,7 @@ async def main():
             if "streamlit_av_queue" in st.session_state:
                 st.session_state.streamlit_av_queue.set_looking_listening(looking, listening)
             if not webrtc_ctx.state.playing:
-                system_one_audio_status.write("Stopped.")
+                system_one_audio_status.write("Camera has stopped.")
                 await asyncio.sleep(0.1)
                 continue
             if charles_actor is None:
@@ -111,14 +114,23 @@ async def main():
                     charles_actor = ray.get_actor("CharlesActor")
                     system_one_audio_status.write("Charles is here.")
                 except ValueError as e:
-                    system_one_audio_status.write("Charles is sleeping.")
-                    pass
+                    charles_actor_proc = subprocess.Popen(["python3", "charles_actor.py"])
+                    system_one_audio_status.write("Creating an instance of Charles...")
+                    # import platform
+                    # print (f"platform:{platform.system()}")
+                    # if platform.system() == 'Linux':
+                    #     charles_actor_proc = subprocess.Popen(["python3", "charles_actor.py"])
+                    #     system_one_audio_status.write("Creating Charles.")
+                    # else:
+                    #     system_one_audio_status.write("Charles is sleeping.")
             if charles_actor is not None: 
                 try:
                     # new_environment_state = await charles_actor.get_environment_state.remote()
                     # environment_state_ouput.markdown(f"{new_environment_state}")
                     charles_debug_str = await charles_actor.get_charles_actor_debug_output.remote()
                     charles_actor_debug_output.markdown(charles_debug_str)
+                    state = await charles_actor.get_state.remote()
+                    system_one_audio_status.write(state)
                 except Exception as e:
                     # assume we disconnected
                     charles_actor = None

charles_actor.py

@@ -29,17 +29,20 @@ class CharlesActor:
     async def _initalize_resources(self):
         # Initialize resources
         print("000 - create StreamlitAVQueue")
+        self._state = "000 - creating StreamlitAVQueue"
         from streamlit_av_queue import StreamlitAVQueue
         self._streamlit_av_queue = StreamlitAVQueue()
         self._out_audio_queue = self._streamlit_av_queue.get_out_audio_queue()
 
         print("001 - create RespondToPromptActor")
+        self._state = "001 - creating RespondToPromptActor"
         from respond_to_prompt_actor import RespondToPromptActor
         self._environment_state_actor = EnvironmentStateActor.remote()
         self._agent_state_actor = AgentStateActor.remote()
         self._respond_to_prompt_actor = RespondToPromptActor.remote(self._environment_state_actor, self._out_audio_queue)
 
         print("002 - create SpeechToTextVoskActor")
+        self._state = "002 - creating SpeechToTextVoskActor"
         from speech_to_text_vosk_actor import SpeechToTextVoskActor
         self._speech_to_text_actor = SpeechToTextVoskActor.remote("small")
         # self._speech_to_text_actor = SpeechToTextVoskActor.remote("big")
@@ -50,6 +53,7 @@ class CharlesActor:
         ]
 
         print("003 - create Prototypes")
+        self._state = "003 - creating Prototypes"
         from prototypes import Prototypes
         self._prototypes = Prototypes()
         print("010")