Update app.py

app.py

@@ -4,6 +4,7 @@ from PIL import Image, ImageEnhance
 import gradio as gr
 import pandas as pd
 import re
+from fuzzywuzzy import process
 from simple_salesforce import Salesforce
 
 # Attribute mappings: readable names to Salesforce API names
@@ -118,13 +119,12 @@ def extract_text(image):
         extracted_text.append(line[1][0])
     return "\n".join(extracted_text)
 
-# Function to find product name from the predefined list
+# Function to find product name from the predefined list using fuzzy matching
 def match_product_name(text):
-    # Normalize text and product names for better matching
-    normalized_text = text.lower()
-    for product_name in PRODUCT_NAMES:
-        if product_name.lower() in normalized_text:
-            return product_name
+    # Normalize text and match against product names
+    best_match, score = process.extractOne(text, PRODUCT_NAMES)
+    if score >= 80:  # Match threshold
+        return best_match
     return None
 
 # Function to find attributes and their values
@@ -242,7 +242,7 @@ def process_image(image, quantity, mode):
             return "No attributes found in the image.", None
 
         numbered_output = "\n".join(
-            [f"{key.replace('__c','')}: {value}" for key, value in attributes.items()]
+            [f"{key.replace('__c', '').replace('_', ' ').title()}: {value}" for key, value in attributes.items()]
         )
 
         file_path = save_to_excel(attributes)
@@ -270,10 +270,9 @@ interface = gr.Interface(
         gr.Text(label="Image Data Viewer"),
         gr.File(label="Data Storage Manager")
     ],
-    title="Processing - VENKATARAMANA MOTORS",
-    description="Process images to update stock in Salesforce and save to Excel.",
+    title="Processing - VENKATA RAMANA MOTORS",
+    description="Process images for Entry or Exit mode to update stock.",
 )
 
 if __name__ == "__main__":
     interface.launch(share=True)
-