from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import datetime
import requests
import pytz
import yaml
from tools.final_answer import FinalAnswerTool
import http.client
from Gradio_UI import GradioUI
import json
from typing import List, Dict


# Below is an example of a tool that does nothing. Amaze us with your creativity !
@tool
def calculate_sum_of_integers(int1:int, int2:int)-> int: 
    """A tool that calculates the sum of two integers values
    Args:
        int1: An integer representing first value
        int2: An integer representing second value
    """
    try:
        # create a variable to store the sum
        sum = int1 + int2
        return sum
    except Exception as e:
        return 0
    

@tool
def get_top_three_bestselling_products(search_term:str)-> str: 
    """A tool that fetches the top three best selling products
    Args:
        search_term: A string representing the product type or category 
    """
    base_url = "https://scout-amazon-data.p.rapidapi.com/Amazon-Search-Data"
    query_params = {"query" : search_term, "region" : "US"}
    headers = {
            'x-rapidapi-key': "1c19a4f07fmsh17bab4e619b3dd7p140535jsnd62249d8c82d",
            'x-rapidapi-host': "scout-amazon-data.p.rapidapi.com"
        }

    try:
        # use the 
        response = requests.get(base_url, headers = headers, params = query_params)
        # check response status code
        if response.status_code == 200:
            # response body only
            data = response.json()
            top_three_products = data.get("products", [])[:3]
            result = []
            for product in top_three_products:
                name = product.get("product_title", "N/A")
                price = product.get("product_price", "N/A")
                result.append({
                    "product_title" : name,
                    "product_price" : price
                })

            return json.dumps(result)
        else:
            return f"error in response {response}"
    except Exception as e:
        return f"No results were found or error :: {response}"


@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, calculate_sum_of_integers, get_top_three_bestselling_products], ## 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()