Edit application file2

README.md

@@ -5,7 +5,7 @@ colorFrom: blue
 colorTo: pink
 sdk: gradio
 sdk_version: 3.21.0
-app_file: ChuanhuChatbot.py
+app_file: app.py
 pinned: false
 license: apache-2.0
 ---

app.py

@@ -0,0 +1,171 @@
+# -*- coding:utf-8 -*-
+import gradio as gr
+import os
+import logging
+import sys
+import argparse
+from utils import *
+from presets import *
+
+logging.basicConfig(level=logging.INFO, format="%(asctime)s [%(levelname)s] [%(filename)s:%(lineno)d] %(message)s")
+
+my_api_key = ""    # 在这里输入你的 API 密钥
+
+#if we are running in Docker
+if os.environ.get('dockerrun') == 'yes':
+    dockerflag = True
+else:
+    dockerflag = False
+
+authflag = False
+
+if dockerflag:
+    my_api_key = os.environ.get('my_api_key')
+    if my_api_key == "empty":
+        logging.error("Please give a api key!")
+        sys.exit(1)
+    #auth
+    username = os.environ.get('USERNAME')
+    password = os.environ.get('PASSWORD')
+    if not (isinstance(username, type(None)) or isinstance(password, type(None))):
+        authflag = True
+else:
+    if not my_api_key and os.path.exists("api_key.txt") and os.path.getsize("api_key.txt"):
+        with open("api_key.txt", "r") as f:
+            my_api_key = f.read().strip()
+    if os.path.exists("auth.json"):
+        with open("auth.json", "r") as f:
+            auth = json.load(f)
+            username = auth["username"]
+            password = auth["password"]
+            if username != "" and password != "":
+                authflag = True
+
+gr.Chatbot.postprocess = postprocess
+
+with gr.Blocks(css=customCSS) as demo:
+    history = gr.State([])
+    token_count = gr.State([])
+    promptTemplates = gr.State(load_template(get_template_names(plain=True)[0], mode=2))
+    TRUECOMSTANT = gr.State(True)
+    FALSECONSTANT = gr.State(False)
+    topic = gr.State("未命名对话历史记录")
+
+    with gr.Row():
+        gr.HTML(title)
+        status_display = gr.Markdown("status: ready", elem_id="status_display")
+
+    with gr.Row(scale=1).style(equal_height=True):
+        with gr.Column(scale=5):
+            with gr.Row(scale=1):
+                chatbot = gr.Chatbot().style(height=600)  # .style(color_map=("#1D51EE", "#585A5B"))
+            with gr.Row(scale=1):
+                with gr.Column(scale=12):
+                    user_input = gr.Textbox(show_label=False, placeholder="在这里输入").style(
+                        container=False)
+                with gr.Column(min_width=50, scale=1):
+                    submitBtn = gr.Button("🚀", variant="primary")
+            with gr.Row(scale=1):
+                emptyBtn = gr.Button("🧹 新的对话",)
+                retryBtn = gr.Button("🔄 重新生成")
+                delLastBtn = gr.Button("🗑️ 删除一条对话")
+                reduceTokenBtn = gr.Button("♻️ 总结对话")
+
+        with gr.Column():
+            with gr.Column(min_width=50,scale=1):
+                with gr.Tab(label="ChatGPT"):
+                    keyTxt = gr.Textbox(show_label=True, placeholder=f"OpenAI API-key...",value=my_api_key, type="password", visible=not HIDE_MY_KEY, label="API-Key")
+                    model_select_dropdown = gr.Dropdown(label="选择模型", choices=MODELS, multiselect=False, value=MODELS[0])
+                    with gr.Accordion("参数", open=False):
+                        top_p = gr.Slider(minimum=-0, maximum=1.0, value=1.0, step=0.05,
+                                        interactive=True, label="Top-p (nucleus sampling)",)
+                        temperature = gr.Slider(minimum=-0, maximum=2.0, value=1.0,
+                                                step=0.1, interactive=True, label="Temperature",)
+                    use_streaming_checkbox = gr.Checkbox(label="实时传输回答", value=True, visible=enable_streaming_option)
+                    use_websearch_checkbox = gr.Checkbox(label="使用在线搜索", value=False)
+
+                with gr.Tab(label="Prompt"):
+                    systemPromptTxt = gr.Textbox(show_label=True, placeholder=f"在这里输入System Prompt...", label="System prompt", value=initial_prompt).style(container=True)
+                    with gr.Accordion(label="加载Prompt模板", open=True):
+                        with gr.Column():
+                            with gr.Row():
+                                with gr.Column(scale=6):
+                                    templateFileSelectDropdown = gr.Dropdown(label="选择Prompt模板集合文件", choices=get_template_names(plain=True), multiselect=False, value=get_template_names(plain=True)[0])
+                                with gr.Column(scale=1):
+                                    templateRefreshBtn = gr.Button("🔄 刷新")
+                            with gr.Row():
+                                with gr.Column():
+                                    templateSelectDropdown = gr.Dropdown(label="从Prompt模板中加载", choices=load_template(get_template_names(plain=True)[0], mode=1), multiselect=False, value=load_template(get_template_names(plain=True)[0], mode=1)[0])
+
+                with gr.Tab(label="保存/加载"):
+                    with gr.Accordion(label="保存/加载对话历史记录", open=True):
+                        gr.Markdown("对话��史默认保存在history文件夹中。")
+                        with gr.Column():
+                            with gr.Row():
+                                with gr.Column(scale=6):
+                                    saveFileName = gr.Textbox(
+                                        show_label=True, placeholder=f"设置文件名: 默认为.json，可选为.md", label="设置保存文件名", value="对话历史记录").style(container=True)
+                                with gr.Column(scale=1):
+                                    saveHistoryBtn = gr.Button("💾 保存对话")
+                                    exportMarkdownBtn = gr.Button("📝 导出为Markdown")
+                            with gr.Row():
+                                with gr.Column(scale=6):
+                                    historyFileSelectDropdown = gr.Dropdown(label="从列表中加载对话", choices=get_history_names(plain=True), multiselect=False, value=get_history_names(plain=True)[0])
+                                with gr.Column(scale=1):
+                                    historyRefreshBtn = gr.Button("🔄 刷新")
+                            with gr.Row():
+                                with gr.Column():
+                                    downloadFile = gr.File(interactive=True)
+
+    gr.Markdown(description)
+
+    # Chatbot
+    user_input.submit(predict, [keyTxt, systemPromptTxt, history, user_input, chatbot, token_count, top_p, temperature, use_streaming_checkbox, model_select_dropdown, use_websearch_checkbox], [chatbot, history, status_display, token_count], show_progress=True)
+    user_input.submit(reset_textbox, [], [user_input])
+
+    submitBtn.click(predict, [keyTxt, systemPromptTxt, history, user_input, chatbot, token_count, top_p, temperature, use_streaming_checkbox, model_select_dropdown, use_websearch_checkbox], [chatbot, history, status_display, token_count], show_progress=True)
+    submitBtn.click(reset_textbox, [], [user_input])
+
+    emptyBtn.click(reset_state, outputs=[chatbot, history, token_count, status_display], show_progress=True)
+
+    retryBtn.click(retry, [keyTxt, systemPromptTxt, history, chatbot, token_count, top_p, temperature, use_streaming_checkbox, model_select_dropdown], [chatbot, history, status_display, token_count], show_progress=True)
+
+    delLastBtn.click(delete_last_conversation, [chatbot, history, token_count], [
+                     chatbot, history, token_count, status_display], show_progress=True)
+
+    reduceTokenBtn.click(reduce_token_size, [keyTxt, systemPromptTxt, history, chatbot, token_count, top_p, temperature, use_streaming_checkbox, model_select_dropdown], [chatbot, history, status_display, token_count], show_progress=True)
+
+    # Template
+    templateRefreshBtn.click(get_template_names, None, [templateFileSelectDropdown])
+    templateFileSelectDropdown.change(load_template, [templateFileSelectDropdown],  [promptTemplates, templateSelectDropdown], show_progress=True)
+    templateSelectDropdown.change(get_template_content, [promptTemplates, templateSelectDropdown, systemPromptTxt],  [systemPromptTxt], show_progress=True)
+
+    # S&L
+    saveHistoryBtn.click(save_chat_history, [saveFileName, systemPromptTxt, history, chatbot], downloadFile, show_progress=True)
+    saveHistoryBtn.click(get_history_names, None, [historyFileSelectDropdown])
+    exportMarkdownBtn.click(export_markdown, [saveFileName, systemPromptTxt, history, chatbot], downloadFile, show_progress=True)
+    historyRefreshBtn.click(get_history_names, None, [historyFileSelectDropdown])
+    historyFileSelectDropdown.change(load_chat_history, [historyFileSelectDropdown, systemPromptTxt, history, chatbot],  [saveFileName, systemPromptTxt, history, chatbot], show_progress=True)
+    downloadFile.change(load_chat_history, [downloadFile, systemPromptTxt, history, chatbot],  [saveFileName, systemPromptTxt, history, chatbot])
+
+
+logging.info(colorama.Back.GREEN + "\n川虎的温馨提示：访问 http://localhost:7860 查看界面" + colorama.Style.RESET_ALL)
+# 默认开启本地服务器，默认可以直接从IP访问，默认不创建公开分享链接
+demo.title = "川虎ChatGPT 🚀"
+
+if __name__ == "__main__":
+    #if running in Docker
+    if dockerflag:
+        if authflag:
+            demo.queue().launch(server_name="0.0.0.0", server_port=7860,auth=(username, password))
+        else:
+            demo.queue().launch(server_name="0.0.0.0", server_port=7860, share=False)
+    #if not running in Docker
+    else:
+        if authflag:
+            demo.queue().launch(share=False, auth=(username, password))
+        else:
+            demo.queue().launch(share=True) # 改为 share=True 可以创建公开分享链接
+        #demo.queue().launch(server_name="0.0.0.0", server_port=7860, share=False) # 可自定义端口
+        #demo.queue().launch(server_name="0.0.0.0", server_port=7860,auth=("在这里填写用户名", "在这里填写密码")) # 可设置用户名与密码
+        #demo.queue().launch(auth=("在这里填写用户名", "在这里填写密码")) # 适合Nginx反向代理