import gradio as gr
import markdown as md
import yaml
from ContentGeneratorAgent import run_crew_cga, set_model, llm_models
from GameBuilderAgent import run_crew_game
from MarketingPostGeneratorAgent import run_crew_mpga
from TripPlanner import run_crew_tp
import base64


def toggle_serper_input(choice):
    return gr.Textbox(visible=(choice == "Yes"))

def update_game_instructions(example_key):
    # Load the game design examples from the YAML file
    with open('gamedesign.yaml', 'r', encoding='utf-8') as f:
        examples = yaml.safe_load(f)
    return examples.get(example_key, "")

def encode_image(image_path):
    with open(image_path, "rb") as image_file:
        return base64.b64encode(image_file.read()).decode('utf-8')

# Encode the images
github_logo_encoded = encode_image("Images/github-logo.png")
linkedin_logo_encoded = encode_image("Images/linkedin-logo.png")
website_logo_encoded = encode_image("Images/ai-logo.png")

# UI Setup
with gr.Blocks(theme=gr.themes.Soft(font=[gr.themes.GoogleFont("Roboto Mono")]),
               css='footer {visibility: hidden}') as demo:
    gr.Markdown("# AI Agent Nexus 🤖🕵🏻")

    with gr.Tabs():
        with gr.TabItem("Intro"):
            gr.Markdown(md.description)

        # Tab for SEO Content Generator Agent
        with gr.TabItem("SEO Content Agent"):
            with gr.Accordion("📔 Description:", open=False):
                gr.Markdown(md.seo_content)
            with gr.Accordion("How to get GEMINI API KEY", open=False):
                gr.Markdown(md.gemini_api_key)
            with gr.Accordion("How to get SERPER API KEY", open=False):
                gr.Markdown(md.serper_api_key)
            with gr.Row():
                with gr.Column(scale=1):
                    model_dropdown = gr.Dropdown(
                        llm_models,
                        label="1. Select AI Model",
                        value=llm_models[0]
                    )
                    gemini_key = gr.Textbox(
                        label="2. Enter Gemini API Key",
                        type="password",
                        placeholder="Paste your Gemini API key here..."
                    )
                    search_toggle = gr.Radio(
                        ["Yes", "No"],
                        label="3. Enable Online Search?",
                        value="No"
                    )
                    serper_key = gr.Textbox(
                        label="4. Enter Serper API Key",
                        type="password",
                        visible=False,
                        placeholder="Paste your Serper API key if enabled..."
                    )
                    topic_input = gr.Textbox(
                        label="5. Enter Content Topic",
                        placeholder="Enter your article topic here..."
                    )
                    run_btn = gr.Button("Generate Content", variant="primary")
                with gr.Column(scale=3):
                    output = gr.Markdown(
                        label="Generated Content",
                        value="Your content will appear here..."
                    )
                    with gr.Accordion("Process Logs", open=True):
                        logs = gr.Markdown()

            # Event handlers for SEO Content Generator
            model_dropdown.change(set_model, model_dropdown)
            search_toggle.change(
                toggle_serper_input,
                inputs=search_toggle,
                outputs=serper_key
            )
            run_btn.click(
                run_crew_cga,
                inputs=[gemini_key, search_toggle, serper_key, topic_input],
                outputs=[output, logs],
                show_progress="full"
            )

        # Tab for Game Dev Agent
        with gr.TabItem("Game Dev Agent"):
            with gr.Accordion('📔 Description:', open=False):
                gr.Markdown(md.game_dev)
            with gr.Accordion("How to get GEMINI API KEY", open=False):
                gr.Markdown(md.gemini_api_key)
            with gr.Row():
                with gr.Column(scale=1):
                    game_model_dropdown = gr.Dropdown(
                        llm_models,
                        label="1. Select AI Model",
                        value=llm_models[0]
                    )
                    game_gemini_key = gr.Textbox(
                        label="2. Enter Gemini API Key",
                        type="password",
                        placeholder="Paste your Gemini API key here..."
                    )
                    game_example_dropdown = gr.Dropdown(
                        choices=["pacman", "pacman2", "snake", "space_invaders", "Tetris", "Frogger", "Chess", "Go",
                                 "Reversi"],
                        label="3. Select Example",
                        value="pacman"
                    )
                    game_load_example_btn = gr.Button("Load Example", variant="secondary")
                    game_instructions = gr.Textbox(
                        label="4. Enter Game Design Instructions",
                        placeholder="Enter your game design instructions here...",
                        lines=5
                    )
                    game_run_btn = gr.Button("Generate Game Code", variant="primary")
                with gr.Column(scale=3):
                    game_output = gr.Markdown(
                        label="Generated Game Code",
                        value="Your game code will appear here..."
                    )
                    with gr.Accordion("Process Logs", open=False):
                        game_logs = gr.Markdown()

            # Event handlers for Game Dev Agent
            game_model_dropdown.change(set_model, game_model_dropdown)
            game_load_example_btn.click(
                update_game_instructions,
                inputs=[game_example_dropdown],
                outputs=[game_instructions]
            )
            game_run_btn.click(
                run_crew_game,
                inputs=[game_gemini_key, game_instructions],
                outputs=[game_output, game_logs],
                show_progress="full"
            )

        # Tab for Marketing Posts Generator Agent
        with gr.TabItem("Marketing Posts Generator Agent"):
            with gr.Accordion("📔 Description: ", open=False):
                gr.Markdown(md.marking_post_gen_agent)
            with gr.Accordion("How to get GEMINI API KEY", open=False):
                gr.Markdown(md.gemini_api_key)
            with gr.Accordion("How to get SERPER API KEY", open=False):
                gr.Markdown(md.serper_api_key)
            with gr.Row():
                with gr.Column(scale=1):
                    mpga_model_dropdown = gr.Dropdown(
                        llm_models,
                        label="1. Select AI Model",
                        value=llm_models[0]
                    )
                    mpga_gemini_key = gr.Textbox(
                        label="2. Enter Gemini API Key",
                        type="password",
                        placeholder="Paste your Gemini API key here..."
                    )
                    mpga_serper_key = gr.Textbox(
                        label="3. Enter Serper API Key",
                        type="password",
                        placeholder="Paste your Serper API key here..."
                    )
                    customer_domain = gr.Textbox(
                        label="4. Enter Customer Domain",
                        placeholder="Enter the customer domain here..."
                    )
                    project_description = gr.Textbox(
                        label="5. Enter Project Description",
                        placeholder="Enter the project description here..."
                    )
                    mpga_run_btn = gr.Button("Generate Marketing Posts", variant="primary")
                with gr.Column(scale=3):
                    mpga_output = gr.Markdown(
                        label="Generated Marketing Posts",
                        value="Your marketing posts will appear here..."
                    )
                    with gr.Accordion("Process Logs", open=False):
                        mpga_logs = gr.Markdown()

            # Event handlers for Marketing Posts Generator Agent
            mpga_model_dropdown.change(set_model, mpga_model_dropdown)
            mpga_run_btn.click(
                run_crew_mpga,
                inputs=[mpga_gemini_key, customer_domain, project_description],
                outputs=[mpga_output, mpga_logs],
                show_progress="full"
            )

        # Tab for Trip Planner Agent
        with gr.TabItem("Trip Planner Agent"):
            with gr.Accordion("📔 Description: ", open=False):
                gr.Markdown(md.trip_planner_agent)
            with gr.Accordion("How to get GEMINI API KEY", open=False):
                gr.Markdown(md.gemini_api_key)
            with gr.Row():
                with gr.Column(scale=1):
                    tp_model_dropdown = gr.Dropdown(
                        llm_models,
                        label="1. Select AI Model",
                        value=llm_models[0]
                    )
                    tp_gemini_key = gr.Textbox(
                        label="2. Enter Gemini API Key",
                        type="password",
                        placeholder="Paste your Gemini API key here..."
                    )
                    origin_input = gr.Textbox(
                        label="3. Enter Origin (Your Starting Location)",
                        placeholder="Enter your origin city..."
                    )
                    cities_input = gr.Textbox(
                        label="4. Enter City Options",
                        placeholder="List the cities you're interested in visiting..."
                    )
                    trip_date_input = gr.Textbox(
                        label="5. Enter Trip Date",
                        placeholder="Enter your trip date (e.g., YYYY-MM-DD to YYYY-MM-DD)..."
                    )
                    interests_input = gr.Textbox(
                        label="6. Enter Traveler Interests",
                        placeholder="Enter your interests/hobbies..."
                    )
                    tp_run_btn = gr.Button("Generate Trip Plan", variant="primary")
                with gr.Column(scale=3):
                    tp_output = gr.Markdown(
                        label="Trip Plan",
                        value="Your trip plan will appear here..."
                    )
                    with gr.Accordion("Process Logs", open=True):
                        tp_logs = gr.Markdown()

            # Event handlers for Trip Planner Agent
            tp_model_dropdown.change(set_model, tp_model_dropdown)
            tp_run_btn.click(
                run_crew_tp,
                inputs=[tp_gemini_key, origin_input, cities_input, trip_date_input, interests_input],
                outputs=[tp_output, tp_logs],
                show_progress="full"
            )

        with gr.TabItem("AI Agent Dev Agent"):
            gr.Markdown("# Comming soon ...")

    gr.HTML(md.footer.format(github_logo_encoded, linkedin_logo_encoded, website_logo_encoded))

if __name__ == "__main__":
    demo.launch()