Update app.py

app.py

@@ -14,6 +14,8 @@ model.to(device)
 # Initialize chat history
 if 'history' not in st.session_state:
     st.session_state['history'] = []
+if 'conversation' not in st.session_state:
+    st.session_state['conversation'] = []
 
 def generate_response(input_text):
     # Encode the new user input, add end of string token
@@ -39,11 +41,20 @@ def generate_response(input_text):
     # Update session state history with the new tokens (flattened)
     st.session_state['history'] = chat_history_ids[0].tolist()
 
+    # Add both user input and bot response to the conversation history for display
+    st.session_state['conversation'].append(f"You: {input_text}")
+    st.session_state['conversation'].append(f"Bot: {bot_output}")
+
     return bot_output
 
 # Streamlit Interface
 st.title("Chat with DialoGPT")
 
+# Display the conversation history
+if st.session_state['conversation']:
+    for message in st.session_state['conversation']:
+        st.markdown(f"<p style='color:gray; padding:5px;'>{message}</p>", unsafe_allow_html=True)
+
 # Create input box for user
 user_input = st.text_input("You: ", "")