Update app.py

app.py

@@ -6,6 +6,11 @@ import openpyxl
 from pdfminer.high_level import extract_text
 import csv
 from pptx import Presentation
+import base64  # For creating PDF download link
+from io import BytesIO  # For PDF generation in memory
+from reportlab.pdfgen import canvas
+from reportlab.lib.pagesizes import letter
+from reportlab.lib.colors import blue, red
 
 # Configure Gemini API
 GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
@@ -60,6 +65,11 @@ def process_document_and_answer(context, question, history=""):
         return f"Error generating answer: {e}"
 
 
+def create_download_link(val, filename):
+    b64 = base64.b64encode(val)  # val looks like b'...'
+    return f'<a href="data:application/octet-stream;base64,{b64.decode()}" download="{filename}">Download conversation as PDF</a>'
+
+
 # Streamlit UI
 st.title("Gemini Document Q&A")
 
@@ -73,6 +83,8 @@ if 'document_content' not in st.session_state:
 if 'question_count' not in st.session_state:
     st.session_state.question_count = 0
 
+if 'display_answer' not in st.session_state:
+    st.session_state.display_answer = False
 
 # File Upload
 uploaded_file = st.file_uploader("Upload a document", type=["docx", "xlsx", "pdf", "csv", "txt", "pptx"])
@@ -96,21 +108,48 @@ if st.session_state.document_content:
     st.text(st.session_state.document_content[:500] + "..." if len(st.session_state.document_content) > 500 else st.session_state.document_content)
 
     # Question Input
-    question = st.text_input("Ask a question about the document (type 'exit' to end):", key=f"question_{st.session_state.question_count}")
+    question = st.text_input("**Ask a question about the document (type 'exit' to end):**", key=f"question_{st.session_state.question_count}")
 
     if question:
         if question.lower() == "exit":
             st.write("Ending conversation.")
         else:
             answer = process_document_and_answer(st.session_state.document_content, question, st.session_state.conversation_history)
-            st.write("Answer:", answer)
+            st.markdown(f"<p style='color:green;'><b>Answer:</b> {answer}</p>", unsafe_allow_html=True)  # Colored Answer
+            st.session_state.display_answer = True # set trigger to allow auto scroll
 
             # Update conversation history and question count
             st.session_state.conversation_history += f"\nQuestion: {question}\nAnswer: {answer}"
             st.session_state.question_count += 1
+    if st.session_state.display_answer:
+      js = f"""
+      <script>
+          var textarea = document.getElementById('question_{st.session_state.question_count -1 }');
+          textarea.scrollIntoView();
+      </script>
+      """
+      st.components.v1.html(js, height=0)
+      st.session_state.display_answer = False # set back to false after scroll
+
 
 
-    # Display conversation history
+    # PDF Export
     if st.session_state.conversation_history:
-        st.subheader("Conversation History:")
-        st.text(st.session_state.conversation_history)
+        pdf_buffer = BytesIO()
+        c = canvas.Canvas(pdf_buffer, pagesize=letter)
+        textobject = c.beginText()
+        textobject.setTextOrigin(10, 730)
+        textobject.setFont("Helvetica", 12)
+
+        for line in st.session_state.conversation_history.split('\n'):
+            textobject.setFillColor(blue)
+            if line.startswith("Answer:"):
+                textobject.setFillColor(red)
+            textobject.textLine(line)
+
+        c.drawText(textobject)
+        c.save()
+        pdf_buffer.seek(0)
+
+        download_link = create_download_link(pdf_buffer.read(), "Conversation.pdf")
+        st.markdown(download_link, unsafe_allow_html=True)