Update app.py

app.py

@@ -35,7 +35,6 @@ def resize_image(img, max_size_mb=5):
         if size_mb <= max_size_mb:
             return img, img_bytes.getvalue()
         
-        # Resize proportionally
         scale = 0.9
         while size_mb > max_size_mb:
             w, h = img.size
@@ -58,13 +57,12 @@ def process_image(img):
     img, img_bytes = resize_image(img)
     weight, confidence = extract_weight_from_image(img)
     
-    if weight == "Not detected" or confidence < 70:
-        return f"{weight} (Confidence: {confidence}%)", ist_time, img, None, gr.update(visible=True), gr.update(visible=False)
+    if weight == "Not detected" or confidence < 80:
+        return f"{weight} (Confidence: {confidence:.2f}%)", ist_time, img, None, gr.update(visible=True), gr.update(visible=False)
     
-    # Encode image for preview
     img_buffer = io.BytesIO(img_bytes)
     img_base64 = base64.b64encode(img_buffer.getvalue()).decode()
-    return f"{weight} kg (Confidence: {confidence}%)", ist_time, img, img_base64, gr.update(visible=True), gr.update(visible=True)
+    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):
     try:
@@ -72,11 +70,9 @@ def save_to_salesforce(weight_text, img_base64):
         if sf is None:
             return "Failed to connect to Salesforce"
         
-        # Extract weight from text (remove "kg" and confidence)
         weight = float(weight_text.split(" ")[0])
         ist_time = datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S")
         
-        # Create custom object record (adjust object and fields as needed)
         record = {
             "Name": f"Weight_Log_{ist_time}",
             "Weight__c": weight,
@@ -123,7 +119,7 @@ with gr.Blocks(title="⚖️ Auto Weight Logger") as demo:
     - Upload a clear, well-lit image of a digital weight scale display.
     - Ensure the image is < 5MB (automatically resized if larger).
     - Review the detected weight and click 'Confirm and Save to Salesforce' to log the data.
-    - The application works on both desktop and mobile browsers.
+    - Works on desktop and mobile browsers.
     """)
 
 demo.launch()