u

app.py

@@ -107,3 +107,19 @@ if uploaded_file:
                 st.download_button("📥 Download Report PDF", f, file_name=file_path, mime="application/pdf")
         else:
             st.warning("Please make sure to run summarization and enter name/note.")
+            
+st.subheader("💬 Ask Questions About the Report")
+
+if 'summary' in st.session_state and 'explanation' in st.session_state:
+    user_question = st.text_input("Ask a question based on the report:")
+    
+    if st.button("Get Answer") and user_question:
+        with st.spinner("Thinking..."):
+            context = st.session_state['summary'] + "\n" + st.session_state['explanation']
+            prompt = f"Context: {context}\n\nQuestion: {user_question}\n\nAnswer in simple words:"
+            response = explainer(prompt)[0]['generated_text']
+            st.success("✅ AI Response:")
+            st.write(response)
+else:
+    st.info("Upload a report and run summarization to use the chatbot.")
+