Update app.py

app.py

@@ -95,6 +95,16 @@ with gr.Blocks() as demo:
             seg_out = gr.Image(label="Gaussian Blurred Image")
             seg_button = gr.Button("Process Gaussian Blur")
             seg_button.click(process_segmentation_blur, inputs=[seg_img, seg_blur], outputs=seg_out)
+        with gr.Tab("Depth Heatmap"):
+            img_input2 = gr.Image(type="pil", label="Upload Image")
+            intensity_slider = gr.Slider(0.5, 5.0, value=1.0, label="Heatmap Intensity")
+            heatmap_output = gr.Image(label="Depth Heatmap")
+            heatmap_button = gr.Button("Generate Depth Heatmap")
+            heatmap_button.click(
+                process_depth_heatmap, 
+                inputs=[img_input2, intensity_slider], 
+                outputs=heatmap_output
+            )
         with gr.Tab("Lens Blur"):
             img_input = gr.Image(type="pil", label="Upload Image")
             blur_slider = gr.Slider(1, 50, value=6, label="Maximum Blur Radius")