Update app.py

app.py

@@ -42,38 +42,20 @@ def respond(
         print(f"Error occurred during message generation: {e}")
         yield "Произошла ошибка при обработке запроса."
 
-    # Формирование ответа в нужном формате
-    # Пример ответа: "Предварительный диагноз: Грипп\nОперация: Не требуется\nУточняющие вопросы: Как давно начались симптомы?"
+    # Формирование ответа в нужном формате на русском языке
     response = response.strip()
 
-    # Обработка предварительного диагноза
-    diagnosis_start = response.find("Предварительный диагноз:")
-    if diagnosis_start != -1:
-        diagnosis_end = response.find("\n", diagnosis_start)
-        if diagnosis_end == -1:
-            diagnosis_end = len(response)
-        diagnosis = response[diagnosis_start:diagnosis_end]
-        response = response[:diagnosis_start] + f"<b>{diagnosis}</b>" + response[diagnosis_end:]
+    # Формирование ответа в структуре: диагноз, операция, уточняющие вопросы
+    # Здесь мы предполагаем, что модель вернет необходимые данные, и мы их форматируем.
 
-    # Операция
-    operation_start = response.find("Операция:")
-    if operation_start != -1:
-        operation_end = response.find("\n", operation_start)
-        if operation_end == -1:
-            operation_end = len(response)
-        operation = response[operation_start:operation_end]
-        response = response[:operation_start] + f"<b>{operation}</b>" + response[operation_end:]
+    # Пример формирования ответа с диагнозом, операцией и уточняющими вопросами
+    response = (
+        f"Предварительный диагноз: {response}\n"
+        "Операция: Не требуется\n"  # Это можно заменить динамически, если модель будет возвращать такую информацию
+        "Уточняющие вопросы для постановки диагноза: Есть ли высокая температура? Когда появились симптомы?\n"
+    )
 
-    # Уточняющие вопросы
-    questions_start = response.find("Уточняющие вопросы:")
-    if questions_start != -1:
-        questions_end = response.find("\n", questions_start)
-        if questions_end == -1:
-            questions_end = len(response)
-        questions = response[questions_start:questions_end]
-        response = response[:questions_start] + f"<b>{questions}</b>" + response[questions_end:]
-
-    # Формирование финального ответа с добавлением идентификационного сообщения
+    # Добавление финального сообщения
     final_response = f"{response}\n\nСоздано больницей EMS штата Alta"
     yield final_response