sliders

app.py

@@ -43,9 +43,9 @@ model, feature_view, weather_fg, project = load_resources()
 # -------------------------
 # FORECAST LOGIC (NEXT 7 DAYS)
 # -------------------------
-def generate_forecast():
+def generate_forecast(days):
     today = datetime.utcnow().date()
-    future_dates = [(today + timedelta(days=i)).strftime("%Y-%m-%d") for i in range(1, 8)]
+    future_dates = [(today + timedelta(days=i)).strftime("%Y-%m-%d") for i in range(1, days + 1)]
 
     df_future = weather_fg.read()
     df_future["date"] = pd.to_datetime(df_future["date"]).dt.date
@@ -78,10 +78,10 @@ def generate_forecast():
 # -------------------------
 # HINDCAST LOGIC (LAST 7 DAYS)
 # -------------------------
-def generate_hindcast():
+def generate_hindcast(days):
     # Read actual + predicted from Feature View
     features_df, labels_df = feature_view.training_data(
-        start_time=datetime.utcnow().date() - timedelta(days=7),
+        start_time=datetime.utcnow().date() - timedelta(days=days)
         end_time=datetime.utcnow().date(),
         statistics_config=False
     )
@@ -110,22 +110,23 @@ def generate_hindcast():
 # -------------------------
 # GRADIO UI
 # -------------------------
-def run_dashboard(_):
-    forecast_plot = generate_forecast()
-    hindcast_plot = generate_hindcast()
-
+def run_dashboard(forecast_days, hindcast_days):
+    forecast_plot = generate_forecast(forecast_days)
+    hindcast_plot = generate_hindcast(hindcast_days)
     return forecast_plot, hindcast_plot
 
 
 iface = gr.Interface(
     fn=run_dashboard,
-    inputs=gr.Button("Generate Dashboard"),
+    inputs=[
+        gr.Slider(3, 10, value=7, step=1, label="Forecast Days (future)"),
+        gr.Slider(3, 10, value=7, step=1, label="Hindcast Days (past)"),
+        gr.Button("Generate")
+    ],
     outputs=[
-        gr.Image(label="PM2.5 Forecast (Next 7 Days)"),
-        gr.Image(label="PM2.5 Hindcast (Past 7 Days)")
+        gr.Image(label="PM2.5 Forecast"),
+        gr.Image(label="PM2.5 Hindcast")
     ],
     title="Air Quality Forecast Dashboard",
-    description="Forecast and Hindcast PM2.5 for Linköping using XGBoost + Hopsworks",
+    description="Forecast and hindcast PM2.5 for Linköping using XGBoost + Hopsworks"
 )
-
-iface.launch()