Update app.py

app.py

@@ -15,7 +15,7 @@ def process_image(img):
     try:
         # Convert Gradio image (PIL) to process
         if img is None:
-            return "No image provided", 0.0, "", None
+            return "No image provided", "", None
         
         # Resize if > 5MB
         img_bytes = img.tobytes()
@@ -33,8 +33,10 @@ def process_image(img):
         # Extract weight
         weight, confidence, unit = extract_weight_from_image(img)
         
-        # Return results
-        return f"{weight} {unit} (Confidence: {confidence:.2f}%)", f"Processed at {datetime.now(pytz.timezone('Asia/Kolkata')).strftime('%d-%m-%Y %I:%M:%S %p IST')}", img
+        # Return results with improved feedback
+        result = f"{weight} {unit} (Confidence: {confidence:.2f}%)" if confidence > 0 else "No weight detected"
+        timestamp = f"Processed at {datetime.now(pytz.timezone('Asia/Kolkata')).strftime('%d-%m-%Y %I:%M:%S %p IST')}"
+        return result, timestamp, img
     except Exception as e:
         logging.error(f"Error in process_image: {str(e)}")
         return f"Error: {str(e)}", "", None