add history download to json

app.py

@@ -1,6 +1,7 @@
 from openai import OpenAI
 import gradio as gr
 import os
+import json
 
 api_key = os.getenv("TYPHOON_API_KEY")  # Replace with your key
 
@@ -11,7 +12,7 @@ client = OpenAI(
 
 def predict(message, history, system_prompt):
     history_openai_format = [{"role": "system", "content": system_prompt}]
-    for human, assistant in history:
+    for human, assistant in history[-3:]:
         if isinstance(human, str) and human.strip():
             history_openai_format.append({"role": "user", "content": human})
         if isinstance(assistant, str) and assistant.strip():
@@ -40,10 +41,21 @@ def chat_bot(user_input, history, system_prompt):
         history[-1] = (user_input, bot_response)
         yield "", history
 
+def get_log(history, system_prompt):
+    history_openai_format = [{"role": "system", "content": system_prompt}]
+    for human, assistant in history:
+        if isinstance(human, str) and human.strip():
+            history_openai_format.append({"role": "user", "content": human})
+        if isinstance(assistant, str) and assistant.strip():
+            history_openai_format.append({"role": "assistant", "content": assistant})
+    
+    history_openai_format_json = '[\n' + ",\n".join([json.dumps(h, ensure_ascii=False) for h in history_openai_format]) + '\n]'
+    return history_openai_format_json
+
 CSS ="""
 .contain { display: flex; flex-direction: column; }
 .gradio-container { height: 100vh !important; }
-#component-0 { height: 100%; }
+#component-0 { height: 80%; }
 #chatbot { flex-grow: 1; overflow: auto;}
 """
 
@@ -67,17 +79,22 @@ with gr.Blocks(css=CSS) as demo:
             inputs=msg,
         )
 
-    with gr.Tab("Setting"):
+    with gr.Tab("Setting") as setting_tab:
         system_prompt = gr.Code(
             value="You are HoraCare, an empathetic Thai girl assistant skilled in psychotherapy and Tarot reading. \nYou provide insights and support using Tarot cards, offering clarity and healing. You always answer in Thai.",
             show_label=True,
             label="System Prompt",
             lines=2
         )
+        
+        msg_log = gr.Code(language='json', label='msg_log')
+        
+        # download_log = gr.Code()
 
     clear.click(lambda: [], [], chatbot)
     msg.submit(chat_bot, [msg, chatbot, system_prompt], [msg, chatbot])
     send.click(chat_bot, [msg, chatbot, system_prompt], [msg, chatbot])
+    setting_tab.select(get_log, [chatbot, system_prompt], [msg_log])
 
     
 demo.launch()