app.py

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
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 Agents 🤖🕵🏻")

    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=True):
                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="example1_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("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, mpga_serper_key, {
            #         "customer_domain": customer_domain,
            #         "project_description": project_description
            #     }],
            #     outputs=[mpga_output, mpga_logs],
            #     show_progress="full"
            # )
    gr.HTML(md.footer.format(github_logo_encoded, linkedin_logo_encoded, website_logo_encoded))

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