Update app.py

app.py

@@ -7,7 +7,7 @@ import datetime
 import requests
 import yaml
 import os
-import pytz # Not allowed for some reason 
+import pytz # Had to give it permission in Code agent  
 
 # Tools 
 from tools.final_answer import FinalAnswerTool
@@ -16,7 +16,7 @@ from tools.web_search import DuckDuckGoSearchTool
 
 from Gradio_UI import GradioUI
 
-# Don't use: there's an issue with pytz and it seems like its being blocked by HF
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     
@@ -75,14 +75,15 @@ combined_prompts = {**prompt_templates, **content_prompts}
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, polite_guard, web_search ], ## add your tools here (don't remove final answer)
+    tools=[final_answer, polite_guard, web_search, get_current_time_in_timezone ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
     name="Content Moderator",
     description=None,
-    prompt_templates=combined_prompts 
+    prompt_templates=combined_prompts,
+    additional_authorized_imports=["pytz"]
  
 )