Update app.py

app.py

@@ -10,6 +10,9 @@ import gradio as gr
 import pandas as pd
 from datetime import datetime
 
+# --- THE FINAL FIX IS HERE: Re-introducing the missing import ---
+import regex as re2
+
 # --- BACKEND IMPORTS ---
 from langchain_cohere import ChatCohere
 
@@ -34,6 +37,7 @@ def load_markdown_text(filepath: str) -> str:
 
 def _sanitize_text(s: str) -> str:
     if not isinstance(s, str): return s
+    # This now works because 're2' is defined from the import above
     return re2.sub(r'[\p{C}--[\n\t]]+', '', s)
 
 def _create_python_script(user_scenario: str, schema_context: str) -> str:
@@ -110,26 +114,16 @@ def handle(user_msg: str, files: list) -> str:
             schema_context = "\n".join(schema_parts)
             analysis_script = _create_python_script(safe_in, schema_context)
 
-            # --- THE FINAL, ROBUST FIX IS HERE: Using standard Python 'exec' ---
-            # Create a safe environment for the script to run in
-            execution_namespace = {
-                "dfs": dataframes,
-                "pd": pd
-            }
-            # Create a string buffer to capture the script's `print` statements
+            execution_namespace = {"dfs": dataframes, "pd": pd}
             output_buffer = io.StringIO()
             
             try:
-                # Use a context manager to redirect stdout to our buffer
                 with redirect_stdout(output_buffer):
-                    # Execute the AI-generated script in the safe namespace
                     exec(analysis_script, execution_namespace)
                 
-                # Get the captured output
                 result = output_buffer.getvalue()
                 return _sanitize_text(result or "(The analysis script ran but produced no output.)")
             except Exception as e:
-                # If execution fails, return the error and the script for debugging
                 return f"An error occurred executing the script: {e}\n\nGenerated Script:\n```python\n{analysis_script}\n```"
         else:
             prompt = f"{GENERAL_CONVERSATION_PROMPT}\n\nUser: {safe_in}\nAssistant:"