Update app.py

app.py

@@ -129,7 +129,11 @@ with input_container:
             with st.spinner("processing....."):
                 response = generate_response(user_message)
                 st.session_state.past.append(user_message)
-                st.session_state.generated.append(response)
+                st.session_state.messages.append(
+                    {"role": "ai", "content": response}
+                )
+                #st.session_state.generated.append(response)
+                
                 # Detect sentiment
                 user_sentiment = chatbot.detect_sentiment(user_message)
     
@@ -167,6 +171,13 @@ with input_container:
                 chatbot.update_q_values(
                     user_sentiment, reward, user_sentiment
                 )
+                
+            with st.chat_message("ai"):
+                st.markdown(llm_reponse_with_quesiton)
+                # Update data to memory
+                st.session_state.user_sentiment = user_sentiment
+                st.session_state.mood_trend = mood_trend
+                st.session_state.mood_trend_symbol = mood_trend_symbol
 
                 # Convert the response to speech
                 speech_fp = text_to_speech(response)
@@ -181,8 +192,7 @@ with input_container:
             message(st.session_state["generated"][i], key=str(I))'''
 
 
-with st.sidebar.expander("Behind the Scene"):
-        st.subheader("What AI is doing:")
+with st.sidebar.expander("Sentiment Analysis"):
         # Use the values stored in session state
         st.write(
             f"- Detected User Tone: {st.session_state.user_sentiment} ({st.session_state.mood_trend.capitalize()}{st.session_state.mood_trend_symbol})"