gradio_app/interface.py

import sys
import gradio as gr
import json
import threading
import subprocess
from pathlib import Path
import time

root_dir = Path(__file__).parent.parent
sys.path.append(str(root_dir))
from gradio_app.interacton_with_swarm import *

SWARM_IS_RUNNING = False
SWARM_THREAD = None

def display_logs():
    return read_swarm_logs()

def display_output():
    return read_swarm_output()

def swarm_interface():
# def swarm_interface(swarm_role, swarm_global_goal, swarm_goals, n_managers, n_analysts, n_googlers):
    global SWARM_IS_RUNNING, SWARM_THREAD
    # please, don't judge me for this hardcoding. it's 3am and it's the first time i use gradio =)))
    # Call the necessary set_ functions with the user inputs
    # set_swarm_role(swarm_role)
    # set_swarm_global_goal(swarm_global_goal)
    # set_swarm_goals(swarm_goals)
    # agents_config = [
    #     {"type": "manager", "n": n_managers},
    #     {"type": "analyst", "n": n_analysts},
    #     {"type": "googler", "n": n_googlers}
    # ]
    # set_swarm_agents_config(agents_config)

    SWARM_THREAD = threading.Thread(target=execute_swarm, args=(SWARM_IS_RUNNING,))
    SWARM_THREAD.start()
    SWARM_IS_RUNNING = True

    print(f"Swarm is running. SWARM_IS_RUNNING = {SWARM_IS_RUNNING}")

def create_gradio_interface():
    global SWARM_IS_RUNNING, SWARM_THREAD
    title = """
    <h1 align="center">🐝🐝 Swarm Intelligence 🐝🐝</h1>
    <div align="center">
    <a style="display:inline-block" href='https://github.com/nicelir1996/GPT-Swarm'><img src='https://img.shields.io/github/stars/nicelir1996/GPT-Swarm?style=social' /></a>
    </div>
    <p align='center' style='font-size: 25px;'>⚠️Attention please⚠️! This version is specifically limited to the input of any case due to the heavy load on the server. We use the case 'VC evaluates the investment potential of Brainamics'</p>
    <p align='center' style='font-size: 18px;'><b>Role: </b>professional venture capital agency, who has a proven track reckord of consistently funding successful startups</p>
    <p align='center' style='font-size: 20px;'><b>Goal: </b>A new startup just send us their pitch. Find if the startup is worth investing in. The startup is called Brainamics and it is in the space of brain computer interfaces.</p>

    <br>
    <div align="center">
    ⚠️⚠️We Highly recommend to use a duplicate space!⚠️⚠️
    <br> Due to huggingface's API limitations, there can only be one instance of the swarm running at a time. You can see it's output, but your input will be ignored. The swarm resets every 10 minutes, so maybe you get lucky.
    <a href="https://huggingface.co/spaces/swarm-agents/swarm-agents?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a>
    </div>
    """

    #display message for themes feature
    theme_addon_msg = """
    The swarm of agents combines a huge number of parallel agents divided into roles, including (for now) managers, analytics, and googlers. 
    The agents all interact with each other through the shared memory and the task queue.
    """

    #Modifying existing Gradio Theme
    theme = gr.themes.Soft(primary_hue="zinc", secondary_hue="green", neutral_hue="green",
                        text_size=gr.themes.sizes.text_lg)       
    
    with gr.Blocks() as demo:
        # Create a container on the left for the inputs
        gr.HTML(title)
        gr.HTML(theme_addon_msg)

        # layout
        with gr.Row():
            # with gr.Column(variant="panel", scale=0.4):
                # with gr.Accordion(label="Swarm goals (can leave empty for default)", open=False):
                #     # Create a textbox for swarm role
                #     swarm_role = gr.Textbox(placeholder=get_swarm_role(), label="Swarm role", disabled=True)
                #     # Create a textbox for swarm global goal
                #     swarm_global_goal = gr.Textbox(placeholder=get_swarm_global_goal(), label="Swarm global goal", disabled=True, description="⚠️Attention please⚠️! This version is specifically limited to the input of any case due to the heavy load on the server. We use the case \"VC evaluates the investment potential of Brainamics\"")
                #     # Create a list for swarm goals
                #     swarm_goals = gr.List(headers=None, col_count=(1, "fixed"), max_cols=1)
                # with gr.Accordion(label="Agents Setup:", open=False):
                    # Create a textbox for number of manager agents
                    # n_managers = gr.Textbox(placeholder=get_swarm_agents_config()[0]["n"], label="Number of manager agents", disabled=True)
                    # Create a textbox for number of analyst agents
                    # n_analysts = gr.Textbox(placeholder=get_swarm_agents_config()[1]["n"], label="Number of analyst agents", disabled=True)
                    # Create a textbox for number of googler agents
                    # n_googlers = gr.Textbox(placeholder=get_swarm_agents_config()[2]["n"], label="Number of googler agents", disabled=True)
                    # create a submit button

            # Create a container on the right for the outputs
            with gr.Column(variant="panel", scale=1):
                    submit = gr.Button(value="Start the Swarm 🚀")
                    # Create a textbox for output
                    output_textbox = gr.Textbox(label="Output", lines=20)
                    # Create a textbox for logs
                    logs_textbox = gr.Textbox(label="Logs", lines=8)
                    update_view_button = gr.Button(value="Update Results Display 🔄")
                    gr.HTML("""<center><p>(If someone knows how to update dynamically, please save us, that's emberrasing 😳)</p></center>""")

        #Event handling
        def update_view_callback():
            return display_logs(), display_output()
        
        def submit_callback():
        # def submit_callback(swarm_role, swarm_global_goal, swarm_goals, n_managers, n_analysts, n_googlers):
            global SWARM_IS_RUNNING, SWARM_THREAD
            if isinstance(SWARM_THREAD, threading.Thread):
                SWARM_IS_RUNNING = SWARM_THREAD.is_alive()
            print(f"Swarm is running. SWARM_IS_RUNNING = {SWARM_IS_RUNNING}")
            if not SWARM_IS_RUNNING:
                swarm_interface()
                # swarm_interface(swarm_role, swarm_global_goal, swarm_goals)
                # swarm_interface(swarm_role, swarm_global_goal, swarm_goals, n_managers, n_analysts, n_googlers)
            return display_logs(), display_output()
        
        submit.click(submit_callback, outputs=[logs_textbox, output_textbox])
        # submit.click(submit_callback, inputs=[swarm_role, swarm_global_goal, swarm_goals, n_managers, n_analysts, n_googlers], outputs=[logs_textbox, output_textbox])
        update_view_button.click(update_view_callback, outputs=[logs_textbox, output_textbox])

    return demo