add retry

app.py

@@ -102,12 +102,12 @@ def predict(question, history=[], behavior=[]):
         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
+    return "", history, response, gr.File.update(value=None, visible=False)
 
 
 def retry(question, history=[], behavior=[]):
     if len(history)<2:
-        return "", history, [], None
+        return "", history, [], gr.File.update(value=None, visible=False)
     question = history[-2]
     history = history[:-2]
     return predict(question, history, behavior)
@@ -115,12 +115,13 @@ def retry(question, history=[], behavior=[]):
 
 def adminInstruct(question, history):
     if "download all dialogue" in question:
-        with open("./all_dialogue.jsonl", "w", encoding="utf-8") as f:
+        filename = f"./all_dialogue_{len(all_dialogue)}.jsonl"
+        with open(filename, "w", encoding="utf-8") as f:
             for dialogue in all_dialogue:
                 f.write(dialogue + "\n")
         response = [(to_md(history[i]),to_md(history[i+1])) for i in range(0,len(history)-1,2)]
-        return "", history, response, gr.File.update(value="./all_dialogue.jsonl", visible=True)
-    return "", history, response, None
+        return "", history, response, gr.File.update(value=filename, visible=True)
+    return "", history, response, gr.File.update(value=None, visible=False)
 
 
 with gr.Blocks() as demo: