Update app.py

app.py

@@ -11,7 +11,7 @@ from typing import Generator, Dict, Any
 import logging
 import time
 
- # =========== Configuration
+# ========== Configuration
 # MODEL NAME
 model = os.getenv("MODEL_NAME")
 # 代理服务器配置
@@ -21,7 +21,7 @@ MAX_RETRIES = int(os.getenv("MAX_RETRIES", 5))
 # 保存历史
 save_history = True
 
-# =========== Configuration
+# ========== Configuration
 
 # 配置日志
 logging.basicConfig(level=logging.INFO)
@@ -155,7 +155,7 @@ def chat_with_retry(history_messages, max_retries=MAX_RETRIES):
                 model=model,
                 messages=history_messages,
                 stream=True,
-                temperature = 0.7, top_p = 0.8
+                temperature=0.7, top_p=0.8
             )
             
             return response
@@ -243,7 +243,6 @@ DEFAULT_THEME = {
     }
 }
 
-
 def format_history(history):
     messages = [{
         "role": "system",
@@ -256,6 +255,27 @@ def format_history(history):
             messages.append({"role": "assistant", "content": item["content"]})
     return messages
 
+def load_history():
+    """加载历史记录"""
+    try:
+        with open('history.json', 'r', encoding='utf-8') as f:
+            return json.load(f)
+    except (FileNotFoundError, json.JSONDecodeError):
+        return {
+            "conversations": [],
+            "conversations_history": {}
+        }
+
+def save_history(state_value):
+    """保存历史记录"""
+    try:
+        with open('history.json', 'w', encoding='utf-8') as f:
+            json.dump({
+                "conversations": state_value["conversations"],
+                "conversations_history": state_value["conversations_history"]
+            }, f, ensure_ascii=False, indent=4)
+    except Exception as e:
+        logger.error(f"Failed to save history: {e}")
 
 class Gradio_Events:
 
@@ -508,7 +528,8 @@ class Gradio_Events:
             return gr.skip()
         state_value["conversation_id"] = ""
         return gr.update(active_key=state_value["conversation_id"]), gr.update(
-            items=DEFAULT_CONVERSATIONS_HISTORY), gr.update(value=state_value)
+            items=DEFAULT_CONVERSATIONS_HISTORY), gr.update(
+                value=state_value)
 
     @staticmethod
     def select_conversation(state_value, e: gr.EventData):
@@ -565,7 +586,7 @@ class Gradio_Events:
 
     @staticmethod
     def update_browser_state(state_value):
-
+        save_history(state_value)
         return gr.update(value=dict(
             conversations=state_value["conversations"],
             conversations_history=state_value["conversations_history"]))
@@ -625,7 +646,6 @@ css = """
 }
 """
 
-
 def logo():
     with antd.Typography.Title(level=1,
                                elem_style=dict(fontSize=24,
@@ -639,8 +659,15 @@ def logo():
                        height=24)
             ms.Span("dots.llm1.inst")
 
-
 with gr.Blocks(css=css, fill_width=True) as demo:
+    if save_history:
+        browser_state = gr.State(load_history())
+    else:
+        browser_state = gr.State({
+            "conversations": [],
+            "conversations_history": {}
+        })
+
     state = gr.State({
         "conversations_history": {},
         "conversations": [],
@@ -974,13 +1001,9 @@ with gr.Blocks(css=css, fill_width=True) as demo:
                                                                maxRows=6),
                                                 elem_style=dict(width="100%"))
     # Events Handler
-    if save_history:
-        browser_state = gr.State({
-        })
-
-        demo.load(fn=Gradio_Events.apply_browser_state,
-                  inputs=[browser_state, state],
-                  outputs=[conversations, state])
+    demo.load(fn=Gradio_Events.apply_browser_state,
+              inputs=[browser_state, state],
+              outputs=[conversations, state])
 
     add_conversation_btn.click(fn=Gradio_Events.new_chat,
                                inputs=[state],
@@ -1035,5 +1058,10 @@ with gr.Blocks(css=css, fill_width=True) as demo:
                       conversations, add_conversation_btn, chatbot, state
                   ])
 
+    # 添加状态更新时的保存逻辑
+    demo.post(fn=Gradio_Events.update_browser_state,
+              inputs=state,
+              outputs=None)
+
 if __name__ == "__main__":
     demo.queue(default_concurrency_limit=200).launch(ssr_mode=False, max_threads=200)