Update app.py

app.py

@@ -10,31 +10,11 @@ import pytz # Had to give it permission in Code agent
 from agents.tools import tools
 
 from check_endpoint import is_huggingface_endpoint
+from agents.prompts import load_prompts
 
 from Gradio_UI import GradioUI
 
 
-@tool
-def get_the_current_time_in_timezone(timezone: str) -> str:
-    
-    """A tool that fetches the current local time in a specified timezone.
-    Args:
-        timezone: A string representing a valid timezone (e.g., 'America/New_York').
-
-    Returns: 
-         string:  A sentence that provides the time using the  12-hour clock including  AM/PM
-    """
-    try:
-        # Create timezone object
-        tz = pytz.timezone(timezone)
-        # Get current time in that timezone
-        local_time = datetime.datetime.now(tz).strftime("%I:%M %p")  # %I for 12-hour clock, %M for minutes, %p for AM/PM
-        return f"The current local time in {timezone} is: {local_time}"
-    except Exception as e:
-        return f"Error fetching time for timezone '{timezone}': {str(e)}"
-
-
-
 my_id = os.getenv("QWEN_URI")
 
 
@@ -68,15 +48,8 @@ model = LiteLLMModel(
 # Import tool from Hub
 #image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
-# Load prompts.yaml
-with open("prompts.yaml", 'r') as stream:
-    prompt_templates = yaml.safe_load(stream)
-
-# Load contentprompts.yml
-with open("contentprompts.yml", 'r') as stream:
-    content_prompts = yaml.safe_load(stream)
-
-combined_prompts = {**prompt_templates, **content_prompts}
+# Load prompts
+combined_prompts = load_prompts()
 
 
 #  web_search, visit_webpage