Update app.py

app.py

@@ -126,11 +126,6 @@ def conversation(qa_chain, message, history):
         st.error(f"Error in conversation: {e}")
         return qa_chain, history, "", []
 
-def download_history(chat_history):
-    formatted_chat_history = format_chat_history(chat_history)
-    history_text = "\n".join(formatted_chat_history)
-    st.download_button(label="Download Chat History", data=history_text, file_name="chat_history.txt", mime="text/plain")
-
 def main():
     st.sidebar.title("PDF Chatbot")
 
@@ -173,18 +168,16 @@ def main():
                 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
-                for i, (user_msg, bot_msg) in enumerate(st.session_state['chat_history']):
-                    st.markdown(f"**User:** {user_msg}")
-                    st.markdown(f"**Assistant:** {bot_msg}")
+        st.markdown("### Chatbot Response")
 
-                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)
+        # Display the chat history in a chat-like interface
+        for i, (user_msg, bot_msg) in enumerate(st.session_state['chat_history']):
+            st.markdown(f"**User:** {user_msg}")
+            st.markdown(f"**Assistant:** {bot_msg}")
 
-                download_history(st.session_state['chat_history'])
+        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 __name__ == "__main__":
     main()