Update app.py

app.py

@@ -176,20 +176,20 @@ cleaning_time_outputs = [
     gr.Textbox(label="Rear Cleaning Time")
 ]
 
-with gr.Blocks() as demo:
+with gr.Blocks(css=".column-container {height: 100%; display: flex; flex-direction: column; justify-content: space-between;}") 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(scale=2):
+        with gr.Column(scale=2, elem_classes="column-container"):
             gr.Markdown("### Input Parameters")
             for inp in inputs:
                 inp.render()
             submit_button = gr.Button(value="Submit", variant="primary")
             clear_button = gr.Button(value="Clear")
 
-        with gr.Column(scale=1, min_width=0, elem_id="center-column"):
+        with gr.Column(scale=1):
             gr.Image(image_path)
 
     # Bottom Section: Outputs (Three columns)