Update app.py

app.py

@@ -198,19 +198,25 @@ with gr.Blocks() as demo:
             )
             gr.Button(value="Clear").click(fn=lambda: None)
             
-        with gr.Column():
+    # Layout for outputs
+    with gr.Row():
+        with gr.Column(scale=2):
             gr.Markdown("### Contamination Predictions")
             for out in contamination_outputs:
                 out.render()
-                
-        with gr.Column():
+
+        with gr.Column(scale=2):
             gr.Markdown("### Gradient Predictions")
             for out in gradients_outputs:
                 out.render()
 
-        with gr.Column():
+        with gr.Column(scale=3):
             gr.Markdown("### Cleaning Time Predictions")
             for out in cleaning_time_outputs:
                 out.render()
 
+        with gr.Column(scale=5):
+            gr.Markdown("### Contamination Levels Over Time")
+            gr.Plot(label="Contamination Levels Over Time").render()
+
 demo.launch()