Update app.py

app.py

@@ -43,17 +43,22 @@ def generate_response(prompt):
         max_new_tokens=40
     )
 
-    # Декодируем и очищаем ответ
+    # Декодируем и очищаем ответ от тегов
     response = tokenizer.decode(generated_token_ids[0], skip_special_tokens=True)
     cleaned_response = re.sub(r'@@ПЕРВЫЙ@@|@@ВТОРОЙ@@', '', response).strip()
 
-    # Добавляем новую реплику в историю
-    chat_history.append(prompt)
-    chat_history.append(cleaned_response)
+    # Добавляем текущий запрос и ответ в историю для отображения в чате
+    chat_history.append(("Ты", prompt))  # Реплика пользователя
+    chat_history.append(("Бот", cleaned_response))  # Ответ бота
 
-    return cleaned_response
+    return chat_history
 
-# Интерфейс Gradio
-iface = gr.Interface(fn=generate_response, inputs="text", outputs="text", title="ruDialoGPT Chatbot")
+# Интерфейс Gradio с компонентом chatbot
+iface = gr.Interface(
+    fn=generate_response, 
+    inputs="text", 
+    outputs="chatbot",  # Вывод в виде чата
+    title="ruDialoGPT Chatbot с Историей"
+)
 
 iface.launch()