Update app.py

app.py

@@ -38,8 +38,26 @@ 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)
+
+        # System prompt to guide the AI
+        system_prompt = """
+        You are a music recommendation assistant specialized in Indian songs. Your task is to suggest popular Indian songs based on the user's mood.
+        - If the mood is "happy," suggest upbeat and joyful Bollywood or Indian pop songs.
+        - If the mood is "sad," suggest emotional and soulful Indian songs.
+        - If the mood is "energetic," suggest high-energy dance or workout songs.
+        - If the mood is "romantic," suggest romantic Bollywood or Indian love songs.
+        - If the mood is "calm," suggest soothing Indian classical or instrumental music.
+        Always suggest 3 songs and provide the song title and artist name in the format:
+        1. Song Title - Artist Name
+        2. Song Title - Artist Name
+        3. Song Title - Artist Name
+        """
+        
+        # User prompt
+        user_prompt = f"Suggest 3 popular Indian {mood} songs."
+        
+        # Combine system and user prompts
+        response = model.generate_content([system_prompt, user_prompt])
         return response.text
     except Exception as e:
         st.error(f"Error using Gemini API: {e}")