Update app.py

app.py

@@ -104,11 +104,34 @@ async def generate(image_id: str):
     }
 
     prompt = f"""
+You are an expense extraction AI.
+
 Extract expense details from the OCR text below:
 
+\"\"\"
 {full_text}
-
-Fill all fields accurately.
+\"\"\"
+
+### Labeling Rules:
+1. Detect the business/merchant name from the text (e.g., KFC, Starbucks, Ying Thai Kitchen).
+2. If items are food or restaurant-related → label must be: "<Business Name> Restaurant".
+3. If it's a store/retail → "<Business Name> Store".
+4. If unclear, infer the closest meaningful category.
+
+### Notes Format:
+Always generate notes EXACTLY in this format:
+"Spent <total_amount> on <label> on <date>."
+
+### Required Output:
+Return structured JSON (via schema) with:
+- total_amount
+- label (following rules above)
+- date
+- time
+- payment_type
+- notes
+
+Fill **every** field with the best possible inference.
 """
 
     try: