update

app.py

@@ -56,9 +56,6 @@ def bot(
                 text_prompt = item.get("text", "").strip()
                 break
 
-    # Debug: print what we received
-    print(f"DEBUG: content={content}, text_prompt='{text_prompt}'")
-
     # Handle cases for text and/or image input
     if not text_prompt and not image_prompt:
         chatbot[-1]["content"] = "Prompt cannot be empty. Please provide input text or an image."
@@ -71,25 +68,32 @@ def bot(
         # If both text and image are provided, combine them
         text_prompt = f"{text_prompt}. Also, analyze the provided image."
 
-    # Configure the model
-    genai.configure(api_key=google_key)
-    model_instance = genai.GenerativeModel(model_name, system_instruction=system_instruction_analysis)
+    # Initialize the client with API key
+    client = genai.Client(api_key=google_key)
     
-    generation_config = genai.types.GenerationConfig(
-        temperature=temperature,
-        max_output_tokens=max_output_tokens,
-        stop_sequences=preprocess_stop_sequences(stop_sequences),
-        top_k=top_k,
-        top_p=top_p,
-    )
+    generation_config = {
+        "temperature": temperature,
+        "max_output_tokens": max_output_tokens,
+        "top_k": top_k,
+        "top_p": top_p,
+    }
+    
+    if preprocess_stop_sequences(stop_sequences):
+        generation_config["stop_sequences"] = preprocess_stop_sequences(stop_sequences)
 
     # Prepare inputs
     inputs = [text_prompt] if image_prompt is None else [text_prompt, preprocess_image(image_prompt)]
 
     # Generate response
     try:
-        response = model_instance.generate_content(inputs, stream=True, generation_config=generation_config)
-        response.resolve()
+        response = client.models.generate_content(
+            model=model_name,
+            contents=inputs,
+            config=genai.types.GenerateContentConfig(
+                system_instruction=system_instruction_analysis,
+                **generation_config
+            ),
+        )
     except Exception as e:
         chatbot[-1]["content"] = f"Error occurred: {str(e)}"
         yield chatbot
@@ -97,12 +101,16 @@ def bot(
 
     # Stream the response back to the chatbot
     chatbot.append({"role": "assistant", "content": ""})
-    for chunk in response:
-        if chunk.text:
-            for i in range(0, len(chunk.text), 10):
-                chatbot[-1]["content"] += chunk.text[i:i + 10]
+    try:
+        if response.text:
+            # Stream the response text character by character
+            for i in range(0, len(response.text), 10):
+                chatbot[-1]["content"] += response.text[i:i + 10]
                 time.sleep(0.01)
                 yield chatbot
+    except Exception as e:
+        chatbot[-1]["content"] = f"Error processing response: {str(e)}"
+        yield chatbot
 # Components
 google_key_component = gr.Textbox(
     label="Google API Key",