fix exec()

app.py

@@ -218,7 +218,6 @@ Rules:
         
         # Debug info
         print("RAW GPT CODE:\n", code)
-        print("FULL RESPONSE:\n", response)
         print("Prompt tokens:", response.usage.prompt_tokens)
         print("Completion tokens:", response.usage.completion_tokens)
         print("Finish reason:", response.choices[0].finish_reason)
@@ -262,10 +261,12 @@ Rules:
         }
 
         # Execute code
-        exec(safe_code, {"__builtins__": safe_builtins}, local_vars)
+        safe_globals = {"__builtins__": safe_builtins, "df1": df1, "df2": df2, "pd": pd}
+        local_vars = {}
+        exec(safe_code, safe_globals, local_vars)
 
         # Fetch result
-        result = local_vars.get("result", "No result produced")
+        result = local_vars.get("result", safe_globals.get("result", "No result produced"))
         return str(format_result(result))
 
     except Exception as e: