added conditions if the message wasn't a dict for multimodal_and_generation

app.py

@@ -102,7 +102,7 @@ def multimodal_and_generation(message, history):
     #             image_path = hist[0][0]
 
     # if image_path is None:
-    input_prompt = message
+    input_prompt = message if isinstance(message, str) else message.get("text", "")
     client = openai.OpenAI(api_key=API_KEY)
     stream = client.chat.completions.create(
         model="gpt-3.5-turbo",
@@ -167,9 +167,10 @@ def bot_comms(message, history):
     """
     Handles communication between Gradio and the models.
     """
-    global mode
-    # mode = mode_manager.get_mode()
-    # logger.debug(f"bot_comms called with message: {message} and mode: {mode}")
+
+    # ensures message is a dictionary
+    if not isinstance(message, dict):
+        message = {"text": message}
 
     if message["text"] == "check cuda":
         logger.debug("Checking CUDA availability.")