Update app.py

app.py

@@ -144,7 +144,7 @@ def predict_and_plot(velocity, temperature, precipitation, humidity):
 
 inputs = [
     gr.Slider(minimum=0, maximum=100, value=50, step=0.05, label="Velocity (mph)"),
-    gr.Slider(minimum=-2, maximum=30, value=0, step=0.5, label="Temperature (°C)"),
+    gr.Slider(minimum=-2, maximum 30, value=0, step=0.5, label="Temperature (°C)"),
     gr.Slider(minimum=0, maximum=1, value=0, step=0.01, label="Precipitation (inch)"),
     gr.Slider(minimum=0, maximum=100, value=50, label="Humidity (%)")
 ]
@@ -186,16 +186,13 @@ with gr.Blocks() as demo:
             gr.Markdown("### Input Parameters")
             for inp in inputs:
                 inp.render()
-            gr.Button(value="Submit", variant="primary").click(
-            fn=predict_and_plot, 
-            inputs=inputs, 
-            outputs= [gr.Plot(label="Contamination Levels Over Time")]+ contamination_outputs + gradients_outputs + cleaning_time_outputs)
-            gr.Button(value="Clear").click(fn=lambda: None)
+            submit_button = gr.Button(value="Submit", variant="primary")
+            clear_button = gr.Button(value="Clear")
 
         with gr.Column(scale=1):
             gr.Image(image_path)
 
-    # Bottom Section: Outputs (Three columns) and Plot Below
+    # Bottom Section: Outputs (Three columns)
     with gr.Row():
         with gr.Column(scale=2):
             gr.Markdown("### Contamination Predictions")
@@ -214,8 +211,13 @@ with gr.Blocks() as demo:
 
     # Graph below the outputs
     with gr.Row():
-        with gr.Column():
-            gr.Markdown("### Contamination Levels Over Time")
-            gr.Plot(label="Contamination Levels Over Time")
+        plot_output = gr.Plot(label="Contamination Levels Over Time")
+
+    submit_button.click(
+        fn=predict_and_plot, 
+        inputs=inputs, 
+        outputs=[plot_output] + contamination_outputs + gradients_outputs + cleaning_time_outputs
+    )
+    clear_button.click(fn=lambda: None)
 
 demo.launch()