Update app.py

app.py

@@ -34,7 +34,7 @@ def resize_image(img, max_size_mb=5):
         size_mb = len(img_bytes.getvalue()) / (1024 * 1024)
         if size_mb <= max_size_mb:
             return img, img_bytes.getvalue()
-
+        
         scale = 0.9
         while size_mb > max_size_mb:
             w, h = img.size
@@ -52,14 +52,14 @@ def resize_image(img, max_size_mb=5):
 def process_image(img):
     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 weight == "Not detected" or confidence < 70:
         return f"{weight} (Confidence: {confidence:.2f}%)", ist_time, img, None, gr.update(visible=True), gr.update(visible=False)
-
+    
     img_buffer = io.BytesIO(img_bytes)
     img_base64 = base64.b64encode(img_buffer.getvalue()).decode()
     return f"{weight} kg (Confidence: {confidence:.2f}%)", ist_time, img, img_base64, gr.update(visible=True), gr.update(visible=True)
@@ -69,10 +69,10 @@ def save_to_salesforce(weight_text, img_base64):
         sf = connect_to_salesforce()
         if sf is None:
             return "Failed to connect to Salesforce"
-
+        
         weight = float(weight_text.split(" ")[0])
         ist_time = datetime.now(pytz.timezone("Asia/Kolkata")).strftime("%Y-%m-%d %H:%M:%S")
-
+        
         record = {
             "Name": f"Weight_Log_{ist_time}",
             "Weight__c": weight,
@@ -122,4 +122,4 @@ with gr.Blocks(title="⚖️ Auto Weight Logger") as demo:
     - Works on desktop and mobile browsers.
     """)
 
-demo.launch()
+demo.launch()