Create app.py

app.py

@@ -0,0 +1,101 @@
+import streamlit as st
+import google.generativeai as genai
+from pytube import Search
+
+# Streamlit App
+st.title("🎶 Mood-Based Indian Song Player")
+
+# Sidebar for user input
+st.sidebar.header("Enter Your Input")
+
+# 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 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')
+        # Enhanced prompt with examples
+        prompt = f"""
+        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
+        2. Song Title - Artist Name
+        3. Song Title - Artist Name
+        """
+        response = model.generate_content(prompt)
+        return response.text
+    except Exception as e:
+        st.error(f"Error using Gemini API: {e}")
+        return None
+
+# Function to search YouTube for a song
+def search_youtube(query):
+    search = Search(query)
+    if search.results:
+        return search.results[0].watch_url
+    else:
+        return None
+
+# Main content
+if user_input and gemini_api_key:
+    # Get combined mood detection and song recommendations from Gemini API
+    response = get_song_recommendations(user_input, gemini_api_key)
+    if response:
+        # Parse mood and song recommendations
+        lines = response.split('\n')
+        if len(lines) > 1 and lines[0].startswith("Mood:"):
+            mood = lines[0].replace("Mood: ", "")
+            st.write(f"Detected Mood/Context: {mood}")
+            songs = [line.strip() for line in lines[1:4] if line.strip()]
+            st.write("🎵 Recommended Songs:")
+            for song in songs:
+                st.write(song)
+                # Search YouTube and provide playback options
+                query = f"{song} song"
+                video_url = search_youtube(query)
+                if video_url:
+                    st.video(video_url)
+                    # 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.")
+        else:
+            st.error("Gemini API response format is incorrect.")
+    else:
+        st.warning("Please provide input to get song recommendations.")
+elif not user_input:
+    st.warning("Please enter your mood and context.")
+elif not gemini_api_key:
+    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.")