update

app.py

@@ -23,8 +23,8 @@ def preprocess_image(image: Image.Image) -> Image.Image:
     image_height = int(image.height * IMAGE_WIDTH / image.width)
     return image.resize((IMAGE_WIDTH, image_height))
 
-def user(text_prompt: str, chatbot: List[Tuple[str, str]]):
-    return "", chatbot + [[text_prompt, None]]
+def user(text_prompt: str, chatbot: List):
+    return "", chatbot + [{"role": "user", "content": text_prompt}]
 
 def bot(
     google_key: str,
@@ -34,17 +34,17 @@ def bot(
     stop_sequences: str,
     top_k: int,
     top_p: float,
-    chatbot: List[Tuple[str, str]]
+    chatbot: List
 ):
     google_key = google_key or GOOGLE_API_KEY
     if not google_key:
         raise ValueError("GOOGLE_API_KEY is not set. Please set it up.")
 
-    text_prompt = chatbot[-1][0].strip() if chatbot[-1][0] else None
+    text_prompt = chatbot[-1]["content"].strip() if chatbot[-1]["content"] else None
 
     # Handle cases for text and/or image input
     if not text_prompt and not image_prompt:
-        chatbot[-1][1] = "Prompt cannot be empty. Please provide input text or an image."
+        chatbot[-1]["content"] = "Prompt cannot be empty. Please provide input text or an image."
         yield chatbot
         return
     elif image_prompt and not text_prompt:
@@ -74,17 +74,18 @@ def bot(
         response = model_instance.generate_content(inputs, stream=True, generation_config=generation_config)
         response.resolve()
     except Exception as e:
-        chatbot[-1][1] = f"Error occurred: {str(e)}"
+        chatbot[-1]["content"] = f"Error occurred: {str(e)}"
         yield chatbot
         return
 
     # Stream the response back to the chatbot
-    chatbot[-1][1] = ""
+    chatbot.append({"role": "assistant", "content": ""})
     for chunk in response:
-        for i in range(0, len(chunk.text), 10):
-            chatbot[-1][1] += chunk.text[i:i + 10]
-            time.sleep(0.01)
-            yield chatbot
+        if chunk.text:
+            for i in range(0, len(chunk.text), 10):
+                chatbot[-1]["content"] += chunk.text[i:i + 10]
+                time.sleep(0.01)
+                yield chatbot
 # Components
 google_key_component = gr.Textbox(
     label="Google API Key",