Update app.py

app.py

@@ -10,17 +10,21 @@ _assistant, _thread = None, None
 
 def create_assistant(client):
     assistant = client.beta.assistants.create(
-        name="Python Code Generator & Executor",
-        instructions="You are a Python programming language expert that generates Pylint-compliant code and explains it. Only execute code when expicitly asked to.",
+        name="Python Code Generator",
+        instructions="You are a Python programming language expert that generates Pylint-compliant code and explains it. Only execute code when explicitly asked to.",
         model="gpt-4o",
         tools=[{"type": "code_interpreter"}],
     )
+    
     show_json("assistant", assistant)
+    
     return assistant
     
 def create_thread(client):
     thread = client.beta.threads.create()
+    
     show_json("thread", thread)
+    
     return thread
 
 def create_message(client, thread, msg):
@@ -29,7 +33,9 @@ def create_message(client, thread, msg):
         thread_id=thread.id,
         content=msg,
     )
+    
     show_json("message", message)
+    
     return message
 
 def create_run(client, assistant, thread):
@@ -37,7 +43,9 @@ def create_run(client, assistant, thread):
         assistant_id=assistant.id,
         thread_id=thread.id,
     )
+    
     show_json("run", run)
+    
     return run
 
 def wait_on_run(client, thread, run):
@@ -46,8 +54,11 @@ def wait_on_run(client, thread, run):
             thread_id=thread.id,
             run_id=run.id,
         )
+    
         time.sleep(0.25)
+    
     show_json("run", run)
+    
     return run
 
 def list_run_steps(client, thread, run):
@@ -56,24 +67,30 @@ def list_run_steps(client, thread, run):
         run_id=run.id,
         order="asc",
     )
+    
     for step in run_steps.data:
         step_details = step.step_details
         show_json("step_details", step_details)
+    
     return run_steps
     
 def list_messages(client, thread):
     messages = client.beta.threads.messages.list(
         thread_id=thread.id
     )
+    
     show_json("messages", messages)
+    
     return messages
     
 def extract_content_values(data):
     content_values = []
+    
     for item in data.data:
         for content in item.content:
             if content.type == "text":
                 content_values.append(content.text.value)
+    
     return content_values
 
 def chat(message, history, openai_api_key):
@@ -87,7 +104,6 @@ def chat(message, history, openai_api_key):
         
     create_message(_client, _thread, message)
 
-    # async
     run = create_run(_client, _assistant, _thread)
     run = wait_on_run(_client, _thread, run)
 
@@ -95,14 +111,17 @@ def chat(message, history, openai_api_key):
     
     messages = list_messages(_client, _thread)
 
-    return extract_content_values(messages)[0]
+    content_values = extract_content_values(messages)
+    
+    return content_values[0]
         
 gr.ChatInterface(
     chat,
     chatbot=gr.Chatbot(height=300),
     textbox=gr.Textbox(placeholder="Ask anything", container=False, scale=7),
-    title="Python Code Generator & Executor",
-    description="Generate, explain, fix, optimize, document, test, help, execute, ...",
+    textbox2=gr.Textbox(placeholder="OpenAI API Key", container=False, scale=7),
+    title="Python Code Generator",
+    description="Generate, explain, fix, optimize, document, test, help, ... Can execute code when asked to.",
     clear_btn="Clear",
     retry_btn="Retry",
     undo_btn="Undo",