Update app.py

app.py

@@ -1,8 +1,8 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
-from typing import List, Dict, Any
+from typing import List, Dict, Any, Tuple  # ← + Tuple!
 
-# 10 моделей (instruct)
+# 10 моделей
 MODELS = [
     "microsoft/Phi-3-mini-4k-instruct",
     "Qwen/Qwen2.5-0.5B-Instruct",
@@ -17,22 +17,20 @@ MODELS = [
 ]
 
 def respond(message: str, 
-            history: List[Dict[str, str]],  # Native messages от Chatbot
+            history: List[Dict[str, str]], 
             model_id: str, 
             system_prompt: str,
             hf_token: str) -> Tuple[List[Dict[str, str]], str, Dict[str, Any]]:
-    """HF API. History — List[dict] → append user/assistant."""
+    """HF API с native messages."""
     try:
         client = InferenceClient(model=model_id, token=hf_token.strip() or None)
         
-        # messages = system + history + new user
         messages = []
         if system_prompt.strip():
             messages.append({"role": "system", "content": system_prompt})
         messages.extend(history)
         messages.append({"role": "user", "content": message})
         
-        # API call
         response = client.chat_completion(
             messages=messages,
             max_tokens=512,
@@ -42,17 +40,16 @@ def respond(message: str,
         
         bot_reply = response.choices[0].message.content
         
-        # New history: + user + assistant
         new_history = history + [
             {"role": "user", "content": message},
             {"role": "assistant", "content": bot_reply}
         ]
-        return new_history, "", gr.update(value="")  # Clear input
+        return new_history, "", gr.update(value="")
     
     except Exception as e:
         error_msg = f"❌ {str(e)}"
         if "429" in str(e) or "rate" in str(e).lower():
-            error_msg += "\n🔥 Rate limit → Вставь HF Token!"
+            error_msg += "\n🔥 Rate limit → HF Token!"
         elif "token" in str(e).lower():
             error_msg += "\n🔑 Проверь токен."
         
@@ -62,20 +59,17 @@ def respond(message: str,
         ]
         return new_history, error_msg, gr.update(value="")
 
-# UI (без State, native messages)
-with gr.Blocks(title="🚀 HF Чат-Тестер (финал!)", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# Тест HF Inference API\n**HF Token** → твои лимиты 1M tokens/мес. [Токен](https://huggingface.co/settings/tokens)")
+# UI
+with gr.Blocks(title="🚀 HF Чат-Тестер (готово!)", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# Тест HF Inference API\n**Token** → 1M tokens/мес с тебя. [Токен](https://huggingface.co/settings/tokens)")
     
-    # Контролы ВЕРХОМ
     with gr.Row(variant="compact"):
         model_dropdown = gr.Dropdown(choices=MODELS, value=MODELS[0], label="🧠 Модель")
-        system_prompt = gr.Textbox(label="📝 System", placeholder="Ты хакер ИИ.", lines=2)
+        system_prompt = gr.Textbox(label="📝 System", placeholder="Ты ИИ-хакер.", lines=2)
         hf_token = gr.Textbox(label="🔑 Token", placeholder="hf_...", type="password")
     
-    # Чат (messages — 0 warning!)
     chatbot = gr.Chatbot(type="messages", height=500)
     
-    # Input + кнопки
     with gr.Row():
         msg_input = gr.Textbox(placeholder="Сообщение... (Enter)", scale=7, container=True)
         send_btn = gr.Button("📤", variant="primary", scale=1)
@@ -84,9 +78,9 @@ with gr.Blocks(title="🚀 HF Чат-Тестер (финал!)", theme=gr.theme
         clear_btn = gr.Button("🗑️ Очистить")
         retry_btn = gr.Button("🔄 Повторить")
     
-    status = gr.Textbox(label="Статус", interactive=False)  # Для ошибок
+    status = gr.Textbox(label="Статус", interactive=False)
     
-    # Send (клик + Enter) — outputs: chatbot, status, msg_input
+    # Events
     send_btn.click(
         fn=respond,
         inputs=[msg_input, chatbot, model_dropdown, system_prompt, hf_token],
@@ -99,13 +93,14 @@ with gr.Blocks(title="🚀 HF Чат-Тестер (финал!)", theme=gr.theme
         outputs=[chatbot, status, msg_input]
     )
     
-    # Кнопки
     def clear():
         return [], "", gr.update(value="")
     clear_btn.click(clear, outputs=[chatbot, status, msg_input])
     
     def retry(history):
-        return history[-2]["content"] if len(history) >= 2 else ""  # Последний user msg
+        if len(history) >= 2 and history[-2]["role"] == "user":
+            return history[-2]["content"]
+        return ""
     retry_btn.click(retry, inputs=[chatbot], outputs=[msg_input])
 
 if __name__ == "__main__":