working ui

app.py

@@ -179,22 +179,42 @@ def generate_hindcast_v2(days):
 
 
 with gr.Blocks() as iface:
+    
+    gr.Markdown("# Air Quality Dashboard (Model v1 & Model v2)")
 
-    gr.Markdown("# Air Quality Forecast Dashboard")
-    gr.Markdown("Choose between Model v1 (no lags) and Model v2 (lag-aware).")
+    with gr.Row():
+        gr.Markdown("### **Model v1 (No lag features)**")
+        gr.Markdown("### **Model v2 (Lag-aware)**")
 
     with gr.Row():
-        forecast_days_v1 = gr.Slider(3, 10, value=7, label="Forecast Days (v1)")
-        forecast_days_v2 = gr.Slider(3, 10, value=7, label="Forecast Days (v2)")
+        days_v1_f = gr.Slider(3, 10, value=7, label="Forecast Days (v1)")
+        days_v2_f = gr.Slider(3, 10, value=7, label="Forecast Days (v2)")
 
     with gr.Row():
-        btn_v1 = gr.Button("Generate Forecast (Model v1)")
-        btn_v2 = gr.Button("Generate Forecast (Model v2)")
+        btn_v1_f = gr.Button("Generate Forecast (v1)")
+        btn_v2_f = gr.Button("Generate Forecast (v2)")
+
+    out_v1_f = gr.Image()
+    out_v2_f = gr.Image()
+
+    btn_v1_f.click(generate_forecast_v1, inputs=days_v1_f, outputs=out_v1_f)
+    btn_v2_f.click(generate_forecast_v2, inputs=days_v2_f, outputs=out_v2_f)
+
+
+    # HINDCAST
+    with gr.Row():
+        days_v1_h = gr.Slider(3, 10, value=7, label="Hindcast Days (v1)")
+        days_v2_h = gr.Slider(3, 10, value=7, label="Hindcast Days (v2)")
+
+    with gr.Row():
+        btn_v1_h = gr.Button("Generate Hindcast (v1)")
+        btn_v2_h = gr.Button("Generate Hindcast (v2)")
+
+    out_v1_h = gr.Image()
+    out_v2_h = gr.Image()
 
-    out_v1 = gr.Image(label="Model v1 Forecast")
-    out_v2 = gr.Image(label="Model v2 Forecast")
+    btn_v1_h.click(generate_hindcast_v1, inputs=days_v1_h, outputs=out_v1_h)
+    btn_v2_h.click(generate_hindcast_v2, inputs=days_v2_h, outputs=out_v2_h)
 
-    btn_v1.click(fn=generate_forecast_v1, inputs=forecast_days_v1, outputs=out_v1)
-    btn_v2.click(fn=generate_forecast_v2, inputs=forecast_days_v2, outputs=out_v2)
 
 iface.launch()