Update app.py

app.py

@@ -46,6 +46,12 @@ def extract_text(image):
     extracted_text = [line[1][0] for line in result[0]]
     return "\n".join(extracted_text)
 
+# Function to extract product name from attributes or use fuzzy matching
+def get_product_name(attributes, extracted_text):
+    if "Product name" in attributes and attributes["Product name"]:
+        return attributes["Product name"]
+    return match_product_name(extracted_text)
+
 # Function to match product name using fuzzy matching
 def match_product_name(extracted_text):
     best_match, best_score = None, 0
@@ -66,10 +72,6 @@ def extract_attributes(extracted_text):
         if match:
             attributes[readable_attr] = match.group(1).strip()
 
-    # Use extracted "Product name" if present; otherwise, apply fuzzy matching
-    if "Product name" not in attributes or not attributes["Product name"]:
-        attributes["Product name"] = match_product_name(extracted_text)
-
     # Extract Quantity from text (e.g., "Quantity: 10")
     quantity_match = re.search(r"\bQuantity[:\-]?\s*(\d+)", extracted_text, re.IGNORECASE)
     if quantity_match:
@@ -101,7 +103,7 @@ def interact_with_salesforce(mode, entry_type, quantity, extracted_text):
             return "❌ Invalid mode or entry type."
 
         attributes = extract_attributes(extracted_text)
-        product_name = attributes.get("Product name")
+        product_name = get_product_name(attributes, extracted_text)
 
         if not product_name:
             return "❌ No matching product found."