Update app.py

app.py

@@ -16,23 +16,29 @@ device = None
 genai.configure(api_key='AIzaSyApqa65vVYTmw4FC4wP-6-_xpBLxXdctxE')
 gemini_model = genai.GenerativeModel("gemini-2.0-flash")
 
-PARSING_PROMPT = """Parse the user's counting instruction into two lists:
-- A (include): objects to count
-- B (exclude): objects to exclude from counting
+PARSING_PROMPT = """Parse sentences of the form "Count A, not B" into two lists—A (include) and B (exclude)—splitting on "and", "or", and commas, and reattaching shared head nouns (e.g., "red and black beans" → "red beans", "black beans").
 
 Rules:
-1. Split on "and", "or", and commas
-2. Reattach shared head nouns (e.g., "red and black beans" → "red beans", "black beans")
-3. Remove from B items that are equivalent to A (synonyms/variants/abbreviations)
-4. Remove from B items that are more specific than A
-5. If B is more general than A but shares head noun, rewrite B to specific non-overlapping forms
-
-Examples:
-- "Count green apples, not red apples" → A: ["green apples"], B: ["red apples"]
-- "Count apples, not green apples" → A: ["apples"], B: []
-- "Count green apples, not apples" → A: ["green apples"], B: ["non-green apples"]
-- "Count fries, not chips" → A: ["fries"], B: []
-- "Count black beans, not poker chips" → A: ["black beans"], B: ["poker chips"]
+- Remove from B items that are equivalent to items in A (synonyms/variants/abbreviations/regional terms)
+- Keep B items that are more specific than A (for fine-grained exclusion)
+- If B is more general than A but shares the head noun, remove B (contradictory)
+
+Case 1 — Different head nouns → Keep B
+Example 1: Count green apples and red beans, not yellow screws and white rice → A: ["green apples", "red beans"], B: ["yellow screws", "white rice"]
+Example 2: Count black beans, not poker chips or nails → A: ["black beans"], B: ["poker chips", "nails"]
+
+Case 2 — Equivalent items → Remove from B
+Example 1: Count fries and TV, not chips and television → A: ["fries", "TV"], B: []
+Example 2: Count garbanzo beans and couch, not chickpeas and sofa → A: ["garbanzo beans", "couch"], B: []
+
+Case 3 — B more specific than A → Keep B (for fine-grained exclusion)
+Example 1: Count apples and beans, not green apples and black beans → A: ["apples", "beans"], B: ["green apples", "black beans"]
+Example 2: Count beans, not white beans or yellow beans → A: ["beans"], B: ["white beans", "yellow beans"]
+Example 3: Count people, not women → A: ["people"], B: ["women"]
+
+Case 4 — B more general than A → Remove B (contradictory)
+Example 1: Count green apples, not apples → A: ["green apples"], B: []
+Example 2: Count red beans and green apples, not beans and apples → A: ["red beans", "green apples"], B: []
 
 User instruction: {instruction}
 
@@ -40,7 +46,6 @@ Respond ONLY with a JSON object in this exact format, no other text:
 {{"A": ["item1", "item2"], "B": ["item3"]}}
 """
 
-
 def parse_counting_instruction(instruction: str) -> tuple[str, str]:
     """
     Parse natural language counting instruction using Gemini 2.0 Flash.