Update app.py

app.py

@@ -121,28 +121,27 @@ else:
 with input_container:
     if user_message:
         st.session_state.entered_text.append(user_message)
-    
         st.session_state.messages.append({"role": "user", "content": user_message})
+        
+        # Display the user's message
         with st.chat_message("user"):
             st.write(user_message)
             
-            with st.spinner("processing....."):
-                response = generate_response(user_message)
-                st.session_state.past.append(user_message)
-                st.session_state.messages.append(
-                    {"role": "ai", "content": response}
-                )
-                #st.session_state.generated.append(response)
-                
-                # Detect sentiment
-                user_sentiment = chatbot.detect_sentiment(user_message)
-    
-                # Update mood history / mood_trend
-                chatbot.update_mood_history()
-                mood_trend = chatbot.check_mood_trend()
-    
-                # Define rewards
-                if user_sentiment in ["Positive", "Moderately Positive"]:
+        # Process the user's message and generate a response
+        with st.spinner("Processing..."):
+            response = generate_response(user_message)
+            st.session_state.past.append(user_message)
+            st.session_state.messages.append({"role": "ai", "content": response})
+            
+            # Detect sentiment
+            user_sentiment = chatbot.detect_sentiment(user_message)
+            
+            # Update mood history / mood_trend
+            chatbot.update_mood_history()
+            mood_trend = chatbot.check_mood_trend()
+
+            # Define rewards
+            if user_sentiment in ["Positive", "Moderately Positive"]:
                     if mood_trend == "increased":
                         reward = +1
                         mood_trend_symbol = " ⬆️"
@@ -151,8 +150,8 @@ with input_container:
                         mood_trend_symbol = ""
                     else:  # decreased
                         reward = -0.2
-                        mood_trend_symbol = " ⬇️"
-                else:
+                        mood_trend_symbol = " ⬇️"    
+            else:
                     if mood_trend == "increased":
                         reward = +1
                         mood_trend_symbol = " ⬆️"
@@ -165,20 +164,19 @@ with input_container:
     
                 print(
                     f"mood_trend - sentiment - reward: {mood_trend} - {user_sentiment} - 🛑{reward}🛑"
-                )
+                )    
     
-                # Update Q-values
-                chatbot.update_q_values(
-                    user_sentiment, reward, user_sentiment
-                )
-                
+
+            # Update Q-values
+            chatbot.update_q_values(user_sentiment, reward, user_sentiment)
+            
+            # Display the AI's response
             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)
                 # Play the speech