Update app.py

app.py

@@ -200,19 +200,32 @@ def detect_defects(image):
         logging.error(f"Detection failed: {str(e)}")
         return None, f"Detection failed: {str(e)}"
 
-demo = gr.Interface(
-    fn=detect_defects,
-    inputs=gr.Image(type="pil", label="Upload Drone Image"),
-    outputs=[
-        gr.Image(label="Detection Result"),
-        gr.Textbox(label="Detected Faults with Severity")
-    ],
-    title="Structural Defect Detection with Salesforce Integration",
-    description=(
-        "Upload drone-captured images to detect structural defects like cracks, rust, spalling, "
-        "and deformations using Faster R-CNN. Detected faults are stored in Salesforce with annotated images."
+# Use gr.Blocks for more control over the UI
+with gr.Blocks() as demo:
+    gr.Markdown(
+        """
+        # Structural Defect Detection with Salesforce Integration
+        Upload drone-captured images to detect structural defects like cracks, rust, spalling, and deformations using Faster R-CNN. Detected faults are stored in Salesforce with annotated images.
+        """
+    )
+    with gr.Row():
+        image_input = gr.Image(type="pil", label="Upload Drone Image")
+        image_output = gr.Image(label="Detection Result")
+    output_text = gr.Textbox(label="Detected Faults with Severity")
+    with gr.Row():
+        clear_btn = gr.Button("Clear")
+        submit_btn = gr.Button("Submit", variant="primary")
+
+    submit_btn.click(
+        fn=detect_defects,
+        inputs=image_input,
+        outputs=[image_output, output_text]
+    )
+    clear_btn.click(
+        fn=lambda: (None, ""),
+        inputs=None,
+        outputs=[image_input, output_text]
     )
-)
 
 if __name__ == "__main__":
-    demo.launch(share=False)
+    demo.launch(share=False, allow_flagging="never")