now can select model

app.py

@@ -13,9 +13,7 @@ import openai
 
 
 DEFAULT_PROMPT = [
-    ["system", "你(assistant)是一名疯狂的摇滚乐手，用户(user)是你的粉丝。"],
-    ["user", "我们来玩一个角色扮演游戏吧！请你扮演一名疯狂的摇滚乐手，而我将扮演你的粉丝。"],
-    ["assistant", "真是个有趣的游戏！我将扮演一名疯狂的摇滚乐手，而你是我的粉丝。听起来真不错！让我们开始吧！"],
+    ["system", "You(assistant) are a helpful AI assistant."],
 ]
 
 
@@ -47,7 +45,7 @@ DEFAULT_PROMPT = [
 
 def on_click_send_btn(
         old_state, api_key_text, chat_input_role, chat_input, prompt_table, chat_use_prompt, chat_use_history, chat_log,
-        temperature, top_p, choices_num, stream, max_tokens, presence_penalty, frequency_penalty, logit_bias,
+        chat_model, temperature, top_p, choices_num, stream, max_tokens, presence_penalty, frequency_penalty, logit_bias,
     ):
 
     print(prompt_table)
@@ -89,7 +87,7 @@ def on_click_send_btn(
     openai.api_key = api_key_text
 
     props = {
-        'model': "gpt-3.5-turbo",
+        'model': chat_model,
         'messages': [xx for xx in map(lambda it: {'role':it[0], 'content':it[1]}, req_hist)],
         'temperature': temperature,
         'top_p': top_p,
@@ -226,6 +224,10 @@ with gradio.Blocks(title="ChatGPT", css=css) as demo:
             with gradio.Column(scale=4):
                 with gradio.Box():
                     gradio.Markdown("See https://platform.openai.com/docs/api-reference/chat/create .")
+                    chat_model = gradio.Dropdown(label="model", choices=[
+                        "gpt-3.5-turbo", "gpt-3.5-turbo-0301",
+                        "gpt-4", "gpt-4-0314", "gpt-4-32k", "gpt-4-32k-0314",
+                    ], value="gpt-3.5-turbo")
                     chat_temperature = gradio.Slider(label="temperature", value=1, minimum=0, maximum=2)
                     chat_top_p = gradio.Slider(label="top_p", value=1, minimum=0, maximum=1)
                     chat_choices_num = gradio.Slider(label="choices num(n)", value=1, minimum=1, maximum=20)
@@ -258,7 +260,7 @@ with gradio.Blocks(title="ChatGPT", css=css) as demo:
                 on_click_send_btn,
                 inputs=[
                     global_state, api_key_text, chat_input_role, chat_input, prompt_table, chat_use_prompt, chat_use_history, chat_log,
-                    chat_temperature, chat_top_p, chat_choices_num, chat_stream, chat_max_tokens, chat_presence_penalty, chat_frequency_penalty, chat_logit_bias,
+                    chat_model, chat_temperature, chat_top_p, chat_choices_num, chat_stream, chat_max_tokens, chat_presence_penalty, chat_frequency_penalty, chat_logit_bias,
                 ],
                 outputs=[global_state, chat_log, chat_log_box, chat_last_resp, chat_last_req, chat_input],
                 api_name="click-send-btn",