set ModeManager class

app.py

@@ -56,6 +56,18 @@ refiner = DiffusionPipeline.from_pretrained(
 )
 refiner.to('cuda')
 
+class ModeManager:
+    def __init__(self):
+        self.mode = ""
+
+    def set_mode(self, mode):
+        self.mode = mode
+
+    def get_mode(self):
+        return self.mode
+
+mode_manager = ModeManager()
+
 def multimodal_and_generation(message, history):
     """
     Generates a response based on the input message and optionally an image.
@@ -127,7 +139,7 @@ def bot_comms(message, history):
     Handles communication between Gradio and the models.
     """
     # global mode
-    mode = spaces.get("mode", default="")
+    mode = mode_manager.get_mode()
     logger.debug(f"bot_comms called with message: {message} and mode: {mode}")
 
     if message == "check cuda":
@@ -137,13 +149,13 @@ def bot_comms(message, history):
     
     if message == "imagery":
         logger.debug("Switching to imagery mode.")
-        spaces.set("mode", "imagery")
+        mode_manager.set_mode("imagery")
         yield "Imagery On! Type your prompt to make the image 🖼️"
         return
     
     if message == "chatting":
         logger.debug("Switching to chatting mode.")
-        spaces.set("mode", "chatting")
+        mode_manager.set_mode("chatting")
         yield "Imagery Off. Ask me any questions. ☄️"
         return