placed spaces.get and set for using the mode variable

app.py

@@ -126,7 +126,8 @@ def bot_comms(message, history):
     """
     Handles communication between Gradio and the models.
     """
-    global mode
+    # global mode
+    mode = spaces.get("mode", default="")
     logger.debug(f"bot_comms called with message: {message} and mode: {mode}")
 
     if message == "check cuda":
@@ -136,13 +137,13 @@ def bot_comms(message, history):
     
     if message == "imagery":
         logger.debug("Switching to imagery mode.")
-        mode = message
+        spaces.set("mode", "imagery")
         yield "Imagery On! Type your prompt to make the image 🖼️"
         return
     
     if message == "chatting":
         logger.debug("Switching to chatting mode.")
-        mode = message
+        spaces.set("mode", "chatting")
         yield "Imagery Off. Ask me any questions. ☄️"
         return