Update app.py

app.py

@@ -1,3 +1,22 @@
+
+import gradio as gr
+from PIL import Image
+from datetime import datetime
+import pytz
+from ocr_engine import extract_weight
+from simple_salesforce import Salesforce
+import base64
+import re
+import os
+
+# Salesforce credentials
+SF_USERNAME = "[email protected]"
+SF_PASSWORD = "autoweight@32"
+SF_TOKEN = "UgiHKWT0aoZRX9gvTYDjAiRY"
+
+# Connect to Salesforce
+sf = Salesforce(username=SF_USERNAME, password=SF_PASSWORD, security_token=SF_TOKEN)
+
 def process_image(image):
     if image is None:
         return "❌ No image provided", "", None, gr.update(visible=True)
@@ -8,11 +27,9 @@ def process_image(image):
         ist = pytz.timezone('Asia/Kolkata')
         timestamp = datetime.now(ist).strftime("%Y-%m-%d %H:%M:%S IST")
 
-        # Handle fallback errors
         if not weight or ("No valid" in weight and "OCR:" not in weight):
             return "❌ Unable to detect. Try again with a clearer image.", "", image, gr.update(visible=True)
 
-        # Extract number and unit
         numeric_match = re.search(r'\d{1,5}(?:\.\d{1,3})?', weight)
         if not numeric_match:
             return "❌ Could not extract number", "", image, gr.update(visible=True)
@@ -20,11 +37,9 @@ def process_image(image):
         numeric_value = float(numeric_match.group())
         unit = "kg" if "kg" in weight.lower() else "g"
 
-        # Save image
         image_path = "snapshot.jpg"
         image.save(image_path)
 
-        # Create Salesforce record
         record = sf.Weight_Log__c.create({
             "Captured_Weight__c": numeric_value,
             "Captured_Unit__c": unit,
@@ -33,7 +48,6 @@ def process_image(image):
             "Status__c": "Confirmed"
         })
 
-        # Upload image to Salesforce
         with open(image_path, "rb") as f:
             encoded_image = base64.b64encode(f.read()).decode("utf-8")
 
@@ -57,3 +71,29 @@ def process_image(image):
         return weight, timestamp, image, gr.update(visible=False)
     except Exception as e:
         return f"Error: {str(e)}", "", None, gr.update(visible=True)
+
+with gr.Blocks(css=".gr-button {background-color: #2e7d32 !important; color: white !important;}") as demo:
+    gr.Markdown("""
+    <h1 style='text-align: center; color: #2e7d32;'>📷 Auto Weight Logger</h1>
+    <p style='text-align: center;'>Upload or capture a digital weight image. Detects weight using AI OCR and logs it into Salesforce.</p>
+    <hr style='border: 1px solid #ddd;' />
+    """)
+
+    with gr.Row():
+        image_input = gr.Image(type="pil", label="📁 Upload or Capture Image")
+
+    detect_btn = gr.Button("🚀 Detect Weight")
+
+    with gr.Row():
+        weight_out = gr.Textbox(label="📦 Detected Weight", placeholder="e.g., 75.5 kg", show_copy_button=True)
+        time_out = gr.Textbox(label="🕒 Captured At (IST)", placeholder="e.g., 2025-07-01 12:00:00")
+
+    snapshot = gr.Image(label="📸 Snapshot Preview")
+    retake_btn = gr.Button("🔁 Retake / Try Again", visible=False)
+
+    detect_btn.click(fn=process_image, inputs=image_input, outputs=[weight_out, time_out, snapshot, retake_btn])
+    retake_btn.click(fn=lambda: ("", "", None, gr.update(visible=False)),
+                     inputs=[], outputs=[weight_out, time_out, snapshot, retake_btn])
+
+if __name__ == "__main__":
+    demo.launch()