Fix bugs

app.py

@@ -89,15 +89,15 @@ def process_csv_file(csv_file):
     except Exception as e:
         return None, f"Error processing file: {str(e)}"
 
-def preview_csv(csv_file):
+def preview_csv(csv_file, use_default):
     """Preview uploaded CSV file immediately"""
     if csv_file is None:
-        return gr.LinePlot(value=pd.DataFrame()), "Please upload a CSV file first", gr.Dropdown()
+        return gr.LinePlot(value=pd.DataFrame()), "Please upload a CSV file first", gr.Dropdown(), False
     
     df, message = process_csv_file(csv_file)
     
     if df is None:
-        return gr.LinePlot(value=pd.DataFrame()), message, gr.Dropdown()
+        return gr.LinePlot(value=pd.DataFrame()), message, gr.Dropdown(), False
     
     # Create dropdown choices
     column_choices = list(df.columns)
@@ -122,7 +122,14 @@ def preview_csv(csv_file):
     
     print("Successfully generated preview!")
     
-    return plot, message, dropdown
+    return plot, message, dropdown, False  # Set use_default to False when file is uploaded
+
+def clear_csv():
+    """Clear uploaded CSV file immediately"""
+    df, message = process_csv_file(None)
+    
+    return gr.LinePlot(value=pd.DataFrame()), message, gr.Dropdown()
+
 
 def update_plot(csv_file, selected_column):
     """Update plot based on selected column"""
@@ -169,12 +176,12 @@ def initialize_interface():
     
     message = "Using default time series with sudden increase at step 100"
     
-    return plot, message, dropdown
+    return plot, message, dropdown, True  # Set use_default to True on initialization
 
 # ─── INFERENCE + VALIDATION ────────────────────────────────────────────────────
 
 @spaces.GPU  # dynamically allocate & release a ZeroGPU device on each call
-def infer_chatts_stream(prompt: str, csv_file):
+def infer_chatts_stream(prompt: str, csv_file, use_default):
     """
     Streaming version of ChatTS inference
     """
@@ -184,15 +191,15 @@ def infer_chatts_stream(prompt: str, csv_file):
         yield "Please enter a prompt"
         return
     
-    # Use default if no file uploaded, otherwise process CSV
-    if csv_file is None:
+    # Use default if no file uploaded and use_default is True
+    if csv_file is None and use_default:
         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}"
+        yield "Please upload a CSV file first or the file contains errors"
         return
     
     try:
@@ -206,17 +213,21 @@ def infer_chatts_stream(prompt: str, csv_file):
                 ts_names.append(name)
                 ts_list.append(trimmed)
         
-        # Add chat_template
+        if not ts_list:
+            yield "No valid time series data found. Please upload time series first."
+            return
+        
+        # Clean prompt
         clean_prompt = prompt.replace("<ts>", "").replace("<ts/>", "")
-        formatted_prompt = f"<|im_start|>system\nYou are a helpful assistant. Your name is ChatTS. You can analyze time series data and provide insights. If user asks who you are, you should give your name and capabilities in the language of the prompt. If no time series are provided, you should say 'I cannot answer this question as you haven't provide the timeseries I need' in the language of the prompt. Always check if the user has provided time series data before answering.<|im_end|><|im_start|>user\n{clean_prompt}<|im_end|><|im_start|>assistant\n"
 
         # Build prompt prefix
         prefix = f"I have {len(ts_list)} time series:\n"
         for name, arr in zip(ts_names, ts_list):
             prefix += f"The {name} is of length {len(arr)}: <ts><ts/>\n"
-        full_prompt = prefix + formatted_prompt
+        
+        full_prompt = f"<|im_start|>system\nYou are a helpful assistant. Your name is ChatTS. You can analyze time series data and provide insights. If user asks who you are, you should give your name and capabilities in the language of the prompt. If no time series are provided, you should say 'I cannot answer this question as you haven't provide the timeseries I need' in the language of the prompt. Always check if the user has provided time series data before answering.<|im_end|><|im_start|>user\n{prefix}{clean_prompt}<|im_end|><|im_start|>assistant\n"
 
-        print(f"[debug] {full_prompt=}, {len(ts_list)=}, {[len(item) for item in ts_list]=}")
+        print(f"[debug] {len(ts_list)=}, {[len(item) for item in ts_list]=}")
 
         # Encode inputs
         inputs = processor(
@@ -254,6 +265,9 @@ with gr.Blocks(title="ChatTS Demo") as demo:
     gr.Markdown("## ChatTS Demo: Time Series Understanding and Reasoning")
     gr.Markdown("Upload a CSV file where each column is a time series, or use the default time series with sudden increase. All columns will be treated as time series data.")
 
+    # State to track whether to use default time series
+    use_default_state = gr.State(value=True)
+
     with gr.Row():
         with gr.Column(scale=1):
             upload = gr.File(
@@ -293,13 +307,19 @@ with gr.Blocks(title="ChatTS Demo") as demo:
     # Initialize interface with default data
     demo.load(
         fn=initialize_interface,
-        outputs=[plot_out, file_status, series_selector]
+        outputs=[plot_out, file_status, series_selector, use_default_state]
     )
 
     # Event handlers
     upload.upload(
         fn=preview_csv,
-        inputs=[upload],
+        inputs=[upload, use_default_state],
+        outputs=[plot_out, file_status, series_selector, use_default_state]
+    )
+
+    upload.clear(
+        fn=clear_csv,
+        inputs=[],
         outputs=[plot_out, file_status, series_selector]
     )
 
@@ -311,7 +331,7 @@ with gr.Blocks(title="ChatTS Demo") as demo:
     
     run_btn.click(
         fn=infer_chatts_stream,
-        inputs=[prompt_input, upload],
+        inputs=[prompt_input, upload, use_default_state],
         outputs=[text_out]
     )