Update app.py

app.py

@@ -29,59 +29,59 @@ def generate_cypher(user_query, query_intent, include_cobranded):
     print("inside cypher query gen")
 
     context_note = f"""
-Contextual Flags:
-- FD Card intent: {query_intent}
-- Include co-branded cards: {include_cobranded}
-"""
+        Contextual Flags:
+        - FD Card intent: {query_intent}
+        - Include co-branded cards: {include_cobranded}
+    """
 
     cypher_prompt = f"""
-You are an expert Neo4j Cypher query generator.
-
-Given a user’s question, graph schema, and **contextual flags**, generate the correct Cypher query.
-
-ONLY output the Cypher query. Do NOT explain anything.
-
----
-
-Graph Schema:
-- Nodes:
-  - (Card): Properties = name, bank_name, card_type, premium, co_branded
-  - (Feature): Properties = name
-- Relationships:
-  - (Card)-[:HAS_FEATURE]->(Feature)
-
-Valid values:
-- card_type: 'FD Card' or 'Regular'
-- premium: true (no concept of false — just include it if applicable)
-- co_branded: true (no concept of false — just include it if applicable)
-
-MANDATORY Condition Rules:
-- If FD Card intent is true → include: `c.card_type = 'FD Card'`
-- Else → include: `c.card_type = 'Regular'`
-- If the query uses words like "premium", "elite", "luxury", "exclusive", "infinia", "black", etc. → include: `AND c.premium = true`
-- If include co-branded is false → include: `AND (c.co_branded IS NULL OR c.co_branded = false)`
-- These conditions are **MANDATORY**. If they apply, include them in the `WHERE` clause. Do not skip them.
-
----
-
-Available features:
-"Dining Benefits", "Railway Benefits", "Railway Lounge", "Airport Lounge Access", "Travel Benefits", 
-"Fuel Surcharge Waiver", "Fuel Benefits", "General Cashback", "General Reward Points", "Movie Benefits", 
-"Daily Spends (Grocery)", "Utility", "Insurance & Health Benefits", "Forex Markup Fee", "Hotel Benefits", 
-"Welcome Bonus", "Air Miles", "No Forex Markup Fee", "Flight Discounts"
-
-Feature Inclusion Rules:
-- Only include relevant features based on user query.
-- Don’t add “General Cashback” or “General Reward Points” unless explicitly mentioned.
-- If fuel is mentioned, include both `Fuel Benefits` and `Fuel Surcharge Waiver`.
-
----
-
-{context_note}
-
-User Query: {user_query}
-Cypher:
-"""
+        You are an expert Neo4j Cypher query generator.
+        
+        Given a user’s question, graph schema, and **contextual flags**, generate the correct Cypher query. The query should retrurn only the cards c.
+        
+        ONLY output the Cypher query. Do NOT explain anything.
+        
+        ---
+        
+        Graph Schema:
+        - Nodes:
+          - (Card): Properties = name, bank_name, card_type, premium, co_branded
+          - (Feature): Properties = name
+        - Relationships:
+          - (Card)-[:HAS_FEATURE]->(Feature)
+        
+        Valid values:
+        - card_type: 'FD Card' or 'Regular'
+        - premium: true (no concept of false — just include it if applicable)
+        - co_branded: true (no concept of false — just include it if applicable)
+        
+        MANDATORY Condition Rules:
+        - If FD Card intent is true → include: `c.card_type = 'FD Card'`
+        - Else → include: `c.card_type = 'Regular'`
+        - If the query uses words like "premium", "elite", "luxury", "exclusive", "infinia", "black", etc. → include: `AND c.premium = true`
+        - If include co-branded is false → include: `AND (c.co_branded IS NULL OR c.co_branded = false)`
+        - These conditions are **MANDATORY**. If they apply, include them in the `WHERE` clause. Do not skip them.
+        
+        ---
+        
+        Available features:
+        "Dining Benefits", "Railway Benefits", "Railway Lounge", "Airport Lounge Access", "Travel Benefits", 
+        "Fuel Surcharge Waiver", "Fuel Benefits", "General Cashback", "General Reward Points", "Movie Benefits", 
+        "Daily Spends (Grocery)", "Utility", "Insurance & Health Benefits", "Forex Markup Fee", "Hotel Benefits", 
+        "Welcome Bonus", "Air Miles", "No Forex Markup Fee", "Flight Discounts"
+        
+        Feature Inclusion Rules:
+        - Only include relevant features based on user query.
+        - Don’t add “General Cashback” or “General Reward Points” unless explicitly mentioned.
+        - If fuel is mentioned, include both `Fuel Benefits` and `Fuel Surcharge Waiver`.
+        
+        ---
+        
+        {context_note}
+        
+        User Query: {user_query}
+        Cypher:
+    """
 
     response = model3.generate_content(cypher_prompt.strip())
     cypher_code = response.text.strip()