Update app.py

app.py

@@ -19,7 +19,6 @@ st.title("🤖 Q&A - Document & Image")
 if "chat_history" not in st.session_state:
     st.session_state.chat_history = []
 
-# Optional: Clear history button
 if st.button("🧹 Clear Chat History"):
     st.session_state.chat_history = []
 
@@ -30,12 +29,9 @@ if not GOOGLE_API_KEY:
     st.stop()
 
 genai.configure(api_key=GOOGLE_API_KEY)
-model = genai.GenerativeModel(
-    model_name="gemini-2.0-flash",
-    tools=[{"google_search": {}}]
-)
+model = genai.GenerativeModel(model_name="gemini-2.0-flash")
 
-# Helper: Extract text from various documents
+# Extract text from supported document formats
 def load_and_extract_text(file_path):
     try:
         ext = file_path.split('.')[-1].lower()
@@ -89,7 +85,7 @@ def load_and_extract_text(file_path):
     except Exception as e:
         return f"Error processing file: {e}"
 
-# Multi-file uploader
+# File uploader
 uploaded_files = st.file_uploader(
     "Upload one or more documents or images",
     type=["docx", "xlsx", "pdf", "csv", "txt", "pptx", "tex", "html", "htm", "jpg", "jpeg", "png"],
@@ -99,7 +95,7 @@ uploaded_files = st.file_uploader(
 # Question input
 question = st.text_input("💬 Enter your question:")
 
-# Processing
+# Process on submit
 if uploaded_files and question:
     combined_context = ""
     image_contents = []
@@ -136,7 +132,7 @@ if uploaded_files and question:
             try:
                 content_blocks = []
 
-                # Add prior Q&A as part of the context
+                # Add prior Q&A history
                 if st.session_state.chat_history:
                     history_text = "\n\n".join(
                         f"Q: {entry['question']}\nA: {entry['answer']}"
@@ -144,17 +140,22 @@ if uploaded_files and question:
                     )
                     content_blocks.append({"text": f"Previous conversation:\n{history_text}"})
 
-                # Add file text content
+                # Add extracted file text
                 if combined_context.strip():
                     content_blocks.append({"text": f"Context:\n{combined_context}"})
 
-                # Add image blocks
+                # Add any images
                 content_blocks.extend(image_contents)
 
-                # Add current question
+                # Add the current question
                 content_blocks.append({"text": f"Question: {question}"})
 
-                response = model.generate_content(contents=content_blocks)
+                # ✅ ENABLE GOOGLE SEARCH HERE
+                response = model.generate_content(
+                    contents=content_blocks,
+                    tools=[{"google_search": {}}]
+                )
+
                 st.success("💡 Answer:")
                 st.write(response.text)
 
@@ -169,9 +170,9 @@ if uploaded_files and question:
     else:
         st.warning("No valid documents or images processed.")
 
-# Optional: Show full conversation history
+# Optional: Show full chat history
 if st.session_state.chat_history:
     st.markdown("### 🗂️ Chat History")
     for i, entry in enumerate(st.session_state.chat_history, 1):
         st.markdown(f"**Q{i}:** {entry['question']}")
-        st.markdown(f"**A{i}:** {entry['answer']}")
+        st.markdown(f"**A{i}:** {entry['answer']}")