Update app.py

app.py

@@ -1,12 +1,12 @@
 import gradio as gr
-import openai, os, time
+import datetime, openai, os, time
 
 from openai import OpenAI
 from utils import show_json
 
-_client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
+client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
-_assistant, _thread = None, None
+assistant, thread = None, None
 
 def create_assistant(client):
     assistant = client.beta.assistants.create(
@@ -19,7 +19,6 @@ def create_assistant(client):
         model="gpt-4o",
         tools=[
                   {"type": "code_interpreter"}, 
-                  {"type": "retrieval"},
               ],
     )
     
@@ -100,50 +99,43 @@ def extract_content_values(data):
     
     return content_values
 
-def chat(message, history, openai_api_key):
-    global _client, _assistant, _thread     
+def chat(message, history):
+    global client, assistant, thread     
        
-    if _assistant == None:
-        _assistant = create_assistant(_client)
+    if assistant == None:
+        assistant = create_assistant(client)
 
-    if _thread == None:
-        _thread = create_thread(_client)
+    if thread == None:
+        thread = create_thread(client)
         
-    create_message(_client, _thread, message)
+    create_message(client, thread, message)
 
-    run = create_run(_client, _assistant, _thread)
-    run = wait_on_run(_client, _thread, run)
+    run = create_run(client, assistant, thread)
+    run = wait_on_run(client, thread, run)
 
-    list_run_steps(_client, _thread, run)
+    list_run_steps(client, thread, run)
     
-    messages = list_messages(_client, _thread)
+    messages = list_messages(client, thread)
 
     content_values = extract_content_values(messages)
     
     return content_values[0]
 
-def vote(data: gr.LikeData):
-    print("voted")
-
 gr.ChatInterface(
         fn=chat,
-        chatbot=gr.Chatbot(height=300),
-        #textbox=gr.Textbox(placeholder="Ask anything", container=False, scale=7),
+        chatbot=gr.Chatbot(height=500),
+        textbox=gr.Textbox(placeholder="Ask anything", 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",
+        description="The assistant can generate, explain, fix, optimize, document, test, and generally help with code. It can also search the internet and execute code.",
+        #clear_btn="Clear",
+        #retry_btn="Retry",
+        #undo_btn="Undo",
         examples=[
-                  ["Generate: NumPy/Pandas/Matplotlib & yfinance trading app", "sk-<BringYourOwn>"],
-                  ["Explain: r'^(?=.*[A-Z])(?=.*[a-z])(?=.*[0-9])(?=.*[\\W]).{8,}$'", "sk-<BringYourOwn>"],
-                  ["Fix: x = [5, 2, 1, 3, 4]; print(x.sort())", "sk-<BringYourOwn>"],
-                  ["Optimize: x = []; for i in range(0, 10000): x.append(i)", "sk-<BringYourOwn>"],
-                  ["Execute: Code to generate the first 20 fibbonaci numbers", "sk-<BringYourOwn>"],
+                  ["Generate: NumPy/Pandas/Matplotlib & yfinance trading app"],
+                  ["Explain: r'^(?=.*[A-Z])(?=.*[a-z])(?=.*[0-9])(?=.*[\\W]).{8,}$'"],
+                  ["Fix: x = [5, 2, 1, 3, 4]; print(x.sort())"],
+                  ["Optimize: x = []; for i in range(0, 10000): x.append(i)"],
+                  ["Execute: Code to generate the first 20 fibbonaci numbers"],
+                  [f"Today is {datetime.date.today()}. Create a plot showing stock gain YTD for NVDA, MSFT, AAPL, and GOOG, x-axis is 'Day' and y-axis is 'YTD Gain %'. Return the result, don't persist to storage."],
                  ],
-        cache_examples=False,
-        additional_inputs=[
-            gr.Textbox("sk-", label="OpenAI API Key", type = "password"),
-        ],
-        multimodal=True,
     ).launch()