Update main.py

main.py

@@ -46,6 +46,8 @@ with gr.Blocks(theme=set_theme, analytics_enabled=False) as demo:
             chatbot.style()
             history = gr.State([])
         with gr.Column(scale=1):
+            with gr.Column(scale=12):
+                api = gr.Textbox(show_label=False, placeholder="Input OpenAI Key.").style(container=False)
             with gr.Row():
                 txt = gr.Textbox(show_label=False, placeholder="Input question here.").style(container=False)
             with gr.Row():
@@ -75,7 +77,7 @@ with gr.Blocks(theme=set_theme, analytics_enabled=False) as demo:
                 top_p = gr.Slider(minimum=-0, maximum=1.0, value=1.0, step=0.01,interactive=True, label="Top-p (nucleus sampling)",)
                 temperature = gr.Slider(minimum=-0, maximum=2.0, value=1.0, step=0.01, interactive=True, label="Temperature",)
 
-    predict_args = dict(fn=predict, inputs=[txt, top_p, temperature, chatbot, history, system_prompt], outputs=[chatbot, history, statusDisplay], show_progress=True)
+    predict_args = dict(fn=predict, inputs=[api, txt, top_p, temperature, chatbot, history, system_prompt], outputs=[chatbot, history, statusDisplay], show_progress=True)
     empty_txt_args = dict(fn=lambda: "", inputs=[], outputs=[txt]) # 用于在提交后清空输入栏
 
     cancel_handles.append(txt.submit(**predict_args))