Update app.py

app.py

@@ -101,27 +101,34 @@ def get_text():
 def generate_response(prompt):
     response = mdl.call_conversational_rag(prompt,final_chain)
     return response['answer']
-        
+
+# Collect user input
+# Add a radio button to choose input mode
+input_mode = st.sidebar.radio("Select input mode:", ["Text", "Speech"])
+user_message = None
+if input_mode == "Speech":
+    # Use the speech_to_text function to capture speech input
+    speech_input = speech_to_text(key="my_stt", callback=speech_recognition_callback)
+    # Check if speech input is available
+    if "speech_input" in st.session_state and st.session_state.speech_input:
+        # Display the speech input
+        # st.text(f"Speech Input: {st.session_state.speech_input}")
+
+        # Process the speech input as a query
+        user_message = st.session_state.speech_input
+        st.session_state.speech_input = None
+else:
+    user_message = st.chat_input("Type your message here:")
      
 ## Applying the user input box        
 with input_container:
-    # Add a radio button to choose input mode
-    input_mode = st.radio("Select input mode:", ["Text", "Speech"])
-
-    if input_mode == "Speech":
-        # Use the speech_to_text function to capture speech input
-        speech_input = speech_to_text(
-            key='my_stt', 
-            callback=speech_recognition_callback
-        )
-
-        # Check if speech input is available
-        if 'speech_input' in st.session_state and st.session_state.speech_input:
-            # Display the speech input
-            st.text(f"Speech Input: {st.session_state.speech_input}")
+    if user_message:
+        st.session_state.entered_text.append(user_message)
+    
+        st.session_state.messages.append({"role": "user", "content": user_message})
+        with st.chat_message("user"):
+            st.write(user_message)
             
-            # Process the speech input as a query
-            query = st.session_state.speech_input
             with st.spinner("processing....."):
                 response = generate_response(query)
                 st.session_state.past.append(query)
@@ -129,67 +136,6 @@ with input_container:
                 # Detect sentiment
                 user_sentiment = chatbot.detect_sentiment(query)
     
-    
-                
-                # 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 = " ⬆️"
-                    elif mood_trend == "unchanged":
-                        reward = +0.8
-                        mood_trend_symbol = ""
-                    else:  # decreased
-                        reward = -0.2
-                        mood_trend_symbol = " ⬇️"
-                else:
-                    if mood_trend == "increased":
-                        reward = +1
-                        mood_trend_symbol = " ⬆️"
-                    elif mood_trend == "unchanged":
-                        reward = -0.2
-                        mood_trend_symbol = ""
-                    else:  # decreased
-                        reward = -1
-                        mood_trend_symbol = " ⬇️"
-    
-                print(
-                    f"mood_trend - sentiment - reward: {mood_trend} - {user_sentiment} - 🛑{reward}🛑"
-                )
-    
-                # Update Q-values
-                chatbot.update_q_values(
-                    user_sentiment, reward, user_sentiment
-                )
-
-                # Convert the response to speech
-                speech_fp = text_to_speech(response)
-                # Play the speech
-                st.audio(speech_fp, format='audio/mp3')
-
-    else:
-        # Add a text input field for query
-        query = st.text_input("Query: ", key="input")
-
-        # Process the query if it's not empty
-        if query:
-            with st.spinner("typing....."):
-                response = generate_response(query)
-                st.session_state.past.append(query)
-                st.session_state.generated.append(response)
-                # Detect sentiment
-                user_sentiment = chatbot.detect_sentiment(query)
-    
-               
-                # Convert the response to speech
-                speech_fp = text_to_speech(response)
-                # Play the speech
-                st.audio(speech_fp, format='audio/mp3')
-                
                 # Update mood history / mood_trend
                 chatbot.update_mood_history()
                 mood_trend = chatbot.check_mood_trend()