Update app.py

app.py

@@ -7,14 +7,21 @@ generator = pipeline("text-generation", model="distilgpt2")
 
 # Функция для ответа "чат-бота"
 def chat(message, history):
-    # берём последние сообщения из истории
+    # Собираем промпт из истории
     prompt = ""
-    for user, bot in history:
-        prompt += f"User: {user}\nBot: {bot}\n"
+    for msg in history:
+        role = msg["role"]
+        content = msg["content"]
+        prompt += f"{role}: {content}\n"
     prompt += f"User: {message}\nBot:"
 
-    # генерируем ответ
-    result = generator(prompt, max_length=200, num_return_sequences=1, pad_token_id=50256)
+    # Генерация ответа
+    result = generator(
+        prompt,
+        max_length=200,
+        num_return_sequences=1,
+        pad_token_id=50256,
+    )
     reply = result[0]["generated_text"].split("Bot:")[-1].strip()
 
     return reply
@@ -22,16 +29,17 @@ def chat(message, history):
 # Интерфейс Gradio
 with gr.Blocks() as demo:
     gr.Markdown("## 🤖 Простой чат-бот на Hugging Face (distilgpt2)")
-    chatbot = gr.Chatbot()
+    chatbot = gr.Chatbot(type="messages")
     msg = gr.Textbox(placeholder="Напиши сообщение и нажми Enter...")
 
     def respond(message, chat_history):
         bot_message = chat(message, chat_history)
-        chat_history.append((message, bot_message))
+        chat_history.append({"role": "user", "content": message})
+        chat_history.append({"role": "assistant", "content": bot_message})
         return "", chat_history
 
     msg.submit(respond, [msg, chatbot], [msg, chatbot])
 
-# Запускаем
+# Запуск
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()