Update app.py

app.py

@@ -12,6 +12,8 @@ from reportlab.pdfgen import canvas
 import docx
 from docx.enum.text import WD_PARAGRAPH_ALIGNMENT
 from io import BytesIO
+import session_state  # Import SessionState
+
 
 # Set up OpenAI API
 openai.api_key = "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"
@@ -109,11 +111,20 @@ if uploaded_file is not None:
             # Add the chatbot functionality
             st.header("1-to-1 Consultation")
             st.write("Ask any questions you have about the radiology report:")
+
+            # Use SessionState to store the chat history
+            state = session_state.get(chat_history=[])
+
             user_input = st.text_input("Enter your question:")
             if user_input:
+                state.chat_history.append(f"You: {user_input}")
                 if user_input.lower() == "thank you":
-                    st.write("You're welcome! If you have any more questions, feel free to ask.")
+                    state.chat_history.append("Bot: You're welcome! If you have any more questions, feel free to ask.")
                 else:
                     # Add the OpenAI API call here and generate the answer to the user's question
                     answer = f"Answer to the user's question based on the generated radiology report: {user_input}"
-                    st.write(answer)
+                    state.chat_history.append(f"Bot: {answer}")
+
+            # Display the chat history
+            for message in state.chat_history:
+                st.write(message)