Update app.py

app.py

@@ -180,25 +180,26 @@ with gr.Blocks() as demo:
     gr.Markdown("<h1 style='text-align: center;'>Environmental Factor-Based Contamination, Gradient, & Cleaning Time Prediction</h1>")
     gr.Markdown("This application predicts the contamination levels, gradients, and cleaning times for different parts of a car's LiDAR system based on environmental factors such as velocity, temperature, precipitation, and humidity.")
     
+    # Top Section: Inputs and Car Image
     with gr.Row():
-        with gr.Column():
+        with gr.Column(scale=2):
             gr.Markdown("### Input Parameters")
             for inp in inputs:
                 inp.render()
 
-            # Centered image display
-            with gr.Row():
-                with gr.Column(scale=1, min_width=0):
-                    gr.Image(image_path)  # Ensure the image is centered
-            
-            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)
-            
-    # Layout for outputs
+        with gr.Column(scale=1):
+            gr.Image(image_path)
+
+    # Middle Section: Submit and Clear Buttons
+    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)
+
+    # Bottom Section: Outputs (Three columns) and Plot Below
     with gr.Row():
         with gr.Column(scale=2):
             gr.Markdown("### Contamination Predictions")
@@ -210,13 +211,15 @@ with gr.Blocks() as demo:
             for out in gradients_outputs:
                 out.render()
 
-        with gr.Column(scale=3):
+        with gr.Column(scale=2):
             gr.Markdown("### Cleaning Time Predictions")
             for out in cleaning_time_outputs:
                 out.render()
 
-        with gr.Column(scale=5):
+    # Graph below the outputs
+    with gr.Row():
+        with gr.Column():
             gr.Markdown("### Contamination Levels Over Time")
-            gr.Plot(label="Contamination Levels Over Time")  # Removed .render()
+            gr.Plot(label="Contamination Levels Over Time")
 
 demo.launch()