Update app.py

app.py

@@ -107,7 +107,7 @@ class ExcelAIQuerySystem:
             return list(self.excel_data.keys())
 
     def query_data(self, query: str, target_sheet: str = "Auto-Select") -> Dict[str, Any]:
-        """--- MODIFIED: Processes a query, either against a specific sheet or by auto-selecting the most relevant ones. ---"""
+        """Processes a query, either against a specific sheet or by auto-selecting the most relevant ones."""
         results = {'query': query, 'relevant_sheets': [], 'sheet_results': {}, 'summary': ''}
         
         try:
@@ -126,7 +126,14 @@ class ExcelAIQuerySystem:
                 if sheet_name not in self.excel_data: continue
                 
                 df = self.excel_data[sheet_name]
-                pandas_agent = create_pandas_dataframe_agent(self.llm, df, agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True)
+                # --- THIS IS THE FIXED LINE ---
+                pandas_agent = create_pandas_dataframe_agent(
+                    self.llm,
+                    df,
+                    agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+                    verbose=True,
+                    allow_dangerous_code=True 
+                )
                 response = pandas_agent.invoke(query)
                 results['sheet_results'][sheet_name] = {'response': response['output']}
             
@@ -148,7 +155,7 @@ class ExcelAIQuerySystem:
 # --- Gradio Interface ---
 
 def process_file(api_key, file_obj):
-    """--- MODIFIED: Also returns the list of sheet names to populate the dropdown. ---"""
+    """Also returns the list of sheet names to populate the dropdown."""
     if not api_key: raise gr.Error("OpenAI API Key is required.")
     if file_obj is None: raise gr.Error("Please upload an Excel file.")
     try:
@@ -168,7 +175,7 @@ def process_file(api_key, file_obj):
         raise gr.Error(f"Failed to process file: {e}")
 
 def generate_response(query, selected_sheet, system_state):
-    """--- MODIFIED: Passes the selected sheet to the query function. ---"""
+    """Passes the selected sheet to the query function."""
     if not query: raise gr.Error("Please enter a query.")
     if system_state is None: raise gr.Error("File not loaded. Please upload and load a file first.")
     
@@ -197,7 +204,6 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Excel AI Query System") as demo:
         
         with gr.Column(scale=2):
             gr.Markdown("### 2. Ask a Question")
-            # --- NEW: Dropdown for sheet selection ---
             sheet_selector = gr.Dropdown(
                 label="Select a sheet to query",
                 info="Choose 'Auto-Select' to let the AI find the best sheet.",