Update app.py

app.py

@@ -36,9 +36,9 @@ def load_model(model_key: str):
 model_cache = {}
 
 def respond(message: str, 
-            history: List[Dict[str, str]], 
+            history: List[List[str]], 
             model_key: str, 
-            system_prompt: str) -> Tuple[List[Dict[str, str]], str, Dict[str, Any]]:
+            system_prompt: str) -> Tuple[List[List[str]], str, Dict[str, Any]]:
     """Локальный чат с pipeline."""
     try:
         if model_key not in model_cache:
@@ -47,11 +47,16 @@ def respond(message: str,
         
         print(f"🚀 Генерация: {model_key}, Msg='{message[:30]}...'")
         
-        # Chat format (system + history + user)
+        # Преобразуем формат истории из Gradio в формат сообщений
         messages = []
         if system_prompt.strip():
             messages.append({"role": "system", "content": system_prompt})
-        messages.extend(history)
+        
+        # Конвертируем историю из формата [["user_msg", "bot_msg"], ...]
+        for user_msg, bot_msg in history:
+            messages.append({"role": "user", "content": user_msg})
+            messages.append({"role": "assistant", "content": bot_msg})
+        
         messages.append({"role": "user", "content": message})
         
         # Apply chat template (для instruct)
@@ -64,38 +69,34 @@ def respond(message: str,
         
         print(f"✅ Ответ: {bot_reply[:50]}...")
         
-        new_history = history + [
-            {"role": "user", "content": message},
-            {"role": "assistant", "content": bot_reply}
-        ]
+        # Обновляем историю в формате Gradio
+        new_history = history + [[message, bot_reply]]
         return new_history, "", gr.update(value="")
     
     except Exception as e:
         error_msg = f"❌ {model_key}: {str(e)}"
         print(f"💥 {error_msg}")
-        new_history = history + [
-            {"role": "user", "content": message},
-            {"role": "assistant", "content": error_msg}
-        ]
+        new_history = history + [[message, error_msg]]
         return new_history, error_msg, gr.update(value="")
 
-# UI - Убрал параметр theme из gr.Blocks()
+# UI
 with gr.Blocks(title="🚀 Локальный HF Чат (на слабом CPU!)") as demo:
     gr.Markdown("# Локальный Inference (без API!)\n**Маленькие модели** — 1-3 сек CPU. Большие думают ооочень долго. Нет limits/token. В качестве примера.")
     
-    with gr.Row(variant="compact"):
+    with gr.Row():
         model_dropdown = gr.Dropdown(choices=list(MODELS.keys()), value="Qwen2.5-0.5B", label="🧠 Модель")
         system_prompt = gr.Textbox(label="📝 System", placeholder="Ты весёлый ИИ.", lines=2)
     
-    chatbot = gr.Chatbot(type="messages", height=500)
+    # Убрал параметр type="messages" для совместимости со старой версией
+    chatbot = gr.Chatbot(height=500)
     
     with gr.Row():
-        msg_input = gr.Textbox(placeholder="Привет! (Enter)", scale=7)
-        send_btn = gr.Button("📤", variant="primary", scale=1)
+        msg_input = gr.Textbox(placeholder="Привет! (Enter)", show_label=False, lines=1)
+        send_btn = gr.Button("📤 Отправить", variant="primary")
     
     with gr.Row():
-        clear_btn = gr.Button("🗑️ Clear")
-        retry_btn = gr.Button("🔄 Retry")
+        clear_btn = gr.Button("🗑️ Очистить")
+        retry_btn = gr.Button("🔄 Повторить")
     
     status = gr.Textbox(label="Логи", interactive=False, lines=4)
     
@@ -108,8 +109,8 @@ with gr.Blocks(title="🚀 Локальный HF Чат (на слабом CPU!)
     clear_btn.click(clear, outputs=[chatbot, status, msg_input])
     
     def retry(history):
-        if len(history) >= 2 and history[-2]["role"] == "user":
-            return history[-2]["content"]
+        if len(history) >= 1:
+            return history[-1][0]  # Последнее сообщение пользователя
         return ""
     retry_btn.click(retry, inputs=[chatbot], outputs=[msg_input])