Update app.py

app.py

@@ -21,7 +21,35 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 cache = {}
 
 web_search = DuckDuckGoSearchTool()
-python_interpreter = PythonInterpreterTool()
+python_interpreter = PythonInterpreterTool(authorized_imports = [
+    # standard library
+    'os',                      # For file path manipulation, checking existence, deletion
+    'glob',                    # Find files matching specific patterns
+    'pathlib',                 # Alternative for path manipulation
+    'sys',
+    'math',
+    'random',
+    'datetime',
+    'time',
+    'json',
+    'csv',
+    're',
+    'collections',
+    'itertools',
+    'functools',
+    'io',
+    'base64',
+    'hashlib',
+    'pathlib',
+    'glob',
+
+    # Third-Party Libraries (ensure they are installed in the execution env)
+    'pandas',         # Data manipulation and analysis
+    'numpy',          # Numerical operations
+    'scipy',          # Scientific and technical computing (stats, optimize, etc.)
+    'sklearn',        # Machine learning
+
+])
 visit_webpage_tool = VisitWebpageTool()
 wiki_tool = WikipediaPageFetcher()
 yt_transcript_fetcher = YoutubeTranscriptFetcher()
@@ -70,8 +98,8 @@ def load_model(model_name):
 # Load the model and tokenizer locally
 # model, tokenizer = load_model()
 
-model_id = "meta-llama/Llama-3.1-8B-Instruct"  # "microsoft/phi-2"# not working out of the box"google/gemma-2-2b-it" #toobig"Qwen/Qwen1.5-7B-Chat"#working but stupid: "meta-llama/Llama-3.2-3B-Instruct"
-model = HfApiModel(model_id)
+#model_id = "meta-llama/Llama-3.1-8B-Instruct"  # "microsoft/phi-2"# not working out of the box"google/gemma-2-2b-it" #toobig"Qwen/Qwen1.5-7B-Chat"#working but stupid: "meta-llama/Llama-3.2-3B-Instruct"
+model = HfApiModel()
 #from smolagents import TransformersModel
 # model = TransformersModel(
 #     model_id=model_id,
@@ -91,7 +119,7 @@ class BasicAgent:
             grammar=None,
             planning_interval=3,
             add_base_tools=True,
-            additional_authorized_imports=['requests', 'wikipedia', 'pandas']
+            additional_authorized_imports=['requests', 'wikipedia', 'pandas','datetime']
 
         )