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

@tool
def get_wikipedia_movie_summary(movie_title:str) -> str:
    """A tool that queries Wikipedia for a movie title in English and returns a page's introduction.
      Args:
          movie_title: the English name of the movie
    """
    # Wikipedia API endpoint
    url = "https://en.wikipedia.org/w/api.php"
    
    params = {
        "action": "query",
        "format": "json",
        "titles": f"{movie_title} movie",
        "prop": "extracts",
        "explaintext": True,
        "exintro": True  # Get only the introductory section
    }
    
    try:
        # Make the API request
        response = requests.get(url, params=params)
        data = response.json()
        
        # Extract the page content
        pages = data["query"]["pages"]
        page_id = next(iter(pages))  # Get the first page ID
        page_content = pages[page_id]["extract"]
        
        return page_content
    except Exception as e:
        return f"Error fetching Wikipedia data: {str(e)}"

# 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,
)

with open("prompts.yaml", 'r') as stream:
    prompt_templates = yaml.safe_load(stream)
    
name = "Bootyful Bot"
task_description = """
- Bootyful Bot will search wikipedia for a movie name
- Bootyful Bot will read the page and tell whether the movie was inspired by factual events or not
- In case it is, tell user about the events
- In case it is not, tell about movie inspiration, if information is available.
- Bootyful Bot should begin every sentence with 'bing bing' and end every sentence with 'bong bong'.
- Bootyful Bot should begin answer with 'Hello, I am Bootyful Bot' and end it with 'Bye bye, I am Bootyful Bot'.
"""

final_answer = FinalAnswerTool(name=name, description=task_description)


agent = CodeAgent(
    model=model,
    tools=[final_answer, get_wikipedia_movie_summary], ## add your tools here (don't remove final answer)
    max_steps=6,
    verbosity_level=1,
    grammar=None,
    planning_interval=None,
    name=None,
    description=task_description,
    prompt_templates=prompt_templates
)



GradioUI(agent).launch()