Update app.py

app.py

@@ -1,6 +1,4 @@
-# TODO:
-#
-# 1. Gradio session / multi-user thread
+# TODO: Gradio session / multi-user thread
 
 # Reference:
 #
@@ -78,16 +76,12 @@ def create_assistant(openai_client):
 
 def load_assistant(openai_client):   
     assistant = openai_client.beta.assistants.retrieve(assistant_id)
-
     show_json("assistant", assistant)
-    
     return assistant
 
 def create_thread(openai_client):
     thread = openai_client.beta.threads.create()
-    
     show_json("thread", thread)
-    
     return thread
 
 def create_message(openai_client, thread, msg):        
@@ -98,7 +92,6 @@ def create_message(openai_client, thread, msg):
     )
     
     show_json("message", message)
-    
     return message
 
 def create_run(openai_client, assistant, thread):
@@ -109,7 +102,6 @@ def create_run(openai_client, assistant, thread):
     )
     
     show_json("run", run)
-    
     return run
 
 def wait_on_run(openai_client, thread, run):
@@ -136,7 +128,6 @@ def get_run_steps(openai_client, thread, run):
     )
 
     show_json("run_steps", run_steps)
-    
     return run_steps
 
 def execute_tool_call(tool_call):
@@ -175,7 +166,6 @@ def get_messages(openai_client, thread):
     )
     
     show_json("messages", messages)
-    
     return messages
                         
 def extract_content_values(data):
@@ -204,21 +194,11 @@ def generate_tool_outputs(tool_call_ids, tool_call_results):
                 "tool_call_id": tool_call_id,
                 "output": tool_call_result.to_json()
             }
-
-            print("###")
-            print(tool_call_id)
-            print(tool_call_result.to_json())
-            print("###")
         except AttributeError:
             tool_output = {
                 "tool_call_id": tool_call_id,
                 "output": tool_call_result
             }
-
-            print("###")
-            print(tool_call_id)
-            print(tool_call_result)
-            print("###")
             
         tool_outputs.append(tool_output)
     
@@ -250,14 +230,6 @@ def chat(message, history):
     tool_call_ids, tool_call_results = execute_tool_calls(run_steps)
     
     if len(tool_call_ids) > 0:
-        print("###")
-        print(len(tool_call_ids))
-        print(tool_call_ids)
-        print(tool_call_ids[0])
-        print(tool_call_results)
-        print(tool_call_results[0])
-        print("###")
-        
         # https://platform.openai.com/docs/api-reference/runs/submitToolOutputs
         tool_output = {}
         
@@ -284,14 +256,6 @@ def chat(message, history):
         tool_call_ids, tool_call_results = execute_tool_calls(run_steps)
             
         if len(tool_call_ids) > 1:
-            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("###")
-
             # https://platform.openai.com/docs/api-reference/runs/submitToolOutputs
             tool_output = {}
             
@@ -325,7 +289,6 @@ def chat(message, history):
     if len(image_values) > 0:
         download_link = f"<p>Download: https://platform.openai.com/storage/files/{image_values[0]}</p>"
     
-    #return f"{text_values[0]}{download_link}"
     return f"{'<hr>'.join(list(reversed(text_values))[1:])}{download_link}"
 
 gr.ChatInterface(
@@ -334,8 +297,8 @@ gr.ChatInterface(
         textbox=gr.Textbox(placeholder="Ask anything", container=False, scale=7),
         title="Python Coding Assistant",
         description=(
-            "The assistant can **generate, explain, fix, optimize, and document Python code, "
-            "create unit test cases, and answer general coding-related questions.** "
+            "The assistant can **generate, explain, fix, optimize,** and **document Python code, "
+            "create unit test cases,** and **answer general coding-related questions.** "
             "It can also **execute code**. "
             "The assistant has access to a <b>today tool</b> (get current date), to a "
             "**yfinance download tool** (get stock data), and to a "
@@ -353,5 +316,5 @@ gr.ChatInterface(
                   ["Execute with tools: Create a plot showing stock gain QTD for NVDA and AMD, x-axis is \"Day\" and y-axis is \"Gain %\""],
                   ["Execute with tools: Get key announcements from the latest OpenAI Dev Day"]
                  ],
-        cache_examples=True,
+        cache_examples=False,
     ).launch()