Update app.py

app.py

@@ -25,9 +25,14 @@ from assistants import (
     extract_content_values,
 )
 
-def chat(message, history):
+def chat(message, history, openai_api_key):
     if not message:
         raise gr.Error("Message is required.")
+
+    if not openai_api_key:
+        raise gr.Error("OpenAI API key is required.")
+    
+    print(openai_api_key)
     
     global assistant, thread     
 
@@ -35,11 +40,8 @@ def chat(message, history):
     # see https://platform.openai.com/playground/assistants.
     # On subsequent runs, load assistant.
     if assistant == None:
-        try: 
-            #assistant = create_assistant()
-            assistant = load_assistant()
-        except:
-            raise gr.Error("Please clone and bring your own credentials.")
+        #assistant = create_assistant()
+        assistant = load_assistant()
 
     # TODO: Use Gradio session to support multiple users
     if thread == None or len(history) == 0:
@@ -81,4 +83,7 @@ gr.ChatInterface(
                   ["1. Execute with tools: Get key announcements from latest OpenAI Dev Day. 2. Show the web references."]
                  ],
         cache_examples=True,
+        additional_inputs=[
+            gr.Textbox(label="OpenAI API Key", type="password"),
+        ],
     ).launch()