Update app.py

app.py

@@ -1,21 +1,3 @@
-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 os
-import re
-
-# 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)
@@ -26,7 +8,7 @@ def process_image(image):
         ist = pytz.timezone('Asia/Kolkata')
         timestamp = datetime.now(ist).strftime("%Y-%m-%d %H:%M:%S IST")
 
-        # Handle missing or invalid output
+        # 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)
 
@@ -42,16 +24,16 @@ def process_image(image):
         image_path = "snapshot.jpg"
         image.save(image_path)
 
-        # Create Weight_Log__c record
+        # Create Salesforce record
         record = sf.Weight_Log__c.create({
             "Captured_Weight__c": numeric_value,
-            "Captured_Unit__c": unit,  # ✅ Must exist in Salesforce
+            "Captured_Unit__c": unit,
             "Captured_At__c": datetime.now(ist).isoformat(),
             "Device_ID__c": "DEVICE-001",
             "Status__c": "Confirmed"
         })
 
-        # Upload image
+        # Upload image to Salesforce
         with open(image_path, "rb") as f:
             encoded_image = base64.b64encode(f.read()).decode("utf-8")
 
@@ -75,28 +57,3 @@ 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])
-
-demo.launch()