Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import json
 import numpy as np
+import textwrap
 from tokenizers import Tokenizer
 import onnxruntime as ort
 from huggingface_hub import hf_hub_download
@@ -43,10 +44,26 @@ class ONNXInferencePipeline:
         
         code_str = os.getenv("banned")
         if not code_str:
-            raise Exception("Environment variable 'code' is not set. Please set it with your complete application code.")
+            raise Exception("Environment variable 'banned' is not set. Please set it with your banned keywords list.")
 
-        # Execute the code loaded from the environment variable
-        exec(code_str)
+        # Create a local namespace to execute the code
+        local_vars = {}
+        
+        # Wrap the code in a function to allow return statements
+        wrapped_code = f"""
+def get_banned_keywords():
+{textwrap.indent(code_str, '    ')}
+"""
+        
+        try:
+            # Execute the wrapped code
+            exec(wrapped_code, globals(), local_vars)
+            # Call the function to get the banned keywords
+            return local_vars['get_banned_keywords']()
+        except Exception as e:
+            print(f"Error loading banned keywords: {e}")
+            # Return a default empty list if there's an error
+            return []
 
     def contains_banned_keyword(self, text):
         """Check if the input text contains any banned keywords."""