Update app.py

app.py

@@ -29,10 +29,10 @@ scheduler = CommitScheduler(
     token=HF_UPLOAD
 )
 
-def save_json(role: str, content: str) -> None:
+def save_json(question: str, answer: str) -> None:
     with scheduler.lock:
         with JSON_DATASET_PATH.open("a") as f:
-            json.dump({"role": role, "content": content, "datetime": datetime.now().isoformat()}, f, ensure_ascii=False)
+            json.dump({"question": question, "answer": answer, "datetime": datetime.now().isoformat()}, f, ensure_ascii=False)
             f.write("\n")
 
 
@@ -101,8 +101,8 @@ def generate(
         outputs.append(text)
         yield "".join(outputs)
         
-    save_json("user", message)
-    save_json("assistant", "".join(outputs))
+   
+    save_json(message, "".join(outputs))