hotfix

app.py

@@ -6,8 +6,8 @@ from typing import List, Dict, Iterator
 from threading import Thread
 from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig, TextIteratorStreamer
 from peft import PeftModel
+import json  
 
-# --- 1) Конфигурация и загрузка модели ---
 BASE_MODEL_ID   = "Tweeties/tweety-7b-tatar-v24a"
 ADAPTER_ID      = os.getenv("ADAPTER_ID")
 YANDEX_API_KEY  = os.getenv("YANDEX_API_KEY")
@@ -115,13 +115,18 @@ def chat_fn(message: str, ui_history: list, messages_state: List[Dict[str, str]]
     messages = messages_state + [{"role": "user", "content": user_tt}]
     ui_history = ui_history + [[user_tt, ""]]
 
+    last = ""  
     for partial in generate_tt_reply_stream(messages):
+        last = partial                 # ADD
         ui_history[-1][1] = partial
         yield ui_history, messages_state + [
             {"role": "user", "content": user_tt},
             {"role": "assistant", "content": partial},
         ]
 
+    final_state = messages + [{"role": "assistant", "content": last}]
+    print("STATE:", json.dumps(final_state, ensure_ascii=False))
+
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("## Татарский чат-бот от команды Сбера")
@@ -147,7 +152,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         return [], [{"role": "system", "content": SYS_PROMPT_TT}]
 
     clear.click(_reset, inputs=None, outputs=[chatbot, messages_state], queue=False)
-    print("Messages_state: " + messages_state)
     clear.click(lambda: "", None, msg, queue=False)
 
 if __name__ == "__main__":