Update app.py

app.py

@@ -37,6 +37,35 @@ ATTRIBUTE_MAPPING = {
     "Usage": "Usage_Application__c",
 }
 
+# 🎨 **CSS Styling for UI**
+CSS_STYLE = """
+body {
+    background: linear-gradient(135deg, #141E30, #243B55);
+    color: white;
+    font-family: 'Poppins', sans-serif;
+}
+.gradio-container {
+    border-radius: 15px;
+    padding: 20px;
+    background: rgba(255, 255, 255, 0.2);
+}
+h1 {
+    text-align: center;
+    color: white;
+    text-shadow: 2px 2px 10px rgba(255, 255, 255, 0.8);
+}
+.gradio-button {
+    background: linear-gradient(to right, #ff416c, #ff4b2b);
+    color: white;
+    font-weight: bold;
+    border-radius: 10px;
+    padding: 12px;
+}
+.gradio-button:hover {
+    background: linear-gradient(to right, #ff4b2b, #ff416c);
+}
+"""
+
 # 🧠 Initialize PaddleOCR
 try:
     ocr = PaddleOCR(use_angle_cls=True, lang='en')
@@ -61,7 +90,7 @@ def match_product_name(extracted_text):
             best_match, best_score = match, score
     return best_match if best_score >= 70 else None  # Threshold: 70
 
-# Function to extract attributes from text
+# Function to extract attributes from text (including quantity)
 def extract_attributes(extracted_text):
     attributes = {}
     for readable_attr, sf_attr in ATTRIBUTE_MAPPING.items():
@@ -69,6 +98,12 @@ def extract_attributes(extracted_text):
         match = re.search(pattern, extracted_text, re.IGNORECASE)
         if match:
             attributes[readable_attr] = match.group(1).strip()
+    
+    # Extract Quantity from text (e.g., "Quantity: 10")
+    quantity_match = re.search(r"\bQuantity[:\-]?\s*(\d+)", extracted_text, re.IGNORECASE)
+    if quantity_match:
+        attributes["Quantity"] = quantity_match.group(1)
+
     return attributes
 
 # Function to export extracted data to Salesforce
@@ -80,20 +115,9 @@ def export_to_salesforce(mode, entry_type, quantity, extracted_text):
             security_token=SALESFORCE_SECURITY_TOKEN
         )
 
-        # Determine Object & Field Mapping
-        object_name, field_name = None, None
-
-        if mode == "Entry":
-            object_name = "Inventory_Management__c" if entry_type == "Sales" else "Un_Billable__c"
-            field_name = "Quantity__c"
-        elif mode == "Exit":
-            object_name = "Inventory_Management__c" if entry_type == "Sales" else "Un_Billable__c"
-            field_name = "Quantity_Sold__c"
-
-        if not object_name or not field_name:
-            return "❌ Invalid mode or entry type."
+        object_name = "Inventory_Management__c" if entry_type == "Sales" else "Un_Billable__c"
+        field_name = "Quantity__c" if mode == "Entry" else "Quantity_Sold__c"
 
-        # Extract Product Name & Attributes
         product_name = match_product_name(extracted_text)
         attributes = extract_attributes(extracted_text)
 
@@ -139,7 +163,7 @@ def process_image(image, mode, entry_type, quantity):
 
 # Gradio App
 def app():
-    with gr.Blocks() as interface:
+    with gr.Blocks(css=CSS_STYLE) as interface:
         gr.Markdown("<h1>🏢 VENKATARAMANA MOTORS</h1>")
 
         with gr.Tab("📥 Stock Entry & Processing"):
@@ -153,7 +177,7 @@ def app():
                 with gr.Column():
                     image_view = gr.Text(label="📜 Extracted Attributes", interactive=False)
                     result_output = gr.Text(label="📝 Salesforce Export Result", interactive=False)
-                    submit_button = gr.Button("🔍 Process & Export")
+                    submit_button = gr.Button("🔍 Process & Export", elem_classes="gradio-button")
 
         # Button Click Events
         submit_button.click(fn=process_image, inputs=[image_input, mode_dropdown, entry_type_radio, quantity_input], outputs=[image_view, result_output])