Update app.py

app.py

@@ -42,7 +42,11 @@ 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)
@@ -51,7 +55,25 @@ def respond(
         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:]
+
+    # Уточняющие вопросы
+    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