Update app.py

app.py

@@ -6,19 +6,23 @@ from ocr_engine import extract_weight
 
 def process_image(image):
     if image is None:
-        return "No image provided", "", None
+        return "No image provided", "", None, gr.update(visible=True)
     try:
         weight = extract_weight(image)
         ist = pytz.timezone('Asia/Kolkata')
         timestamp = datetime.now(ist).strftime("%Y-%m-%d %H:%M:%S IST")
-        return weight, timestamp, image
+
+        if not weight:
+            return "❌ Unable to detect. Try again with clearer image.", "", image, gr.update(visible=True)
+
+        return weight, timestamp, image, gr.update(visible=False)
     except Exception as e:
-        return f"Error: {str(e)}", "", None
+        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;'>Detect weights (kg or grams) from digital balance display using AI OCR.</p>
+    <p style='text-align: center;'>Upload or capture a digital weight image. Detects weight with AI OCR.</p>
     <hr style='border: 1px solid #ddd;'/>
     """)
 
@@ -32,12 +36,14 @@ with gr.Blocks(css=".gr-button {background-color: #2e7d32 !important; color: whi
         time_out = gr.Textbox(label="🕒 Captured At (IST)", placeholder="e.g., 2025-06-30 14:32:10")
 
     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])
+    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])
 
     gr.Markdown("""
-    <br><p style='text-align: center; color: gray;'>Developed by Shalu • Powered by Hugging Face OCR 🚀</p>
+    <p style='text-align: center; color: gray;'>Tip: Ensure image is glare-free and focused. <br>Developed by Shalu • Hugging Face OCR ⚡</p>
     """)
 
-# 🚨 REQUIRED for Hugging Face to recognize the app
 demo.launch()