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 get_news_headlines()-> str: #it's import to specify the return type
    #Keep this format for the description / args / args description but feel free to modify the tool
    """A dummy tool that imitates fetching of the top 3 headlines from the news aggregator omni.se 
    """

    dummy = """
    Trump hävdar felaktigt att Ukraina startade kriget
USA:s president Donald Trump påstår felaktigt att det var Ukraina som startade kriget med Ryssland. Angående att Kyiv inte haft en plats vid förhandlingsbordet i Saudiarabien säger han att Ukraina ”borde avslutat” det för tre år sedan.
– Ni borde aldrig ha startat det. Ni kunde ha slutit ett avtal, säger Trump vid en pressträff på Mar-a-Lago.

Tidöpartierna vill inte ta bort ordet ”kvinna” ur abortlag
Tidöpartierna är kritiska till förslaget att ta bort ordet ”kvinna” ur den svenska abortlagstiftningen och i stället ersätta det med ”den som är gravid”.
– Det är kvinnor som blir gravid och det är kvinnor som föder barn. Det är den biologiska grunden, så enkelt är det, säger Linda Lindberg, gruppledare i SD, till Ekot.
Förslaget kommer eftersom även den som har juridiskt kön ”man” kan bli gravid, och då måste ha möjlighet att göra abort.

Asap Rocky frias i rätten: ”Räddade mitt liv”
Rapartisten Asap Rocky frias av juryn. Rakim Mayers, som han egentligen heter, stod åtalad för att ha skjutit mot sin barndomskompis och riskerade upp till 24 års fängelse vid en fällande dom.
– Tack för att ni räddade mitt liv, sa han till juryn.
    """
    
    return dummy

@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=[get_news_headlines, final_answer], ## 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()