Update app.py

app.py

@@ -1,23 +1,30 @@
 import gradio as gr
-from datetime import datetime
-import pytz
-from ocr_engine import extract_weight_from_image
+import cv2
+import pytesseract
 from PIL import Image
 import io
 import base64
+from datetime import datetime
+import pytz
 from simple_salesforce import Salesforce
 import logging
+import numpy as np
+import os
 
 # Set up logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-# Salesforce configuration (replace with your credentials)
-SF_USERNAME = "your_salesforce_username"
-SF_PASSWORD = "your_salesforce_password"
-SF_SECURITY_TOKEN = "your_salesforce_security_token"
-SF_DOMAIN = "login"  # or "test" for sandbox
+# Configure Tesseract path for Hugging Face
+pytesseract.pytesseract.tesseract_cmd = '/usr/bin/tesseract'
+
+# Salesforce configuration (use environment variables in production)
+SF_USERNAME = os.getenv("SF_USERNAME", "your_salesforce_username")
+SF_PASSWORD = os.getenv("SF_PASSWORD", "your_salesforce_password")
+SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN", "your_salesforce_security_token")
+SF_DOMAIN = os.getenv("SF_DOMAIN", "login")  # or "test" for sandbox
 
 def connect_to_salesforce():
+    """Connect to Salesforce with error handling."""
     try:
         sf = Salesforce(username=SF_USERNAME, password=SF_PASSWORD, security_token=SF_SECURITY_TOKEN, domain=SF_DOMAIN)
         logging.info("Connected to Salesforce successfully")
@@ -49,15 +56,44 @@ def resize_image(img, max_size_mb=5):
         logging.error(f"Image resizing failed: {str(e)}")
         return img, None
 
+def extract_weight(img):
+    """Extract weight from image using Tesseract OCR."""
+    try:
+        # Convert PIL image to OpenCV format
+        img_cv = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
+        gray = cv2.cvtColor(img_cv, cv2.COLOR_BGR2GRAY)
+        # Preprocess image for better OCR accuracy
+        _, thresh = cv2.threshold(gray, 150, 255, cv2.THRESH_BINARY)
+        # Configure Tesseract for 7-segment display (digits only, single line)
+        config = '--psm 7 digits'
+        text = pytesseract.image_to_string(thresh, config=config)
+        # Extract numeric values (digits and decimal point)
+        weight = ''.join(filter(lambda x: x in '0123456789.', text))
+        # Validate weight (ensure it’s a valid number)
+        try:
+            weight_float = float(weight)
+            # Simplified confidence: 95% if valid number, else 0%
+            confidence = 95.0 if weight_float > 0 else 0.0
+            return weight, confidence
+        except ValueError:
+            return "Not detected", 0.0
+    except Exception as e:
+        logging.error(f"OCR processing failed: {str(e)}")
+        return "Not detected", 0.0
+
 def process_image(img):
+    """Process uploaded or captured image and extract weight."""
     if img is None:
         return "No image uploaded", None, None, None, gr.update(visible=False), gr.update(visible=False)
     
     ist_time = datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%d-%m-%Y %I:%M:%S %p")
     img, img_bytes = resize_image(img)
-    weight, confidence = extract_weight_from_image(img)
+    if img_bytes is None:
+        return "Image processing failed", ist_time, img, None, gr.update(visible=False), gr.update(visible=False)
+    
+    weight, confidence = extract_weight(img)
     
-    if weight == "Not detected" or confidence < 70:
+    if weight == "Not detected" or confidence < 95.0:
         return f"{weight} (Confidence: {confidence:.2f}%)", ist_time, img, None, gr.update(visible=True), gr.update(visible=False)
     
     img_buffer = io.BytesIO(img_bytes)
@@ -65,6 +101,7 @@ def process_image(img):
     return f"{weight} kg (Confidence: {confidence:.2f}%)", ist_time, img, img_base64, gr.update(visible=True), gr.update(visible=True)
 
 def save_to_salesforce(weight_text, img_base64):
+    """Save weight and image to Salesforce Weight_Log__c object."""
     try:
         sf = connect_to_salesforce()
         if sf is None:
@@ -87,6 +124,7 @@ def save_to_salesforce(weight_text, img_base64):
         logging.error(f"Salesforce save failed: {str(e)}")
         return f"Failed to save to Salesforce: {str(e)}"
 
+# Gradio Interface
 with gr.Blocks(title="⚖️ Auto Weight Logger") as demo:
     gr.Markdown("## ⚖️ Auto Weight Logger")
     gr.Markdown("📷 Upload or capture an image of a digital weight scale (max 5MB).")
@@ -123,4 +161,5 @@ with gr.Blocks(title="⚖️ Auto Weight Logger") as demo:
     - Works on desktop and mobile browsers.
     """)
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()