add retry

app.py

@@ -8,6 +8,8 @@ import os
 openai.api_key = os.getenv('API_KEY')
 openai.request_times = 0
 
+all_dialogue = []
+
 def ask(question, history, behavior):
     openai.request_times += 1
     print(f"request times {openai.request_times}: {datetime.datetime.now()}: {question}")
@@ -30,6 +32,7 @@ def ask(question, history, behavior):
         print(e)
         response = 'Timeout! Please wait a few minutes and retry'
     history = history + [question, response]
+    record_dialogue(history)
     return history
 
 def num_tokens_from_messages(messages, model="gpt-3.5-turbo"):
@@ -60,6 +63,15 @@ def forget_long_term(messages, max_num_tokens=4000):
             messages = messages[1:]
     return messages
 
+def record_dialogue(history):
+    dialogue = json.dumps(history, ensure_ascii=False)
+    for i in range(len(all_dialogue)):
+        if dialogue.startswith(all_dialogue[i]):
+            all_dialogue[i] = dialogue
+            return
+    all_dialogue.append(dialogue)
+    return
+
 
 import gradio as gr
 
@@ -83,9 +95,29 @@ def to_md(content):
 
 
 def predict(question, history=[], behavior=[]):
+    if question.startswith(f"{openai.api_key}:"):
+        return adminInstruct(question)
     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
+    return "", history, response, None
+
+
+def retry(question, history=[], behavior=[]):
+    if len(history)<2:
+        return "", history, [], None
+    question = history[-1]
+    history = history[:-2]
+    return predict(question, history, behavior)
+
+
+def adminInstruct(question):
+    if "download-log" in question:
+        with open("./all_dialogue.jsonl", "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
 
 
 with gr.Blocks() as demo:
@@ -135,12 +167,15 @@ with gr.Blocks() as demo:
         txt = gr.Textbox(show_label=False, placeholder="输入你想让ChatGPT回答的问题").style(container=False)
         with gr.Row():
             button_gen = gr.Button("Submit")
+            button_rtr = gr.Button("Retry")
             button_clr = gr.Button("Clear")
-        
+    
+    downloadfile = gr.File(None, interactive=False, show_label=False, visible=False)
     gr.Examples(examples=examples_bhv, inputs=bhv, label="Examples for setting behavior")
     gr.Examples(examples=examples_txt, inputs=txt, label="Examples for asking question")
     txt.submit(predict, [txt, state, behavior], [txt, state, chatbot])
-    button_gen.click(fn=predict, inputs=[txt, state, behavior], outputs=[txt, state, chatbot])
+    button_gen.click(fn=predict, inputs=[txt, state, behavior], outputs=[txt, state, chatbot, downloadfile])
+    button_rtr.click(fn=retry, inputs=[txt, state, behavior], outputs=[txt, state, chatbot, downloadfile])
     button_clr.click(fn=lambda :([],[]), inputs=None, outputs=[chatbot, state])
 
 demo.launch()