add retry

app.py

@@ -81,8 +81,11 @@ def to_md(content):
     count_backtick = 0
     output_spans = []
     for i in range(len(content)):
-        if content[i]=="\n" and not is_inside_code_block:
-            output_spans.append("<br>")
+        if content[i]=="\n":
+            if not is_inside_code_block:
+                output_spans.append("<br>")
+            else:
+                output_spans.append("\n\n")
         elif content[i]=="`":
             count_backtick += 1
             if count_backtick == 3:
@@ -96,7 +99,7 @@ def to_md(content):
 
 def predict(question, history=[], behavior=[]):
     if question.startswith(f"{openai.api_key}:"):
-        return adminInstruct(question)
+        return adminInstruct(question, history)
     history = ask(question, history, behavior)
     response = [(to_md(history[i]),to_md(history[i+1])) for i in range(0,len(history)-1,2)]
     return "", history, response, None
@@ -105,13 +108,13 @@ def predict(question, history=[], behavior=[]):
 def retry(question, history=[], behavior=[]):
     if len(history)<2:
         return "", history, [], None
-    question = history[-1]
+    question = history[-2]
     history = history[:-2]
     return predict(question, history, behavior)
 
 
-def adminInstruct(question):
-    if "download-log" in question:
+def adminInstruct(question, history):
+    if "download all dialogue" in question:
         with open("./all_dialogue.jsonl", "w", encoding="utf-8") as f:
             for dialogue in all_dialogue:
                 f.write(dialogue + "\n")