Update app.py

app.py

@@ -0,0 +1,103 @@
+import streamlit as st
+from transformers import pipeline
+import google.generativeai as genai
+from pytube import Search
+import speech_recognition as sr
+
+# Load sentiment analysis model
+mood_classifier = pipeline("sentiment-analysis")
+
+# Speech-to-text function
+def speech_to_text():
+    recognizer = sr.Recognizer()
+    with sr.Microphone() as source:
+        st.write("🎤 Speak now...")
+        audio = recognizer.listen(source)
+        try:
+            text = recognizer.recognize_google(audio)
+            st.write(f"🎤 You said: **{text}**")
+            return text
+        except sr.UnknownValueError:
+            st.error("Sorry, I could not understand your audio.")
+            return None
+        except sr.RequestError:
+            st.error("Sorry, there was an issue with the speech recognition service.")
+            return None
+
+# Functions
+def detect_mood(text):
+    result = mood_classifier(text)[0]
+    if result['label'] == 'POSITIVE':
+        return "joyful"
+    elif result['label'] == 'NEGATIVE':
+        return "sad"
+    else:
+        return "neutral"
+
+def get_song_recommendations(mood, api_key):
+    try:
+        genai.configure(api_key=api_key)
+        model = genai.GenerativeModel('gemini-pro')
+        prompt = f"Suggest 3 popular Indian {mood} songs."
+        response = model.generate_content(prompt)
+        return response.text
+    except Exception as e:
+        st.error(f"Error using Gemini API: {e}")
+        return None
+
+def search_youtube(query):
+    search = Search(query)
+    return search.results[0].watch_url
+
+# Streamlit App
+st.title("🎵 Mood-Based Indian Song Player")
+
+# Sidebar for user input
+st.sidebar.header("How are you feeling today?")
+mood_options = ["happy", "sad", "energetic", "romantic", "calm"]
+
+# Input for Gemini API key
+gemini_api_key = st.sidebar.text_input("Enter your Gemini API Key:", type="password")
+
+# Add a button for speech input
+if st.sidebar.button("🎤 Speak your mood"):
+    user_mood = speech_to_text()
+else:
+    user_mood = st.sidebar.selectbox("Or select your mood:", mood_options)
+
+# Playlist
+if 'playlist' not in st.session_state:
+    st.session_state.playlist = []
+
+# Main content
+if user_mood and gemini_api_key:
+    mood = detect_mood(user_mood)
+    st.write(f"🎭 Detected Mood: **{mood}**")
+    
+    st.write("🎵 Recommended Songs:")
+    recommendations = get_song_recommendations(mood, gemini_api_key)
+    if recommendations:
+        st.write(recommendations)
+        
+        song_names = recommendations.split("\n")
+        for song in song_names:
+            if song.strip():
+                st.write(f"🔍 Searching for: **{song}**")
+                video_url = search_youtube(song)
+                st.video(video_url)
+                
+                # Add to playlist button
+                if st.button(f"Add '{song}' to Playlist"):
+                    st.session_state.playlist.append((song, video_url))
+                    st.success(f"Added '{song}' to your playlist!")
+    else:
+        st.error("Failed to get song recommendations. Please check your API key.")
+elif not gemini_api_key:
+    st.warning("Please enter your Gemini API key to get started.")
+
+# Display the playlist
+st.sidebar.header("Your 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_{idx}"):
+        st.video(url)