improve interface ui

app.py

@@ -76,6 +76,7 @@ def completion(model_engine, prompt, max_tokens, temperature, top_p, top_k, n, s
 def app():
     # 左侧栏
     st.sidebar.title("参数设置")
+    model_engine_list = st.sidebar.multiselect("请选择要对比的模型", ["davinci-003", "WeLM"], default=["davinci-003", "WeLM"])
     default_top_p = 0.95
     default_top_k = 0
     default_temperature = 0.85
@@ -89,18 +90,21 @@ def app():
     stop_tokens = ""
     if st.sidebar.checkbox("使用换行符作为截断", value=True):
         stop_tokens = "\n"
-    model_engine_list = st.sidebar.multiselect("请选择模型", ["davinci-003", "WeLM"], default=["davinci-003", "WeLM"])
 
     # 主界面
     st.title("对比不同模型生成文本的效果")
     st.text('Tips: ')
+    st.text("* Davinci是是GPT-3语言生成模型，可以一定程度上理解用户的指令")
     st.text("* WeLM不是一个直接的对话机器人，而是一个补全用户输入信息的生成模型")
     st.text("* 因此 prompt 需要经过一定的设计，才能有比较好的效果")
     st.text("* 修改Prompt可以更多参考 https://welm.weixin.qq.com/docs/introduction/ 或者使用下方的例子")
 
     examples()
     prompt = st.text_area("请输入Prompt：", key="prompt")
-    if st.button("生成", disabled=(len(prompt)==0)):
+    if st.button("生成"):
+        if prompt.strip() == '':
+            st.error("请输入内容")
+            st.stop()
         columns = st.columns(len(model_engine_list))
         for col, model_engine in zip(columns, model_engine_list):
             with col: