Update angle_detection_app.py

angle_detection_app.py

@@ -78,6 +78,9 @@ def process_image(
     """
     Process the image and return the visualization and angle measurements.
     """
+    if image is None:
+        return None, "Please upload an image."
+    
     if isinstance(image, dict):
         image = image['image']
     if isinstance(image, str):
@@ -109,39 +112,28 @@ def process_image(
     return result_img, measurements
 
 def create_gradio_interface():
-    with gr.Blocks(title="Angle Detection App", theme=gr.themes.Soft()) as interface:
-        gr.Markdown("# Angle Detection App")
-        gr.Markdown("Upload an image to detect bends and measure angles.")
-        
-        with gr.Row():
-            with gr.Column():
-                input_image = gr.Image(label="Input Image", type="numpy")
-                
-                with gr.Accordion("Algorithm Parameters", open=False):
-                    blur_kernel_size = gr.Slider(minimum=3, maximum=15, step=2, value=7, label="Blur Kernel Size")
-                    canny_threshold1 = gr.Slider(minimum=0, maximum=100, step=10, value=30, label="Canny Threshold 1")
-                    canny_threshold2 = gr.Slider(minimum=100, maximum=300, step=10, value=150, label="Canny Threshold 2")
-                    dilation_kernel_size = gr.Slider(minimum=1, maximum=5, step=1, value=2, label="Dilation Kernel Size")
-                    hough_threshold = gr.Slider(minimum=10, maximum=100, step=10, value=50, label="Hough Threshold")
-                    min_line_length = gr.Slider(minimum=5, maximum=50, step=5, value=10, label="Minimum Line Length")
-                    max_line_gap = gr.Slider(minimum=10, maximum=100, step=10, value=60, label="Maximum Line Gap")
-                    bend_threshold = gr.Slider(minimum=5, maximum=30, step=5, value=15, label="Bend Threshold")
-                
-                process_btn = gr.Button("Process Image", variant="primary")
-            
-            with gr.Column():
-                output_image = gr.Image(label="Result")
-                output_text = gr.Textbox(label="Measurements", lines=10)
-        
-        process_btn.click(
-            fn=process_image,
-            inputs=[input_image, blur_kernel_size, canny_threshold1, canny_threshold2, dilation_kernel_size,
-                    hough_threshold, min_line_length, max_line_gap, bend_threshold],
-            outputs=[output_image, output_text]
-        )
-        
+    interface = gr.Interface(
+        fn=process_image,
+        inputs=[
+            gr.Image(label="Input Image", type="numpy"),
+            gr.Slider(minimum=3, maximum=15, step=2, value=7, label="Blur Kernel Size"),
+            gr.Slider(minimum=0, maximum=100, step=10, value=30, label="Canny Threshold 1"),
+            gr.Slider(minimum=100, maximum=300, step=10, value=150, label="Canny Threshold 2"),
+            gr.Slider(minimum=1, maximum=5, step=1, value=2, label="Dilation Kernel Size"),
+            gr.Slider(minimum=10, maximum=100, step=10, value=50, label="Hough Threshold"),
+            gr.Slider(minimum=5, maximum=50, step=5, value=10, label="Minimum Line Length"),
+            gr.Slider(minimum=10, maximum=100, step=10, value=60, label="Maximum Line Gap"),
+            gr.Slider(minimum=5, maximum=30, step=5, value=15, label="Bend Threshold")
+        ],
+        outputs=[
+            gr.Image(label="Result"),
+            gr.Textbox(label="Measurements", lines=10)
+        ],
+        title="Angle Detection App",
+        description="Upload an image to detect bends and measure angles. Adjust parameters as needed."
+    )
     return interface
 
 if __name__ == "__main__":
     interface = create_gradio_interface()
-    interface.launch(share=False)  # Set share=False to avoid issues when deploying
+    interface.launch(share=True)