Update app.py

app.py

@@ -12,8 +12,7 @@ import traceback
 # ==============================================================
 # API Configuration - Add your key here
 # ==============================================================
-GEMINI_API_KEY = "AIzaSyDgNzVud08vXVrvFb5Mz0bHX8vBQ1LihNs"  # Replace with your key
-
+GEMINI_API_KEY = "AIzaSyDgNzVud08vXVrvFb5Mz0bHX8vBQ1LihNs"  
 # ==============================================================
 # Enhanced extraction prompt with better instructions
 # ==============================================================
@@ -38,16 +37,16 @@ Extract and structure the data as valid JSON only (no markdown, no commentary):
   "inventories": {
     "items": [
       {
-        "quantityShipped": number | null,
+        "quantityShipped": "Quantity as number, no of packages",
         "inventoryUnits": string | null,
         "pcs": number | null,
         "productName": string | null,
         "productCode": string | null,
         "product": {
-          "category": string | null,
+          "category": "Product category (OSB/Lumber/etc)",
           "defaultUnits": string | null,
-          "unit": number | null,
-          "pcs": number | null,
+          "unit": "Unit type from document (MBF, FBM, SF, UNIT etc.)",
+          "pcs": "pcs": "Pieces per unit",
           "mbf": number | null,
           "sf": number | null,
           "pcsHeight": number | null,
@@ -63,13 +62,14 @@ Extract and structure the data as valid JSON only (no markdown, no commentary):
 EXTRACTION RULES:
 1. Extract ALL product line items - create one inventory item per product
 2. Parse dimensions: "2X6X14" → pcsHeight=2, pcsWidth=6, pcsLength=14
-3. Convert BF to MBF: BF ÷ 1000
-4. customFields format: "Key||Value" (e.g., "Mill||Tolko")
-5. Look for: PO numbers, shipping info, quantities, product codes, dimensions
-6. Check headers, footers, stamps, handwritten notes, and table cells
-7. If multiple documents, consolidate all items into one JSON
-8. Return null for missing fields
-9. attachments should list all provided filenames
+3. BF = totalQuantity
+4. Convert BF to MBF: BF ÷ 1000
+5. customFields format: "Key||Value" (e.g., "Mill||Tolko")
+6. Look for: PO numbers, shipping info, quantities, product codes, dimensions
+7. Check headers, footers, stamps, handwritten notes, and table cells
+8. If multiple documents, consolidate all items into one JSON
+9. Return null for missing fields
+10.attachments should list all provided filenames
 
 Return ONLY valid JSON matching this exact structure."""