Update app.py

app.py

@@ -10,7 +10,6 @@ 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")
@@ -31,9 +30,7 @@ if not GOOGLE_API_KEY:
     st.stop()
 
 genai.configure(api_key=GOOGLE_API_KEY)
-client = genai.Client()
-model_id = "gemini-2.0-flash"
-google_search_tool = Tool(google_search=GoogleSearch())
+model = genai.GenerativeModel(model_name="gemini-2.0-flash")
 
 # Helper: Extract text from various documents
 def load_and_extract_text(file_path):
@@ -136,6 +133,7 @@ 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']}"
@@ -143,21 +141,17 @@ 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)
-                content_blocks.append({"text": f"Question: {question}"})
 
-                response = client.models.generate_content(
-                    model=model_id,
-                    contents=content_blocks,
-                    config=GenerateContentConfig(
-                        tools=[google_search_tool],
-                        response_modalities=["TEXT"],
-                    )
-                )
+                # Add current question
+                content_blocks.append({"text": f"Question: {question}"})
 
+                response = model.generate_content(contents=content_blocks)
                 st.success("💡 Answer:")
                 st.write(response.text)