Update default time series

app.py

@@ -33,10 +33,17 @@ model.eval()
 
 def create_default_timeseries():
     """Create default time series with sudden increase"""
-    seq_len = 256
-    y = np.zeros(seq_len, dtype=np.float32)
-    y[100:] += 1
-    df = pd.DataFrame({"default_series": y})
+    x1 = np.arange(256)
+    x2 = np.arange(256)
+    ts1 = np.sin(x1 / 10) * 5.0
+    ts1[103:] -= 10.0
+    ts2 = x2 * 0.01
+    ts2[100] += 10.0
+
+    df = pd.DataFrame({
+        "TS1": ts1,
+        "TS2": ts2
+    })
     return df
 
 def process_csv_file(csv_file):
@@ -131,12 +138,15 @@ def clear_csv():
     return gr.LinePlot(value=pd.DataFrame()), message, gr.Dropdown()
 
 
-def update_plot(csv_file, selected_column):
+def update_plot(csv_file, selected_column, use_default_state):
     """Update plot based on selected column"""
-    if csv_file is None or selected_column is None:
+    if (csv_file is None and not use_default_state) or selected_column is None :
         return gr.LinePlot(value=pd.DataFrame())
     
-    df, _ = process_csv_file(csv_file)
+    if csv_file is None and use_default_state:
+        df = create_default_timeseries()
+    else:
+        df, _ = process_csv_file(csv_file)
     if df is None:
         return gr.LinePlot(value=pd.DataFrame())
     
@@ -174,7 +184,7 @@ def initialize_interface():
         label="Select a Column to Visualize"
     )
     
-    message = "Using default time series with sudden increase at step 100"
+    message = "Using default time series"
     
     return plot, message, dropdown, True  # Set use_default to True on initialization
 
@@ -333,7 +343,7 @@ with gr.Blocks(title="ChatTS Demo") as demo:
 
     series_selector.change(
         fn=update_plot,
-        inputs=[upload, series_selector],
+        inputs=[upload, series_selector, use_default_state],
         outputs=[plot_out]
     )
     
@@ -344,4 +354,4 @@ with gr.Blocks(title="ChatTS Demo") as demo:
     )
 
 if __name__ == '__main__':
-    demo.launch()
+    demo.launch()