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_random_ayah_number(surah_number: int = None) -> str:
    """Returns random Quran ayah location info.
    Args:
        surah_number: The surah number. If None, a random surah is chosen.
    """
    import random
    ayahs_meta = [7, 286, 200, 176, 120, 165, 206, 75, 129, 109, 123, 111, 43, 52, 99, 128,111, 110, 98, 135, 112, 78, 118, 64, 77, 227, 93, 88, 69, 60, 34, 30, 73, 54, 45, 83, 182, 88, 75, 85, 54, 53, 89, 59, 37, 35, 38, 29, 18, 45, 60, 49, 62, 55, 78, 96, 29, 22, 24, 13, 14, 11, 11, 18, 12, 12, 30, 52, 52, 44, 28, 28, 20, 56, 40, 31, 50, 40, 46, 42, 29, 19, 36, 25, 22, 17, 19, 26, 30, 20, 15, 21, 11, 8, 8, 19, 5, 8, 8, 11, 11, 8, 3, 9, 5, 4, 7, 3, 6, 3, 5, 4, 5, 6]
    if surah_number is not None:
        ayah = random.randint(1, ayahs_meta[surah_number-1])
    else:
        surah = random.randint(1, 114)
        ayah = random.randint(1, ayahs_meta[surah-1])
    return "Surah " + str(surah) + " ayah " + str(ayah)

@tool
def get_quran_ayah(surah_number: int, ayah_number: int) -> str:
    """Fetches a specific ayah arabic and translation textes from the Quran API for the specified surah and ayah numbers.
    Args:
        surah_number: The surah number.
        ayah_number: The ayah number.
    """
    url = f"https://api.alquran.cloud/v1/ayah/{surah_number}:{ayah_number}/editions/quran-simple,en.asad"
    
    try:
        response = requests.get(url)
        response.raise_for_status()
        data = response.json()
        arabic_text = data["data"][0]["text"]
        english_text = data["data"][1]["text"]
        return f"The text for ayah {ayah_number} of surah {surah_number} is:\nArabic: {arabic_text}\nEnglish: {english_text}"
    except requests.RequestException as e:
        response_data = response.json()
        if response_data.get("status") == "NOT FOUND":
            return f"Error: NOT FOUND - Invalid surah/ayah reference ({surah_number}:{ayah_number})"
        return f"Error fetching ayah {ayah_number} of surah {surah_number}: {e}"

@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_random_ayah_number, get_quran_ayah, 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()