Update app.py

app.py

@@ -104,9 +104,9 @@ def create_sales_agent(client):
     return client.beta.assistants.create(
         name="Sales Agent",
         instructions=(
-            "You are a sales agent for ACME Inc."
-            "Always answer in a sentence or less."
-            "Follow the following routine with the user:"
+            "You are a sales agent for ACME Inc. "
+            "Always answer in a sentence or less. "
+            "Follow the following routine with the user: "
             "1. Ask them about any problems in their life related to catching roadrunners.\n"
             "2. Casually mention one of ACME's crazy made-up products can help.\n"
             " - Don't mention price.\n"
@@ -144,7 +144,7 @@ def create_issues_repairs_agent(client):
 
 def create_thread(client):
     thread = client.beta.threads.create()
-    #show_json("thread", thread)
+    show_json("thread", thread)
     
     return thread
 
@@ -154,7 +154,7 @@ def create_message(client, thread, msg):
         thread_id=thread.id,
         content=msg,
     )
-    #show_json("message", message)
+    show_json("message", message)
     
     return message
 
@@ -163,7 +163,7 @@ def create_run(client, assistant, thread):
         assistant_id=assistant.id,
         thread_id=thread.id,
     )
-    #show_json("run", run)
+    show_json("run", run)
     
     return run
 
@@ -174,7 +174,7 @@ def wait_on_run(client, thread, run):
             run_id=run.id,
         )
         time.sleep(0.25)
-    #show_json("run", run)
+    show_json("run", run)
     
     return run
 
@@ -195,10 +195,10 @@ def list_run_steps(client, thread, run):
 
     for step in run_steps.data:
         step_details = step.step_details
-        #show_json("step_details", step_details)
+        show_json("step_details", step_details)
 
         if not hasattr(step_details, "tool_calls"):
-            break
+            continue
 
         for tool_call in step_details.tool_calls:
             execute_tool_call(tool_call)
@@ -209,7 +209,7 @@ def list_messages(client, thread):
     messages = client.beta.threads.messages.list(
         thread_id=thread.id
     )
-    #show_json("messages", messages)
+    show_json("messages", messages)
     
     return messages
 
@@ -229,26 +229,22 @@ current_agent, current_thread = None, None
 
 def set_current_agent(agent):
     global current_agent
-    #print("### set_current_agent ###")
     current_agent = agent
-    #show_json("Current Agent", current_agent)
+    show_json("Current Agent", current_agent)
 
 def set_current_thread(thread):
     global current_thread
-    #print("### set_current_thread ###")
     current_thread = thread
-    #show_json("Current Thread", current_thread)
+    show_json("Current Thread", current_thread)
 
 def get_current_agent():
     global current_agent
-    #print("### get_current_agent ###")
-    #show_json("Current Agent", 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)
+    show_json("Current Thread", current_thread)
     return current_thread
 
 #
@@ -271,30 +267,24 @@ def chat(message, history, openai_api_key):
     global client
 
     assistant = get_current_agent()
-    show_json("Current Agent", assistant)
     
     thread = get_current_thread()
-    show_json("Current Thread", thread)
 
-    #print("111")
     create_message(client, thread, message)
 
     # async
     run = create_run(client, assistant, thread)
     run = wait_on_run(client, thread, run)
 
-    print("222")
     list_run_steps(client, thread, run)
 
-    print("333")
     messages = list_messages(client, thread)
 
-    print("444")
     return extract_content_values(messages)[0]
 
 gr.ChatInterface(
     chat,
-    chatbot=gr.Chatbot(height=500),
+    chatbot=gr.Chatbot(),
     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",