Update app.py

app.py

@@ -54,6 +54,27 @@ class DataAnalyzer:
         except Exception as e:
             return f"API Error: {str(e)}"
 
+    def evaluate_code(self, code: str, state: Dict = None) -> Tuple[Any, str]:
+        """Safely evaluate Python code with proper state management and security"""
+        if state is None:
+            state = {"print_outputs": ""}
+            
+        # Create safe environment with allowed imports
+        safe_env = {
+            "pd": pd,
+            "np": np,
+            "plt": plt,
+            "sns": sns,
+            "stats": stats,
+            "print": lambda *args: state.update({"print_outputs": state["print_outputs"] + " ".join(map(str, args)) + "\n"}),
+        }
+        
+        try:
+            exec(code, safe_env, state)
+            return state.get("result", None), state["print_outputs"]
+        except Exception as e:
+            raise RuntimeError(f"Code execution failed: {str(e)}")
+
     def analyze_data_types(self, df: pd.DataFrame) -> Dict:
         """Analyze data types and basic statistics of the DataFrame"""
         analysis = {