Update app.py

app.py

@@ -10,6 +10,7 @@ from io import StringIO
 from bs4 import BeautifulSoup
 import re
 import google.generativeai as genai
+from google.generativeai.types import Tool, GoogleSearch, GenerateContentConfig
 
 # Setup
 st.set_page_config(page_title="Gemini Q&A App", layout="centered")
@@ -30,7 +31,9 @@ if not GOOGLE_API_KEY:
     st.stop()
 
 genai.configure(api_key=GOOGLE_API_KEY)
-model = genai.GenerativeModel(model_name="gemini-2.0-flash")
+client = genai.Client()
+model_id = "gemini-2.0-flash"
+google_search_tool = Tool(google_search=GoogleSearch())
 
 # Helper: Extract text from various documents
 def load_and_extract_text(file_path):
@@ -133,7 +136,6 @@ if uploaded_files and question:
             try:
                 content_blocks = []
 
-                # Add prior Q&A as part of the context
                 if st.session_state.chat_history:
                     history_text = "\n\n".join(
                         f"Q: {entry['question']}\nA: {entry['answer']}"
@@ -141,17 +143,21 @@ if uploaded_files and question:
                     )
                     content_blocks.append({"text": f"Previous conversation:\n{history_text}"})
 
-                # Add file text content
                 if combined_context.strip():
                     content_blocks.append({"text": f"Context:\n{combined_context}"})
 
-                # Add image blocks
                 content_blocks.extend(image_contents)
-
-                # Add current question
                 content_blocks.append({"text": f"Question: {question}"})
 
-                response = model.generate_content(contents=content_blocks)
+                response = client.models.generate_content(
+                    model=model_id,
+                    contents=content_blocks,
+                    config=GenerateContentConfig(
+                        tools=[google_search_tool],
+                        response_modalities=["TEXT"],
+                    )
+                )
+
                 st.success("💡 Answer:")
                 st.write(response.text)