Update app.py

app.py

@@ -9,6 +9,8 @@ import requests
 import csv
 
 my_api_key = ""    # 在这里输入你的 API 密钥
+HIDE_MY_KEY = False # 如果你想在UI中隐藏你的 API 密钥，将此值设置为 True
+
 initial_prompt = "You are a helpful assistant."
 API_URL = "https://api.openai.com/v1/chat/completions"
 HISTORY_DIR = "history"
@@ -38,30 +40,35 @@ if dockerflag:
 
 def parse_text(text):
     lines = text.split("\n")
+    lines = [line for line in lines if line != ""]
     count = 0
+    firstline = False
     for i, line in enumerate(lines):
         if "```" in line:
             count += 1
             items = line.split('`')
             if count % 2 == 1:
-                lines[i] = f'<pre><code class="{items[-1]}">'
+                lines[i] = f'<pre><code class="{items[-1]}" style="display: block; white-space: pre; padding: 0 1em 1em 1em; color: #fff; background: #000;">'
+                firstline = True
             else:
                 lines[i] = f'</code></pre>'
         else:
             if i > 0:
                 if count % 2 == 1:
                     line = line.replace("&", "&amp;")
-                    line = line.replace("\"", "&quot;")
-                    line = line.replace("\'", "&apos;")
+                    line = line.replace("\"", "`\"`")
+                    line = line.replace("\'", "`\'`")
                     line = line.replace("<", "&lt;")
                     line = line.replace(">", "&gt;")
                     line = line.replace(" ", "&nbsp;")
-                if not line.startswith("<br/>"):
-                    lines[i] = '<br/>'+line
-    return "".join(lines)
+                lines[i] = "<br>"+line
+    text = "".join(lines)
+    return text
 
 def predict(inputs, top_p, temperature, openai_api_key, chatbot=[], history=[], system_prompt=initial_prompt, retry=False, summary=False):  # repetition_penalty, top_k
 
+    print(f"chatbot 1: {chatbot}")
+
     headers = {
         "Content-Type": "application/json",
         "Authorization": f"Bearer {openai_api_key}"
@@ -135,7 +142,7 @@ def predict(inputs, top_p, temperature, openai_api_key, chatbot=[], history=[],
                     break
             except Exception as e:
                 chatbot.pop()
-                chatbot.append((history[-1], f"☹️发生了错误\n返回值：{response.text}\n异常：{e}"))
+                chatbot.append((history[-1], f"☹️发生了错误<br>返回值：{response.text}<br>异常：{e}"))
                 history.pop()
                 yield chatbot, history
                 break
@@ -231,7 +238,7 @@ description = """<div align=center>
 with gr.Blocks() as demo:
     gr.HTML(title)
     keyTxt = gr.Textbox(show_label=True, placeholder=f"在这里输入你的OpenAI API-key...",
-                        value=my_api_key, label="API Key", type="password").style(container=True)
+                        value=my_api_key, label="API Key", type="password", visible=not HIDE_MY_KEY).style(container=True)
     chatbot = gr.Chatbot()  # .style(color_map=("#1D51EE", "#585A5B"))
     history = gr.State([])
     promptTemplates = gr.State({})