Update app.py

app.py

@@ -7,22 +7,35 @@ st.title("🎶 Mood-Based Indian Song Player")
 
 # Sidebar for user input
 st.sidebar.header("Enter Your Input")
-input_method = "Text"  # Only text input is available
 
 # Input for Gemini API key
 gemini_api_key = st.sidebar.text_input("Enter your Gemini API Key:", type="password")
 
-# Text input for user's mood or context
-user_input = st.text_input("Enter your mood or context:")
+# Text input for user's mood and context
+user_input = st.text_input("Enter your mood and context:", 
+                           placeholder="e.g., 'I'm feeling happy on a sunny day'")
+
+# Initialize playlist in session state
+if 'playlist' not in st.session_state:
+    st.session_state.playlist = []
 
 # Function to get song recommendations using Gemini API
 def get_song_recommendations(user_input, api_key):
     try:
         genai.configure(api_key=api_key)
         model = genai.GenerativeModel('gemini-pro')
-        # Prompt to guide the AI
+        # Enhanced prompt with examples
         prompt = f"""
-        Based on the input: '{user_input}', determine the mood or context and suggest 3 popular Indian songs that match this mood or context.
+        Given the input: '{user_input}', determine the mood or context and suggest 3 popular Indian songs that match this mood or context, considering the specific details provided.
+        
+        Examples:
+        - Input: 'I'm feeling stressed at work'
+          Mood: stressed
+          Songs: calming instrumental music
+        - Input: 'I'm excited about my vacation'
+          Mood: excited
+          Songs: upbeat and joyful songs
+        
         Output the mood followed by the song recommendations in the following format:
         Mood: [mood]
         1. Song Title - Artist Name
@@ -62,8 +75,9 @@ if user_input and gemini_api_key:
                 video_url = search_youtube(query)
                 if video_url:
                     st.video(video_url)
-                    if st.button(f"Add '{song}' to Playlist"):
-                        # Add to playlist logic here
+                    # Add to playlist button
+                    if st.button(f"Add '{song}' to Playlist", key=f"add_{song}"):
+                        st.session_state.playlist.append((song, video_url))
                         st.success(f"Added '{song}' to your playlist!")
                 else:
                     st.warning(f"Could not find '{song}' on YouTube.")
@@ -72,6 +86,16 @@ if user_input and gemini_api_key:
     else:
         st.warning("Please provide input to get song recommendations.")
 elif not user_input:
-    st.warning("Please enter your mood or context.")
+    st.warning("Please enter your mood and context.")
 elif not gemini_api_key:
-    st.warning("Please enter your Gemini API key to get started.")
+    st.warning("Please enter your Gemini API key to get started.")
+
+# Display the playlist
+st.sidebar.header("Your Playlist")
+if st.session_state.playlist:
+    for idx, (song, url) in enumerate(st.session_state.playlist):
+        st.sidebar.write(f"{idx + 1}. {song}")
+        if st.sidebar.button(f"Play {song}", key=f"play_{song}"):
+            st.video(url)
+else:
+    st.sidebar.write("Your playlist is empty.")