Update app.py

app.py

@@ -1,36 +1,41 @@
 import gradio as gr
 from transformers import pipeline
 
-# Загружаем модель
+# Загружаем пайплайн генерации (быстрая и лёгкая модель)
 generator = pipeline("text2text-generation", model="google/flan-t5-small")
 
 # История сообщений
 chat_history = []
 
-# Функция чата
 def chat(message):
     global chat_history
 
     if not message.strip():
-        return chat_history + [["Ты", message], ["FlareGPT", "Пожалуйста, напиши что-то!"]]
+        return chat_history + [["Ты", message], ["FlareGPT", "Пожалуйста, введите сообщение."]]
 
     # Генерация ответа
     response = generator(message, max_new_tokens=100)[0]["generated_text"]
 
-    # Обновляем историю
+    # Добавляем в историю
     chat_history.append(["Ты", message])
     chat_history.append(["FlareGPT", response])
 
     return chat_history
 
-# Интерфейс
-iface = gr.Interface(
-    fn=chat,
-    inputs=gr.Textbox(label="Сообщение"),
-    outputs=gr.Chatbot(label="FlareGPT Light"),
-    title="FlareGPT Light",
-    description="Поговори с FlareGPT!",
-    theme="default"
-)
-
-iface.launch()
+def clear_history():
+    global chat_history
+    chat_history = []
+    return []
+
+# Интерфейс Gradio
+with gr.Blocks(title="FlareGPT Light") as demo:
+    gr.Markdown("## Поговори с FlareGPT!")
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(label="Сообщение")
+    send_btn = gr.Button("Отправить")
+    clear_btn = gr.Button("Очистить")
+
+    send_btn.click(chat, inputs=msg, outputs=chatbot)
+    clear_btn.click(clear_history, outputs=chatbot)
+
+demo.launch()