from smolagents import CodeAgent, HfApiModel,load_tool,tool
import datetime
import requests
import pytz
import yaml
import spaces

# Add dummy function to satisfy ZeroGPU startup check
@spaces.GPU(duration=1)
def dummy_gpu():
    return None

from tools.final_answer import FinalAnswerTool
from tools.web_search import DuckDuckGoSearchTool
from tools.visit_webpage import VisitWebpageTool
from tools.markdown_movie_release_sorter import MarkdownMovieReleaseSorterTool

from Gradio_UI import GradioUI

@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()
search_tool = DuckDuckGoSearchTool()
visit_tool = VisitWebpageTool()
sorter_tool = MarkdownMovieReleaseSorterTool()

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

# 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.2,
    model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
    custom_role_conversions=None,
)

with open("prompts.yaml", 'r') as stream:
    prompt_templates = yaml.safe_load(stream)

# Cinema radar instructions
custom_instructions = """
You are 'Cinema Radar', an automated assistant that finds upcoming theatrical movie releases.

STRICT OPERATIONAL RULES:
- NEVER run python's `input()` function. Use details directly from the user prompt.
- If no country is specified, default to Spain.
- NEVER put URL paths or slashes inside `site:`. Only use root domains (e.g., `site:sensacine.com`).
- Maximum 2 searches allowed. If search does not return release dates, directly call `visit_webpage` on:
  - Spain: 'https://www.sensacine.com/peliculas/estrenos/'
  - Global/US: 'https://www.rottentomatoes.com/browse/movies_coming_soon/'
- YOU MUST call `sorter_tool` on the collected dates before calling `final_answer`. Sorting dates manually in code or thought is forbidden.

STEP-BY-STEP WORKFLOW:
1. Identify the genre and country from the user request.
2. Run `web_search`:
   - Spain example: `web_search(query="site:sensacine.com estrenos [genre] 2026")`
   - Global example: `web_search(query="upcoming [genre] movies release dates 2026")`
3. Call `visit_webpage(url=...)` on the best calendar/aggregator link found.
4. Format the found movies into markdown bullet lines:
   raw_text = \"\"\"
   - Movie Title 1: 2026-10-24 - Notes
   - Movie Title 2: 2026-11-15 - Notes
   \"\"\"
5. Run: `sorted_releases = sorter_tool(markdown_text=raw_text)`
6. Pass `sorted_releases` with concise summaries into `final_answer`.
"""

# Append the custom instructions to the existing system_prompt template
prompt_templates["system_prompt"] += f"\n\n{custom_instructions}"
    
agent = CodeAgent(
    model=model,
    tools=[
        get_current_time_in_timezone,
        search_tool, 
        visit_tool, 
        sorter_tool, 
        final_answer
    ],
    max_steps=10,
    verbosity_level=1,
    grammar=None,
    planning_interval=None,
    name=None,
    description=None,
    prompt_templates=prompt_templates,
    additional_authorized_imports=["re", "json", "datetime", "math"]
)


GradioUI(agent).launch()