add

app.py

@@ -1,4 +1,5 @@
 import sys
+sys.path.append("../../Gradio_Config")
 
 from gradio_base import UIHelper, WebUI
 import os
@@ -8,6 +9,7 @@ from typing import List, Tuple, Any
 import gradio as gr
 import time
 
+
 class DebateUI(WebUI):
     FORMAT = "{}\n<debate topic>\n{}\nAffirmative viewpoint:{}\nNegative viewpoint:{}\n<debate topic>{}"
     AUDIENCE = "Audience" 
@@ -77,13 +79,13 @@ class DebateUI(WebUI):
         render_data = self.render_bubble(history, self.data_history, node_name, render_node_name= True or state % 10 == 2)
         return render_data
 
-    def start_button_when_click(self, theme, positive, negative, choose, mode):
+    def start_button_when_click(self, theme, positive, negative, choose, mode, api_key):
         """
         inputs=[self.text_theme, self.text_positive, self.text_negative, self.radio_choose],
         outputs=[self.chatbot, self.btn_send]
         """
         cosplay = None if choose == self.AUDIENCE else choose.split("(")[0]
-        message = dict(theme=theme, positive=positive, negative=negative, cosplay=cosplay, mode=mode)
+        message = dict(theme=theme, positive=positive, negative=negative, cosplay=cosplay, mode=mode, api_key=api_key)
         self.send_start_cmd(message=message)
         return gr.Chatbot.update(
             visible=True
@@ -209,6 +211,11 @@ class DebateUI(WebUI):
         with gr.Blocks(css=gc.CSS) as demo:
             with gr.Row():
                 with gr.Column():
+                    self.text_api = gr.Textbox(
+                        value = self.cache["api_key"],
+                        placeholder="openai key",
+                        label="Please input valid openai key for gpt-3.5-turbo-16k."
+                    )
                     self.radio_mode = gr.Radio(
                         [Client.AUTO_MODE, Client.SINGLE_MODE],
                         value=Client.AUTO_MODE,
@@ -268,7 +275,7 @@ class DebateUI(WebUI):
                     
             self.btn_start.click(
                 fn=self.start_button_when_click,
-                inputs=[self.text_theme, self.text_positive, self.text_negative, self.radio_choose, self.radio_mode],
+                inputs=[self.text_theme, self.text_positive, self.text_negative, self.radio_choose, self.radio_mode, self.text_api],
                 outputs=[self.chatbot, self.btn_start]
             ).then(
                 fn=self.start_button_after_click,
@@ -350,4 +357,4 @@ class DebateUI(WebUI):
 if __name__ == '__main__':
     ui = DebateUI(client_cmd=["python","gradio_backend.py"])
     ui.construct_ui()
-    ui.run()
+    ui.run()

gradio_backend.py

@@ -2,14 +2,12 @@ import yaml
 import os
 import argparse
 import sys
-sys.path.append("../../../src/agents")
-sys.path.append("../../Gradio_Config")
 from agents.SOP import SOP
 from agents.Agent import Agent
 from agents.Environment import Environment
 from agents.Memory import Memory
 from gradio_base import Client
-from app import DebateUI
+from run_gradio import DebateUI
 
 def process(action):
     response = action.response
@@ -43,7 +41,7 @@ def gradio_process(action,current_state):
                 if item.startswith("<USER>"):
                     content = item.split("<USER>")[1]
                     break
-            # print(f"client: recieved `{content}` from server")
+            # print(f"client: received `{content}` from server")
             action.response = content
             break
         else:
@@ -83,12 +81,12 @@ def init(config):
 def run(agents,sop,environment):
     while True:      
         current_state,current_agent= sop.next(environment,agents)
-        block_when_next(current_agent, current_state)
         if sop.finished:
             print("finished!")
-            Client.send_server(str([99, ' ', ' ', current_state.name]))
+            Client.send_server(str([99, ' ', ' ', "done"]))
             os.environ.clear()
             break
+        block_when_next(current_agent, current_state)
         action = current_agent.step(current_state,"")   #component_dict = current_state[self.role[current_node.name]]   current_agent.compile(component_dict) 
         gradio_process(action,current_state)
         memory = process(action)
@@ -107,12 +105,14 @@ def prepare(agents, sop, environment):
             "negative": f"{parse_data[2]}",
             "agents_name": DebateUI.convert2list4agentname(sop)[0],
             "only_name":  DebateUI.convert2list4agentname(sop)[0],
-            "default_cos_play_id": -1
+            "default_cos_play_id": -1,
+            "api_key": os.environ["API_KEY"]
         }
     )
     client.listening_for_start_()
     client.mode = Client.mode = client.cache["mode"]
     # cover config and then start
+    os.environ["API_KEY"] = client.cache["api_key"]
     if Client.cache["cosplay"] is not None:
         agents[Client.cache["cosplay"]].is_user = True
     sop.states['Negative_Task_Allocation_state'] = sop.states['Affirmative_Task_Allocation_state'].begin_query = \
@@ -135,4 +135,3 @@ if __name__ == '__main__':
 
     run(agents,sop,environment)
 
-