Expand safe_builtins and add them in the prompt

app.py

@@ -149,6 +149,36 @@ def increment_usage_count():
         f.write(str(count))
 
 def answer_question(history, message):
+    # Define a restricted set of built-ins
+    safe_builtins = {
+        "abs": abs,
+        "all": all,
+        "any": any,
+        "bool": bool,
+        "dict": dict,
+        "float": float,
+        "int": int,
+        "len": len,
+        "list": list,
+        "max": max,
+        "min": min,
+        "range": range,
+        "str": str,
+        "sum": sum,
+        "round": round,
+        "KeyError": KeyError,
+        "ValueError": ValueError,
+        "sorted": sorted,
+        "enumerate": enumerate,
+        "zip": zip,
+        "set": set,
+        "type": type,
+        "isinstance": isinstance,
+        "pd": pd,
+        "np": np,
+        "print": print,
+        "Exception": Exception
+    }
     if get_usage_count() >= MAX_REQUESTS_PER_DAY:
         return "OpenAI API usage limit reached for today."
     increment_usage_count()
@@ -199,6 +229,7 @@ Rules:
 - If result is numeric, round to 2 decimal places.
 - If result is a list, return the full list (not just the first element).
 - If a required column or metric is missing, return a clear error string in `result` instead of crashing.
+- List of allowed functions: {list(safe_builtins.keys())}
     """
 
 
@@ -236,30 +267,6 @@ Rules:
         # Prepare sandbox
         local_vars = {"df1": df1, "df2": df2, "pd": pd}
 
-        # Define a restricted set of built-ins
-        safe_builtins = {
-            "abs": abs,
-            "all": all,
-            "any": any,
-            "bool": bool,
-            "dict": dict,
-            "float": float,
-            "int": int,
-            "len": len,
-            "list": list,
-            "max": max,
-            "min": min,
-            "range": range,
-            "str": str,
-            "sum": sum,
-            "round": round,
-            "KeyError": KeyError,
-            "ValueError": ValueError,
-            "sorted": sorted,
-            "enumerate": enumerate,
-            "zip": zip,
-        }
-
         # Execute code
         safe_globals = {"__builtins__": safe_builtins, "df1": df1, "df2": df2, "pd": pd}
         local_vars = {}