Update app.py

app.py

@@ -1,200 +1,200 @@
-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()
+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()