Update app.py

app.py

@@ -19,6 +19,7 @@ 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 = []
 
@@ -31,7 +32,7 @@ if not GOOGLE_API_KEY:
 genai.configure(api_key=GOOGLE_API_KEY)
 model = genai.GenerativeModel(model_name="gemini-2.0-flash")
 
-# Extract text from supported document formats
+# Helper: Extract text from various documents
 def load_and_extract_text(file_path):
     try:
         ext = file_path.split('.')[-1].lower()
@@ -85,7 +86,7 @@ def load_and_extract_text(file_path):
     except Exception as e:
         return f"Error processing file: {e}"
 
-# File uploader
+# Multi-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"],
@@ -95,7 +96,7 @@ uploaded_files = st.file_uploader(
 # Question input
 question = st.text_input("💬 Enter your question:")
 
-# Process on submit
+# Processing
 if uploaded_files and question:
     combined_context = ""
     image_contents = []
@@ -132,7 +133,7 @@ if uploaded_files and question:
             try:
                 content_blocks = []
 
-                # Add prior Q&A history
+                # 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']}"
@@ -140,22 +141,17 @@ if uploaded_files and question:
                     )
                     content_blocks.append({"text": f"Previous conversation:\n{history_text}"})
 
-                # Add extracted file text
+                # Add file text content
                 if combined_context.strip():
                     content_blocks.append({"text": f"Context:\n{combined_context}"})
 
-                # Add any images
+                # Add image blocks
                 content_blocks.extend(image_contents)
 
-                # Add the current question
+                # Add current question
                 content_blocks.append({"text": f"Question: {question}"})
 
-                # ✅ ENABLE GOOGLE SEARCH HERE
-                response = model.generate_content(
-                    contents=content_blocks,
-                    tools=[{"google_search": {}}]
-                )
-
+                response = model.generate_content(contents=content_blocks)
                 st.success("💡 Answer:")
                 st.write(response.text)
 
@@ -170,9 +166,9 @@ if uploaded_files and question:
     else:
         st.warning("No valid documents or images processed.")
 
-# Optional: Show full chat history
+# Optional: Show full conversation 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']}")