Update app.py

app.py

@@ -62,19 +62,25 @@ def generate(prompt, history):
         return f"Error: {str(e)}"
 
 def format_prompt(message, history):
-    prompt = """You are a mood analyzer for music recommendations. Classify user mood as: Happy, Sad, Instrumental, or Party.
-
-Instructions:
-1. Chat with the user to understand their mood
-2. When clear, respond with ONLY one word: Happy, Sad, Instrumental, or Party
-3. If unclear, ask a follow-up question
-4. Maximum 5 exchanges
-
-"""
+    fixed_prompt = """
+    You are a smart mood analyzer tasked with determining the user's mood for a music recommendation system. Your goal is to classify the user's mood into one of four categories: Happy, Sad, Instrumental, or Party.
+    Instructions:
+    1. Engage in a conversation with the user to understand their mood.
+    2. Ask relevant questions to guide the conversation towards mood classification.
+    3. If the user's mood is clear, respond with a single word: "Happy", "Sad", "Instrumental", or "Party".
+    4. If the mood is unclear, continue the conversation with a follow-up question.
+    5. Limit the conversation to a maximum of 5 exchanges.
+    6. Do not classify the mood prematurely if it's not evident from the user's responses.
+    7. Focus on the user's emotional state rather than specific activities or preferences.
+    8. If unable to classify after 5 exchanges, respond with "Unclear" to indicate the need for more information.
+    Remember: Your primary goal is mood classification. Stay on topic and guide the conversation towards understanding the user's emotional state.
+    """
+    prompt = f"{fixed_prompt}\n"
     
-    for user_msg, bot_msg in history:
-        if user_msg.strip():
-            prompt += f"User: {user_msg}\nAssistant: {bot_msg}\n"
+    for i, (user_prompt, bot_response) in enumerate(history):
+        prompt += f"User: {user_prompt}\nAssistant: {bot_response}\n"
+        if i == 3:
+            prompt += "Note: This is the last exchange. Classify the mood if possible or respond with 'Unclear'.\n"
     
     prompt += f"User: {message}\nAssistant:"
     return prompt