Update app.py

app.py

@@ -6,11 +6,12 @@ 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
+import base64
+from io import BytesIO
 from reportlab.pdfgen import canvas
 from reportlab.lib.pagesizes import letter
 from reportlab.lib.colors import blue, red
+import streamlit.components.v1 as components
 
 # Configure Gemini API
 GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
@@ -69,6 +70,26 @@ 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>'
 
+def move_element(element, location="bottom"):
+    js_code = f"""
+        <script>
+        window.onload = function() {{
+            var element = document.getElementById('{element}');
+            if (element) {{
+                element.style.position = 'fixed';
+                element.style.bottom = '0';
+                element.style.left = '0';
+                element.style.width = '100%';
+                element.style.backgroundColor = 'white';
+                element.style.zIndex = '9999';
+                element.style.padding = '10px';
+                element.style.boxShadow = '0px -2px 5px #888888';
+                document.body.appendChild(element);
+            }}
+        }}
+        </script>
+    """
+    components.html(js_code, height=0)
 
 # Streamlit UI
 st.title("Gemini Document Q&A")
@@ -86,6 +107,10 @@ if 'question_count' not in st.session_state:
 if 'display_answer' not in st.session_state:
     st.session_state.display_answer = False
 
+if 'answers' not in st.session_state:
+    st.session_state.answers = []
+
+
 # File Upload
 uploaded_file = st.file_uploader("Upload a document", type=["docx", "xlsx", "pdf", "csv", "txt", "pptx"])
 
@@ -102,35 +127,30 @@ if uploaded_file is not None:
         st.error(st.session_state.document_content)
         st.session_state.document_content = None
 
-
 if st.session_state.document_content:
-    st.subheader("Document Content Preview:")
-    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}")
+    # Display Answers above
+    if st.session_state.answers:
+        for i, answer in enumerate(st.session_state.answers):
+            st.markdown(f"<p style='color:green;'><b>Answer {i+1}:</b> {answer}</p>", unsafe_allow_html=True)
+
+    # Question Input at bottom using javascript to stick at the bottom
+    move_element("question_input")
+    question = st.text_input("**Ask a question (type 'exit' to end):**", key="question_input", value="")
 
     if question:
         if question.lower() == "exit":
             st.write("Ending conversation.")
+            st.stop() # this one is more appropiate to stop
         else:
             answer = process_document_and_answer(st.session_state.document_content, question, st.session_state.conversation_history)
-            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
+            st.session_state.answers.append(answer)
+            st.session_state.display_answer = True
 
             # 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
-
+            st.rerun() # Rerun to display new question at bottom and answer at top
 
 
     # PDF Export
@@ -151,5 +171,5 @@ if st.session_state.document_content:
         c.save()
         pdf_buffer.seek(0)
 
-        download_link = create_download_link(pdf_buffer.read(), "Conversation.pdf")
+        download_link = create_download_link(pdf_buffer.read(), "conversation.pdf")
         st.markdown(download_link, unsafe_allow_html=True)