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,27 +243,19 @@ DEFAULT_THEME = {
     }
 }
 
-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}")
+def format_history(history):
+    messages = [{
+        "role": "system",
+        "content": "You are a helpful assistant named Flare.",
+    }]
+    for item in history:
+        if item["role"] == "user":
+            messages.append({"role": "user", "content": item["content"]})
+        elif item["role"] == "assistant":
+            messages.append({"role": "assistant", "content": item["content"]})
+    return messages
+
 
 class Gradio_Events:
 
@@ -516,8 +508,7 @@ 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):
@@ -574,7 +565,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"]))
@@ -634,6 +625,7 @@ css = """
 }
 """
 
+
 def logo():
     with antd.Typography.Title(level=1,
                                elem_style=dict(fontSize=24,
@@ -647,15 +639,8 @@ 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": {}
-        })
 
+with gr.Blocks(css=css, fill_width=True) as demo:
     state = gr.State({
         "conversations_history": {},
         "conversations": [],
@@ -989,9 +974,13 @@ with gr.Blocks(css=css, fill_width=True) as demo:
                                                                maxRows=6),
                                                 elem_style=dict(width="100%"))
     # Events Handler
-    demo.load(fn=Gradio_Events.apply_browser_state,
-              inputs=[browser_state, state],
-              outputs=[conversations, state])
+    if save_history:
+        browser_state = gr.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],
@@ -1046,8 +1035,5 @@ with gr.Blocks(css=css, fill_width=True) as demo:
                       conversations, add_conversation_btn, chatbot, state
                   ])
 
-    # 添加状态更新时的保存逻辑 (исправленный вариант)
-    state.change(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)