Fix intent classification bug

app.py

@@ -533,23 +533,29 @@ def handle_query_classification(user_query):
     genai.configure(api_key='AIzaSyDGc7l82E1sa4GarTfGiTXC6dLD7Crk8oo')  
     model1 = genai.GenerativeModel('gemini-1.5-flash-latest')
     prompt = f"""
-You are a smart financial assistant.
-
-### User's Query:
-{user_query}
-
-### Task:
-Classify the user's intent into one of the following categories:
-1. "retrieve" → If the user is asking for card suggestions, recommendations, or showing cards (e.g., "suggest a card", "need a travel card").
-2. "specific" → If the user is asking about a particular card by name, the card names need not end with "card" keyword, so also consider that (e.g., "Tell me about HDFC Regalia", "Is SBI Elite good?").
-3. "no_retrieval" → If the query is generic, chit-chat, or doesn't need credit card data lookup (e.g., "What is interest in credit cards?").
-
-Respond in the following JSON format:
-{{
-  "intent": "retrieve" | "specific" | "no_retrieval",
-  "response": "Only include this if intent is 'no_retrieval'"
-}}
-"""
+        You are a smart financial assistant.
+        
+        ### User's Query:
+        {user_query}
+        
+        ### Task:
+        Classify the user's intent into one of the following categories:
+        1. "retrieve" → If the user is asking for card suggestions, recommendations, or showing cards (e.g., "suggest a card", "need a travel card") OR if they mention their lifestyle, income, spending, or needs (e.g., travel, shopping, fuel, rewards, luxury).
+        2. "specific" → If the user is asking about a particular credit card by name (even if the word "card" is not used). Examples: "Tell me about HDFC Regalia", "Is SBI Elite good?".
+        3. "no_retrieval" → ONLY if the query is generic (e.g., “What is credit score?”), casual chit-chat (e.g., “Hi”), or doesn’t mention any lifestyle, financial needs, or specific card names.
+
+        
+        Respond ONLY in the following JSON format:
+        If intent is "no_retrieval", you MUST include a helpful 'response' field.
+        If intent is "retrieve" or "specific", do NOT include any response or explanation.
+        
+        Respond in this exact format:
+        {{
+          "intent": "retrieve" | "specific" | "no_retrieval",
+          "response": "Only include this if intent is 'no_retrieval'"
+        }}
+        
+        """
 
     raw_response = model1.generate_content(prompt).text.strip()
 
@@ -566,9 +572,10 @@ Respond in the following JSON format:
         print("JSON parsing error:", e)
         print("Raw response from LLM:", raw_response)
         raise
-
-# print(handle_query_classification("what is a credit card"))
-
+# result = handle_query_classification("Want to optimize my spending – travel often, premium hotels, and online shopping.")
+# if result["intent"] == "no_retrieval":
+#     print(result['response'])
+    
 #passing the card mentioned in the user query
 def find_matching_card(user_query):
     lowered_query = user_query.lower()