Update app.py

app.py

@@ -1,5 +1,4 @@
 # app.py
-# app.py
 from __future__ import annotations
 import os
 import traceback
@@ -110,8 +109,9 @@ def handle(user_msg: str, history_messages: List[Dict[str, str]], files: list) -
                 agent = create_pandas_dataframe_agent(
                     llm,
                     df,
-                    agent_type=AgentType.OPENAI_FUNCTIONS, # Recommended for reliability
-                    verbose=True # Set to False in production
+                    agent_type=AgentType.OPENAI_FUNCTIONS,
+                    verbose=True,
+                    allow_dangerous_code=True  # <-- THE FIX IS HERE
                 )
 
                 # Run the agent with the user's scenario text. The agent will
@@ -126,12 +126,10 @@ def handle(user_msg: str, history_messages: List[Dict[str, str]], files: list) -
 
         # --- ORIGINAL LOGIC: Fallback for scenarios without files or general chat ---
         elif is_healthcare_scenario(safe_in, bool(file_paths)) and USE_SCENARIO_ENGINE:
-            # This block now primarily handles scenarios where no data files are provided,
-            # relying on the original deterministic analysis logic.
-            registry = DataRegistry() # This part might be simplified if files always trigger the agent
+            registry = DataRegistry()
             rag = RAGIndex()
             try:
-                ing = extract_text_from_files(file_paths) # For text extraction from markdown/txt
+                ing = extract_text_from_files(file_paths)
                 rag.add(ing.get("chunks", []))
             except Exception as e:
                 log_event("rag_ingest_error", None, {"err": str(e)})
@@ -204,7 +202,6 @@ with gr.Blocks(analytics_enabled=False) as demo:
     ping_btn.click(lambda: ping_cohere(), outputs=[ping_out])
 
 if __name__ == "__main__":
-    # Ensure you have your COHERE_API_KEY set as an environment variable
     if not os.getenv("COHERE_API_KEY"):
         print("🔴 COHERE_API_KEY environment variable not set. Application may not function correctly.")