app.py

from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool

from Gradio_UI import GradioUI


# Below is an example of a tool that does nothing. Amaze us with your creativity !
@tool
def my_custom_tool(arg1: str, arg2: int) -> str:
    """
    A tool that shares interesting facts about Chakwal, highlighting its flora, fauna, natural landscapes,
    and cultural heritage. It also shows the current local time in Chakwal (Asia/Karachi timezone).
    
    Args:
        arg1: A custom note or additional message to include in the output.
        arg2: The number of detailed facts to include about Chakwal (1 to 4).
              (1: Flora only, 2: Flora & Fauna, 3: + Natural Landscapes, 4: + Cultural Heritage)
    """
    messages = []
    
    # Introduction about Chakwal
    messages.append("Welcome to Chakwal, a city with a rich history, vibrant culture, and breathtaking natural beauty!")
    
    # Facts about Chakwal's natural and cultural highlights
    facts = [
        ("Flora", "The region boasts a diverse array of native plants, lush greenery, and seasonal wildflowers that add vibrant colors to the landscape."),
        ("Fauna", "Local wildlife includes a variety of bird species, reptiles, and small mammals that thrive in Chakwal's varied ecosystems."),
        ("Natural Landscapes", "Chakwal is surrounded by scenic hills, valleys, and water bodies, offering breathtaking views and numerous opportunities for outdoor adventures."),
        ("Cultural Heritage", "Deep-rooted traditions and lively festivals reflect Chakwal's rich cultural tapestry and historical significance.")
    ]
    
    # Determine the number of facts to display (minimum 1, maximum 4)
    num_facts = min(max(arg2, 1), len(facts))
    for i in range(num_facts):
        title, detail = facts[i]
        messages.append(f"{title}: {detail}")
    
    # Add current local time in Chakwal (using Asia/Karachi timezone)
    try:
        tz = pytz.timezone("Asia/Karachi")
        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
        messages.append(f"Local Time in Chakwal (Asia/Karachi): {local_time}")
    except Exception as e:
        messages.append("Error fetching local time for Chakwal.")
    
    # Include any custom note provided via arg1
    if arg1:
        messages.append(f"Note: {arg1}")
    
    return "\n".join(messages)




@tool
def get_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').
    """
    try:
        # Create timezone object
        tz = pytz.timezone(timezone)
        # Get current time in that timezone
        local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S")
        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)}"


final_answer = FinalAnswerTool()

# If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
# model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 

model = HfApiModel(
max_tokens=2096,
temperature=0.5,
model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
custom_role_conversions=None,
)


# Import tool from Hub
image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)

with open("prompts.yaml", 'r') as stream:
    prompt_templates = yaml.safe_load(stream)
    
agent = CodeAgent(
    model=model,
    tools=[final_answer,my_custom_tool,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=None,
    description=None,
    prompt_templates=prompt_templates
)


GradioUI(agent).launch()