Update app.py

app.py

@@ -85,6 +85,63 @@ def user(text_prompt: str, chatbot: CHAT_HISTORY):
     return "", chatbot
 
 
+# def bot(
+#     google_key: str,
+#     model_name: str,
+#     files: Optional[List[str]],
+#     temperature: float,
+#     max_output_tokens: int,
+#     stop_sequences: str,
+#     top_k: int,
+#     top_p: float,
+#     chatbot: CHAT_HISTORY
+# ):
+#     if len(chatbot) == 0:
+#         return chatbot
+
+#     google_key = google_key if google_key else GOOGLE_API_KEY
+#     if not google_key:
+#         raise ValueError(
+#             "GOOGLE_API_KEY is not set. "
+#             "Please follow the instructions in the README to set it up.")
+
+#     genai.configure(api_key=google_key)
+#     generation_config = genai.types.GenerationConfig(
+#         temperature=temperature,
+#         max_output_tokens=max_output_tokens,
+#         stop_sequences=preprocess_stop_sequences(stop_sequences=stop_sequences),
+#         top_k=top_k,
+#         top_p=top_p)
+
+#     if files:
+#         text_prompt = [chatbot[-1][0]] \
+#             if chatbot[-1][0] and isinstance(chatbot[-1][0], str) \
+#             else []
+#         image_prompt = [Image.open(file).convert('RGB') for file in files]
+#         model = genai.GenerativeModel(model_name)
+#         response = model.generate_content(
+#             text_prompt + image_prompt,
+#             stream=True,
+#             generation_config=generation_config)
+#     else:
+#         messages = preprocess_chat_history(chatbot)
+#         model = genai.GenerativeModel(model_name)
+#         response = model.generate_content(
+#             messages,
+#             stream=True,
+#             generation_config=generation_config)
+
+#     # streaming effect
+#     chatbot[-1][1] = ""
+#     for chunk in response:
+#         for i in range(0, len(chunk.text), 10):
+#             section = chunk.text[i:i + 10]
+#             chatbot[-1][1] += section
+#             time.sleep(0.01)
+#             yield chatbot
+
+# -------------------------------------------------------------------
+
 def bot(
     google_key: str,
     model_name: str,
@@ -134,11 +191,23 @@ def bot(
     # streaming effect
     chatbot[-1][1] = ""
     for chunk in response:
-        for i in range(0, len(chunk.text), 10):
-            section = chunk.text[i:i + 10]
-            chatbot[-1][1] += section
-            time.sleep(0.01)
-            yield chatbot
+        if not chunk.text:
+            print("chunk.text is empty")
+            continue
+
+        print(f"chunk.text: {chunk.text}")
+
+        try:
+            for i in range(0, len(chunk.text), 10):
+                section = chunk.text[i:i + 10]
+                chatbot[-1][1] += section
+                time.sleep(0.01)
+                yield chatbot
+        except IndexError as e:
+            print(f"IndexError: {e}")
+            # Handle the error appropriately
+
+# -------------------------------------------------------------------
 
 model_selection = gr.Dropdown(
     ["gemini-1.5-flash",