Update app.py

app.py

@@ -160,14 +160,6 @@ def main():
     st.title("Chat with your Document")
 
     if st.session_state['qa_chain']:
-        message = st.text_input("Ask a question")
-
-        if st.button("Submit"):
-            with st.spinner("Generating response..."):
-                qa_chain, chat_history, response_answer, sources = conversation(st.session_state['qa_chain'], message, st.session_state['chat_history'])
-                st.session_state['qa_chain'] = qa_chain
-                st.session_state['chat_history'] = chat_history
-
         st.markdown("### Chatbot Response")
 
         # Display the chat history in a chat-like interface
@@ -175,9 +167,22 @@ def main():
             st.markdown(f"**User:** {user_msg}")
             st.markdown(f"**Assistant:** {bot_msg}")
 
+        st.markdown("### Relevant context from the source document")
+
         with st.expander("Relevant context from the source document"):
-            for source in sources:
-                st.text_area(f"Source - Page {source['page']}", value=source["content"], height=100)
+            if 'sources' in st.session_state:
+                for i, source in enumerate(st.session_state['sources']):
+                    st.text_area(f"Source {i + 1} - Page {source['page']}", value=source["content"], height=100)
+
+        message = st.text_input("Ask a question", key="message")
+
+        if st.button("Submit"):
+            with st.spinner("Generating response..."):
+                qa_chain, chat_history, response_answer, sources = conversation(st.session_state['qa_chain'], st.session_state['message'], st.session_state['chat_history'])
+                st.session_state['qa_chain'] = qa_chain
+                st.session_state['chat_history'] = chat_history
+                st.session_state['sources'] = sources
+                st.session_state['message'] = ""
 
 if __name__ == "__main__":
     main()