Update app.py

app.py

@@ -2,54 +2,52 @@ import gradio as gr
 from PIL import Image
 from datetime import datetime
 import pytz
-import os
-from simple_salesforce import Salesforce
 from ocr_engine import extract_weight
-from dotenv import load_dotenv
+from simple_salesforce import Salesforce
+import base64
+import os
 
-# Load environment variables
-load_dotenv()
+# Salesforce credentials (for safety, you can later store these in environment variables)
+SF_USERNAME = "[email protected]"
+SF_PASSWORD = "autoweight@32"
+SF_TOKEN = "UgiHKWT0aoZRX9gvTYDjAiRY"
 
-# Salesforce connection
-sf = Salesforce(
-    username=os.getenv("SF_USERNAME"),
-    password=os.getenv("SF_PASSWORD"),
-    security_token=os.getenv("SF_TOKEN")
-)
+# 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)
-
     try:
         weight = extract_weight(image)
         ist = pytz.timezone('Asia/Kolkata')
-        now = datetime.now(ist)
-        timestamp_display = now.strftime("%Y-%m-%d %H:%M:%S IST")
-        timestamp_iso = now.isoformat()
-
+        timestamp = datetime.now(ist).strftime("%Y-%m-%d %H:%M:%S IST")
         if not weight or "No valid" in weight:
             return "❌ Unable to detect. Try again with a clearer image.", "", image, gr.update(visible=True)
 
-        # Create Salesforce record
+        # Save snapshot image to base64 (optional field, depends on usage)
+        buffered_image = image.copy()
+        buffered_image.save("snapshot.jpg")
+        with open("snapshot.jpg", "rb") as img_file:
+            snapshot_base64 = base64.b64encode(img_file.read()).decode('utf-8')
+
+        # Create record in Salesforce
         sf.Weight_Log__c.create({
-            'Captured_Weight__c': float(weight.replace("kg", "").strip()),
-            'Captured_At__c': timestamp_iso,
-            'Device_ID__c': 'Device-01',  # Static or dynamic device ID
-            'Snapshot_Image__c': 'Manual Entry',
-            'Status__c': 'Captured'
+            "Captured_Weight__c": float(weight.replace("kg", "").strip()),
+            "Captured_At__c": datetime.now(ist).isoformat(),
+            "Snapshot_Image__c": snapshot_base64,
+            "Device_ID__c": "DEVICE-001",   # Static or dynamic based on your design
+            "Status__c": "Captured"
         })
 
-        return weight, timestamp_display, image, gr.update(visible=False)
-
+        return weight, timestamp, image, gr.update(visible=False)
     except Exception as e:
         return f"Error: {str(e)}", "", None, gr.update(visible=True)
 
-# Gradio UI
 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 stores the result in Salesforce.</p>
+    <p style='text-align: center;'>Upload or capture a digital weight image. Detects weight using AI OCR.</p>
     <hr style='border: 1px solid #ddd;'/>
     """)