update to interactive plot

app.py

@@ -9,6 +9,8 @@ import gradio as gr
 import spaces
 import torch
 import pandas as pd
+import plotly.graph_objects as go
+from plotly.subplots import make_subplots
 
 # External imports
 
@@ -210,8 +212,6 @@ def create_test_dataloader(
     )
 
 def plot(ts_index, test_dataset, forecasts, prediction_length):
-    fig, ax = plt.subplots(figsize=(12, 8), facecolor='white')
-
     # Length of the target data
     target_length = len(test_dataset[ts_index]['target'])
 
@@ -223,26 +223,34 @@ def plot(ts_index, test_dataset, forecasts, prediction_length):
     ).to_timestamp()
 
     # Plotting actual data
-    ax.plot(
-        index[:target_length],
-        test_dataset[ts_index]['target'],
-        label="Actual"
+    actual_data = go.Scatter(
+        x=index[:target_length],
+        y=test_dataset[ts_index]['target'],
+        name="Actual",
+        mode='lines',
     )
 
     # Plotting the forecast data
-    # Forecast starts right after the last actual data point
-    forecast_start_index = target_length
-    ax.plot(
-        index[forecast_start_index:],
-        forecasts[ts_index][0][:prediction_length],  # Use forecasts[ts_index][0][:prediction_length] to slice the forecast values
-        label="Prediction"
+    forecast_data = go.Scatter(
+        x=index[target_length:],
+        y=forecasts[ts_index][0][:prediction_length],
+        name="Prediction",
+        mode='lines',
     )
 
-    ax.set_ylim(0, 140000)
-    ax.xaxis.set_major_locator(mdates.MonthLocator(bymonth=(1, 7)))
-    ax.xaxis.set_minor_locator(mdates.MonthLocator())
+    # Create the figure
+    fig = make_subplots(rows=1, cols=1)
+    fig.add_trace(actual_data, row=1, col=1)
+    fig.add_trace(forecast_data, row=1, col=1)
+
+    # Set layout and title
+    fig.update_layout(
+        xaxis_title="Date",
+        yaxis_title="Value",
+        title="Actual vs. Predicted Values",
+        xaxis_rangeslider_visible=True,
+    )
 
-    plt.legend()
     return fig
 
 def do_prediction(days_to_predict: int):

requirements.txt

@@ -118,6 +118,7 @@ parso==0.8.4
 pexpect==4.9.0
 pillow==10.3.0
 platformdirs==4.2.1
+plotly==5.22.0
 pretty-errors==1.2.25
 prometheus_client==0.20.0
 prompt-toolkit==3.0.43
@@ -165,6 +166,7 @@ spaces==0.28.0
 stack-data==0.6.3
 starlette==0.37.2
 sympy==1.12
+tenacity==8.3.0
 tensorboard==2.16.2
 tensorboard-data-server==0.7.2
 terminado==0.18.1