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",