Update app.py

app.py

@@ -31,11 +31,11 @@ def convert_audio_to_wav(audio_bytes):
 def detect_mood(text):
     result = mood_classifier(text)[0]
     if result['label'] == 'POSITIVE':
-        return "joyful"
+        return "happy"
     elif result['label'] == 'NEGATIVE':
         return "sad"
     else:
-        return "neutral"
+        return "calm"
 
 def speech_to_text():
     # Initialize recognizer
@@ -130,8 +130,14 @@ user_text = None
 
 if input_method == "Text":
     # Text input
-    user_mood = st.sidebar.selectbox("Select your mood:", mood_options)
-    user_text = user_mood
+    user_text = st.sidebar.text_input("Enter text describing your mood:")
+    if st.sidebar.button("Submit"):
+        if user_text:
+            user_mood = detect_mood(user_text)
+            st.write(f"Based on your text, I detect your mood as: **{user_mood}**")
+        else:
+            st.sidebar.warning("Please enter your mood description.")
+            user_mood = None
 else:
     # Speech input
     st.write("📢 Click the button below and tell me about your day...")
@@ -147,12 +153,11 @@ if 'playlist' not in st.session_state:
     st.session_state.playlist = []
 
 # Main content
-if user_text and gemini_api_key:
-    detected_mood = detect_mood(user_text)
-    st.write(f"🎭 Detected Mood: **{detected_mood}**")
+if user_mood and gemini_api_key:
+    st.write(f"🎭 Detected Mood: **{user_mood}**")
     
     st.write("🎵 Recommended Songs:")
-    recommendations = get_song_recommendations(detected_mood, gemini_api_key)
+    recommendations = get_song_recommendations(user_mood, gemini_api_key)
     if recommendations:
         st.write(recommendations)