Update app.py

app.py

@@ -3,22 +3,22 @@ from transformers import pipeline
 import torch
 import logging
 
-# Настройка логирования для диагностики
+# Настройка логирования
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Загружаем модель через pipeline (локально из Hugging Face Hub)
+# Загружаем модель
 model_name = "distilgpt2"
 try:
     logger.info(f"Попытка загрузки модели {model_name}...")
     generator = pipeline(
         "text-generation",
         model=model_name,
-        device=-1,  # CPU для бесплатного Spaces
+        device=-1,
         framework="pt",
         max_length=512,
         truncation=True,
-        model_kwargs={"torch_dtype": torch.float32}  # Указываем тип данных для совместимости
+        model_kwargs={"torch_dtype": torch.float32}
     )
     logger.info("Модель успешно загружена.")
 except Exception as e:
@@ -27,13 +27,11 @@ except Exception as e:
 
 def respond(message, history, max_tokens=256, temperature=0.7, top_p=0.9):
     history = history or []
-    # Формируем входной текст
     input_text = ""
     for user_msg, bot_msg in history:
         input_text += f"User: {user_msg}\nAssistant: {bot_msg}\n"
     input_text += f"User: {message}"
 
-    # Генерация ответа
     try:
         logger.info(f"Генерация ответа для: {message}")
         outputs = generator(
@@ -51,7 +49,6 @@ def respond(message, history, max_tokens=256, temperature=0.7, top_p=0.9):
         logger.error(f"Ошибка генерации ответа: {e}")
         return f"Ошибка генерации: {e}", history
 
-    # Форматируем ответ
     formatted_response = format_response(response)
     history.append((message, formatted_response))
     return formatted_response, history
@@ -73,10 +70,9 @@ def extract_treatment(response):
     sentences = response.split(".")
     return sentences[-1].strip() if len(sentences) > 1 else "Не указано"
 
-# Gradio интерфейс
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("## Медицинский чат-бот на базе DistilGPT-2")
-    chatbot = gr.Chatbot(label="Чат", height=400)
+    chatbot = gr.Chatbot(label="Чат", height=400, type='messages')
     with gr.Row():
         msg = gr.Textbox(
             label="Ваше сообщение",
@@ -101,21 +97,18 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     def clear_chat():
         return [], [], ""
 
-    # Кнопка "Отправить"
     submit_btn.click(
         fn=submit_message,
         inputs=[msg, state, max_tokens, temperature, top_p],
         outputs=[chatbot, state, msg],
         queue=True
     )
-    # Поддержка Enter
     msg.submit(
         fn=submit_message,
         inputs=[msg, state, max_tokens, temperature, top_p],
         outputs=[chatbot, state, msg],
         queue=True
     )
-    # Кнопка "Очистить"
     clear_btn.click(
         fn=clear_chat,
         outputs=[chatbot, state, msg]