Update app.py

app.py

@@ -7,6 +7,7 @@ from utils import function_to_schema, show_json
 # Tools
 
 triage_agent, sales_agent, issues_repairs_agent = None, None, None
+triage_thread, sales_thread, issues_repairs_thread = None, None, None
 
 def transfer_to_triage_agent():
     """Call this if the user brings up a topic outside of your purview,
@@ -19,14 +20,14 @@ def transfer_to_triage_agent():
 def transfer_to_sales_agent():
     """Use for anything sales or buying related."""
     print("=> transfer_to_sales_agent")
-    global sales_agent
+    global sales_agent, sales_thread
     set_current_agent(sales_agent)
     set_current_thread(sales_thread)
 
 def transfer_to_issues_repairs_agent():
     """Use for issues, repairs, or refunds."""
     print("=> transfer_to_issues_repairs_agent")
-    global issues_repairs_agent
+    global issues_repairs_agent, issues_repairs_thread
     set_current_agent(issues_repairs_agent)
     set_current_thread(issues_repairs_thread)
 
@@ -228,26 +229,26 @@ current_agent, current_thread = None, None
 
 def set_current_agent(agent):
     global current_agent
-    print("### set_current_agent ###")
-    current_agent = agent
+    #print("### set_current_agent ###")
+    #current_agent = agent
     show_json("Current Agent", current_agent)
 
 def set_current_thread(thread):
     global current_thread
-    print("### set_current_thread ###")
-    current_thread = thread
+    #print("### set_current_thread ###")
+    #current_thread = thread
     show_json("Current Thread", current_thread)
 
 def get_current_agent():
     global current_agent
-    print("### get_current_agent ###")
-    show_json("Current Agent", current_agent)
+    #print("### get_current_agent ###")
+    #show_json("Current Agent", current_agent)
     return current_agent
 
 def get_current_thread():
     global current_thread
-    print("### get_current_thread ###")
-    show_json("Current Thread", current_thread)
+    #print("### get_current_thread ###")
+    #show_json("Current Thread", current_thread)
     return current_thread
 
 #
@@ -275,7 +276,7 @@ def chat(message, history, openai_api_key):
     thread = get_current_thread()
     show_json("Current Thread", thread)
 
-    print("111")
+    #print("111")
     create_message(client, thread, message)
 
     # async
@@ -293,8 +294,8 @@ def chat(message, history, openai_api_key):
 
 gr.ChatInterface(
     chat,
-    chatbot=gr.Chatbot(height=300),
-    textbox=gr.Textbox(placeholder="Question", container=False, scale=7),
+    chatbot=gr.Chatbot(height=500),
+    textbox=gr.Textbox(placeholder="Type here", container=False, scale=7),
     title="Multi-Agent Orchestration",
     description="Demo using hand-off pattern: triage agent, sales agent, and issues & repairs agent",
     clear_btn=None,