Update app.py

app.py

@@ -63,18 +63,22 @@ def process_document_and_answer(context, question, history=""):
 # Streamlit UI
 st.title("Gemini Document Q&A")
 
-# Initialize session state for conversation history
+# Initialize session state
 if 'conversation_history' not in st.session_state:
     st.session_state.conversation_history = ""
 
 if 'document_content' not in st.session_state:
     st.session_state.document_content = None
 
+if 'question_count' not in st.session_state:
+    st.session_state.question_count = 0
+
+
 # File Upload
 uploaded_file = st.file_uploader("Upload a document", type=["docx", "xlsx", "pdf", "csv", "txt", "pptx"])
 
 if uploaded_file is not None:
-    # Save the uploaded file to a temporary location (important for Streamlit)
+    # Save the uploaded file to a temporary location
     temp_file_path = "temp." + uploaded_file.name.split('.')[-1].lower()
     with open(temp_file_path, "wb") as temp_file:
         temp_file.write(uploaded_file.read())
@@ -84,31 +88,29 @@ if uploaded_file is not None:
 
     if "Error processing file" in st.session_state.document_content:
         st.error(st.session_state.document_content)
-        st.session_state.document_content = None # Clear document content to prevent further processing
+        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)  # Display a snippet
-
+    st.text(st.session_state.document_content[:500] + "..." if len(st.session_state.document_content) > 500 else st.session_state.document_content)
 
-# Conversation Loop
-while True:
-    question = st.text_input("Ask a question about the document (type 'exit' to end):", key=f"question_{len(st.session_state.conversation_history)}") # Unique key for each input
+    # Question Input
+    question = st.text_input("Ask a question about the document (type 'exit' to end):", key=f"question_{st.session_state.question_count}")
 
-    if question.lower() == "exit":
-        st.write("Ending conversation.")
-        break
-
-    if question and st.session_state.document_content:
-        answer = process_document_and_answer(st.session_state.document_content, question, st.session_state.conversation_history)
-
-        st.write("Answer:", answer)
+    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)
 
-        # Update conversation history
-        st.session_state.conversation_history += f"\nQuestion: {question}\nAnswer: {answer}"
+            # Update conversation history and question count
+            st.session_state.conversation_history += f"\nQuestion: {question}\nAnswer: {answer}"
+            st.session_state.question_count += 1
 
 
-# Display conversation history
-if st.session_state.conversation_history:
-    st.subheader("Conversation History:")
-    st.text(st.session_state.conversation_history)
+    # Display conversation history
+    if st.session_state.conversation_history:
+        st.subheader("Conversation History:")
+        st.text(st.session_state.conversation_history)