Update app.py

app.py

@@ -187,17 +187,18 @@ with gr.Blocks() as demo:
             for inp in inputs:
                 inp.render()
             # Submit and Clear Buttons under the inputs
-            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)
+            with gr.Row():
+                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)
 
         with gr.Column(scale=1):
             gr.Image(image_path)
 
-    # Bottom Section: Outputs (Three columns) and Plot Below
+    # Middle Section: Outputs (Three columns)
     with gr.Row():
         with gr.Column(scale=2):
             gr.Markdown("### Contamination Predictions")
@@ -214,7 +215,7 @@ with gr.Blocks() as demo:
             for out in cleaning_time_outputs:
                 out.render()
 
-    # Graph below the outputs
+    # Bottom Section: Graph at the very end
     with gr.Row():
         with gr.Column():
             gr.Markdown("### Contamination Levels Over Time")