Update app.py

app.py

@@ -29,7 +29,8 @@ def generate_image_url(prompt):
     image_url = response["data"][0]["url"]
     return image_url
 
-def greet(user_id, user_question, clear_history):
+def greet(user_id, user_question, api_key, clear_history):
+    openai.api_key = api_key
     global user_contexts
     if user_id not in user_contexts:
         user_contexts[user_id] = [
@@ -70,6 +71,7 @@ demo = gr.Interface(
     fn=greet,
     inputs=[
         gr.Textbox(lines=1, label='请输入用户ID', placeholder='请输入用户ID'),
+        gr.Textbox(lines=1, label='请输入你的OpenAI API密钥', placeholder='请输入你的OpenAI API密钥'),
         gr.Textbox(lines=15, label='请输入问题', placeholder='请输入您的问题'),
         gr.Checkbox(label='清空聊天记录', default=False)
     ],