Update app.py

app.py

@@ -214,7 +214,7 @@ def chat(message, history):
     if not message:
         raise gr.Error("Message is required.")
     
-    global client, assistant, thread     
+    global assistant, thread     
     
     if assistant == None:
         assistant = load_assistant(client)
@@ -225,13 +225,13 @@ def chat(message, history):
     create_message(client, thread, message)
 
     run = create_run(client, assistant, thread)
-    run = wait_on_run(client, thread, run)
 
+    run = wait_on_run(client, thread, run)
     run_steps = get_run_steps(client, thread, run)
 
+    ### TODO
     tool_call_ids, tool_call_results = execute_tool_calls(run_steps)
 
-    ### TODO
     print("###")
     print(len(tool_call_ids))
     print(tool_call_ids)
@@ -257,32 +257,32 @@ def chat(message, history):
         run = wait_on_run(client, thread, run)
         run_steps = get_run_steps(client, thread, run)
     ###
-    tool_call_ids, tool_call_results = execute_tool_calls(run_steps)
-
-    print("###")
-    print(len(tool_call_ids))
-    print(tool_call_ids)
-    print(tool_call_ids[1])
-    print(tool_call_results)
-    print(tool_call_results[1])
-    print("###")
+        tool_call_ids, tool_call_results = execute_tool_calls(run_steps)
     
-    if tool_call_ids[1]:
-        # https://platform.openai.com/docs/api-reference/runs/submitToolOutputs
-        run = client.beta.threads.runs.submit_tool_outputs(
-            thread_id=thread.id,
-            run_id=run.id,
-            #tool_outputs=generate_tool_outputs(tool_call_ids, tool_call_results)
-            tool_outputs=[
-                {
-                    "tool_call_id": tool_call_ids[1],
-                    "output": tool_call_results[1].to_json()
-                }
-            ]
-        )
-    
-        run = wait_on_run(client, thread, run)
-        run_steps = get_run_steps(client, thread, run)    
+        print("###")
+        print(len(tool_call_ids))
+        print(tool_call_ids)
+        print(tool_call_ids[1])
+        print(tool_call_results)
+        print(tool_call_results[1])
+        print("###")
+        
+        if tool_call_ids[1]:
+            # https://platform.openai.com/docs/api-reference/runs/submitToolOutputs
+            run = client.beta.threads.runs.submit_tool_outputs(
+                thread_id=thread.id,
+                run_id=run.id,
+                #tool_outputs=generate_tool_outputs(tool_call_ids, tool_call_results)
+                tool_outputs=[
+                    {
+                        "tool_call_id": tool_call_ids[1],
+                        "output": tool_call_results[1].to_json()
+                    }
+                ]
+            )
+        
+            run = wait_on_run(client, thread, run)
+            run_steps = get_run_steps(client, thread, run)    
     ###
     
     messages = get_messages(client, thread)
@@ -304,9 +304,9 @@ gr.ChatInterface(
         title="Python Code Generator",
         description=(
             "The assistant can **generate, explain, fix, optimize, document, and test code**. "
-            "It can also <b>execute code</b>. "
-            "It has access to <b>today tool</b> (to get current date) and "
-            "<b>yfinance download tool</b> (to get stock data)."
+            "It can also **execute code**. "
+            "It has access to <b>today tool</b> (get current date) and "
+            "to **yfinance download tool** (get stock data)."
         ),
         clear_btn="Clear",
         retry_btn=None,