Sync from GitHub: 3363d9befffa58a6417f0f97bfce1803aadacf1e

inference.py

@@ -65,19 +65,11 @@ Output rules:
 
 # Two-step Chain of Thought prompts (reasoning mode) - OPTIMIZED FOR SPEED
 REASONING_PROMPT = """
-Analyze this Indian tractor invoice and share your observations about extracting these 4 fields:
-1. dealer_name
-2. model_name  
-3. horse_power
-4. asset_cost
+Analyze this Indian tractor invoice and share your observations about extracting these 2 fields:
+'model name' and its corresponding 'Horse Power'
 
 Think through each field:
 
-DEALER NAME:
-- Where do you see the dealer/company name? Describe location and appearance,generally it is at top middle
-- What language is it in? Any spelling issues?
-- Write what you observe exactly as you see it
-
 MODEL NAME:
 - How is the model presented? (checkbox list, handwritten field, printed text)
 - If checkboxes exist, which one is marked/selected?
@@ -89,21 +81,18 @@ HORSE POWER:
 - Is it explicit (like "49 HP") or in a checkbox list?
 - If checkboxes, which HP value is selected?
 - What exact text shows the HP?
-
-ASSET COST:
-- Where is the final total amount located?
-- Which number is the final amount after all taxes?
-- What is the exact amount with currency symbol as shown?
+- Horse power must come ONLY from explicit HP text, never from model numbers.
+- Horse power may appear as "HP", handwritten like "49 HP", "63hp", "HP-30".
 
 Express your observations naturally. Be specific about what you see and any uncertainties.
 """
 
 
 EXTRACTION_WITH_CONTEXT_PROMPT = """
-Based on your analysis:
+Based on the image and following analysis regarding 'model':
 {reasoning_output}
 
-Extract these fields:
+Extract these fields from image and analysis:
 
 {{
   "dealer_name": string,
@@ -112,11 +101,19 @@ Extract these fields:
   "asset_cost": number
 }}
 
-Rules:
-1. DEALER/MODEL: Copy EXACTLY in original language, don't translate
-2. HP: Number only (e.g., "49 HP" → 49). Use selected checkbox if applicable
-3. ASSET COST: Final total as number (remove ₹, commas: "1,50,000" → 150000)
-4. Checkboxes: Extract only marked options
+Critical rules:
+
+- Dealer name must be copied exactly from the image in the original language and spelling.
+- Model name must be copied exactly from the image without translation.
+- HP: Number only (e.g., "49 HP" → 49). Use selected checkbox if applicable
+- ASSET COST: Final total as number (remove ₹, commas: "1,50,000" → 150000)
+- Checkboxes: Extract only marked options
+
+Extraction hints:
+- Asset cost is the total amount, usually the largest number on the page, the total amount after TAX, final price or final cost.
+- Dealer name is usually at the top header or company name.
+- Model name often appears near words like Model, Tractor, Variant.
+- If handwriting is unclear, make your best reasonable interpretation of the characters — but preserve language.
 
 Output ONLY valid JSON, no markdown.
 """