Update app.py

app.py

@@ -15,6 +15,14 @@ import google.generativeai as genai
 st.set_page_config(page_title="Gemini Q&A App", layout="centered")
 st.title("🤖 Q&A - Document & Image")
 
+# Initialize chat history
+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 = []
+
 # Load API key
 GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY") or st.secrets.get("GOOGLE_API_KEY")
 if not GOOGLE_API_KEY:
@@ -97,7 +105,6 @@ if uploaded_files and question:
         file_ext = uploaded_file.name.split('.')[-1].lower()
 
         if file_ext in ['jpg', 'jpeg', 'png']:
-            # Store image data for later processing
             image_bytes = uploaded_file.read()
             encoded_image = base64.b64encode(image_bytes).decode("utf-8")
             image_contents.append({
@@ -109,7 +116,6 @@ if uploaded_files and question:
             st.image(image_bytes, caption=uploaded_file.name, use_container_width=True)
             continue
 
-        # Process document files
         temp_file_path = "temp." + file_ext
         with open(temp_file_path, "wb") as temp_file:
             temp_file.write(uploaded_file.read())
@@ -122,27 +128,47 @@ if uploaded_files and question:
         else:
             combined_context += f"\n\n---\nDocument: {uploaded_file.name}\n\n{extracted_text}"
 
-    # Process if there's valid content
     if combined_context.strip() or image_contents:
         with st.spinner("Generating answer..."):
             try:
                 content_blocks = []
 
-                # Add text context
+                # 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']}"
+                        for entry in st.session_state.chat_history
+                    )
+                    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 user question
+                # Add current question
                 content_blocks.append({"text": f"Question: {question}"})
 
                 response = model.generate_content(contents=content_blocks)
                 st.success("💡 Answer:")
                 st.write(response.text)
 
+                # Save to session history
+                st.session_state.chat_history.append({
+                    "question": question,
+                    "answer": response.text
+                })
+
             except Exception as e:
                 st.error(f"Error generating answer: {e}")
     else:
-        st.warning("No valid documents or images processed.")
+        st.warning("No valid documents or images processed.")
+
+# 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']}")