Update app.py

app.py

@@ -45,9 +45,13 @@ def generate_text_based_on_mood(emotion, response_type):
     try:
         if response_type == "Joke":
             prompt = f"Generate a light-hearted joke for someone who is feeling {emotion}."
-        else:  # Motivational Message
+        elif response_type == "Motivational Message":
             prompt = f"Generate a motivational message for someone who is feeling {emotion}."
-        
+        elif response_type == "Compliment":
+            prompt = f"Generate a heartfelt compliment for someone who is feeling {emotion}."
+        elif response_type == "Song Recommendation":
+            prompt = f"Recommend a song that would suit someone who is feeling {emotion}. Include the song title and artist."
+
         # Call OpenAI's API using GPT-4.
         response = openai.ChatCompletion.create(
             model="gpt-4",  # Specify the GPT-4 model
@@ -97,20 +101,26 @@ def main():
             st.write(f"Detected emotion: {emotion}")
 
             # Dropdown for selecting response type.
-            response_type = st.selectbox("Select the type of response:", ["Joke", "Motivational Message"])
+            response_type = st.selectbox("Select the type of response:", ["Joke", "Motivational Message", "Compliment", "Song Recommendation"])
 
             # Generate text based on detected emotion and user preference.
             if st.button("Get Response"):
                 message = generate_text_based_on_mood(emotion, response_type)
+                
+                # Display the generated message (song recommendation or other response)
                 st.write("Here's your response:")
                 st.write(message)
 
-                # Convert the generated message to audio.
-                audio_file = text_to_speech(message)
+                # If the response is a song recommendation, provide extra formatting if desired.
+                if response_type == "Song Recommendation":
+                    st.write("Song Recommendation:")
+                    st.write(message)
 
-                # Provide an audio player in the Streamlit app if audio file exists.
-                if audio_file:
-                    st.audio(audio_file)  # Streamlit will handle playback.
+                # Convert the generated message to audio if it's not a song recommendation (optional)
+                if response_type != "Song Recommendation":
+                    audio_file = text_to_speech(message)
+                    if audio_file:
+                        st.audio(audio_file)  # Streamlit will handle playback.
 
 #############################################################################################################################
 # Run the application.