Debug

app.py

@@ -89,16 +89,12 @@ def process_csv_file(csv_file):
     except Exception as e:
         return None, f"Error processing file: {str(e)}"
 
-def get_current_data(csv_file):
-    """Get current data (either uploaded CSV or default)"""
-    if csv_file is None:
-        return create_default_timeseries(), "Using default time series with sudden increase at step 100"
-    else:
-        return process_csv_file(csv_file)
-
 def preview_csv(csv_file):
     """Preview uploaded CSV file immediately"""
-    df, message = get_current_data(csv_file)
+    if csv_file is None:
+        return gr.LinePlot(value=pd.DataFrame()), "Please upload a CSV file first", gr.Dropdown()
+    
+    df, message = process_csv_file(csv_file)
     
     if df is None:
         return gr.LinePlot(value=pd.DataFrame()), message, gr.Dropdown()
@@ -130,10 +126,10 @@ def preview_csv(csv_file):
 
 def update_plot(csv_file, selected_column):
     """Update plot based on selected column"""
-    if selected_column is None:
+    if csv_file is None or selected_column is None:
         return gr.LinePlot(value=pd.DataFrame())
     
-    df, _ = get_current_data(csv_file)
+    df, _ = process_csv_file(csv_file)
     if df is None:
         return gr.LinePlot(value=pd.DataFrame())
     
@@ -183,13 +179,18 @@ def infer_chatts_stream(prompt: str, csv_file):
     Streaming version of ChatTS inference
     """
     print("Start inferring!!!")
-        
+    
     if not prompt.strip():
         yield "Please enter a prompt"
         return
     
-    # Get current data (CSV or default)
-    df, error_msg = get_current_data(csv_file)
+    # Use default if no file uploaded, otherwise process CSV
+    if csv_file is None:
+        df = create_default_timeseries()
+        error_msg = None
+    else:
+        df, error_msg = process_csv_file(csv_file)
+    
     if df is None:
         yield f"Error: {error_msg}"
         return
@@ -215,6 +216,8 @@ def infer_chatts_stream(prompt: str, csv_file):
             prefix += f"The {name} is of length {len(arr)}: <ts><ts/>\n"
         full_prompt = prefix + formatted_prompt
 
+        print(f"[debug] {full_prompt=}, {len(ts_list)=}, {[len(item) for item in ts_list]=}")
+
         # Encode inputs
         inputs = processor(
             text=[full_prompt],
@@ -260,7 +263,7 @@ with gr.Blocks(title="ChatTS Demo") as demo:
             )
             
             prompt_input = gr.Textbox(
-                lines=4,
+                lines=6,
                 placeholder="Enter your analysis prompt here...",
                 label="Analysis Prompt",
                 value="Please analyze this time series and provide insights about the trends, seasonality, and local fluctuations."