Update app.py

app.py

@@ -178,7 +178,7 @@ def safe_exec_code(code, df, capture_charts=True, interactive=False, extra_globa
         return (f"Generated code references missing columns: {missing_cols}\nPlease adjust your prompt or data.",
                 [])
     
-    # Expanded safe built-ins. (Added 'str' to avoid NameError.)
+    # Expanded safe built-ins. Including float, int, bool, etc.
     safe_builtins = {
         "abs": abs,
         "min": min,
@@ -197,6 +197,11 @@ def safe_exec_code(code, df, capture_charts=True, interactive=False, extra_globa
         "pd": pd,
         "plt": plt,
         "str": str,
+        "float": float,
+        "int": int,
+        "bool": bool,
+        "complex": complex,
+        "round": round,
         "__import__": __import__,
     }
     safe_globals = {"__builtins__": safe_builtins, "df": df, "plt": plt, "charts": []}
@@ -234,6 +239,8 @@ def safe_exec_code(code, df, capture_charts=True, interactive=False, extra_globa
             error_details += "\nHint: The generated code might be using a fixed datetime format. Consider using pd.to_datetime() without a fixed format or with dayfirst=True."
         if "KeyError" in error_details:
             error_details += "\nHint: The generated code might be referencing columns that do not exist in your DataFrame."
+        if "NameError" in error_details:
+            error_details += "\nHint: Ensure that all required built-in types and libraries (like float, int, etc.) are included in the safe built-ins."
         return f"An error occurred during code execution:\n{error_details}", safe_globals["charts"]
     
     if capture_charts and not safe_globals["charts"]: