Программирование на питон

Question

def examination(name, age, symptoms): 
    print(&#34;Пациент пришел на осмотр.&#34;) 
    print('Имя: ' + name + '. Возраст: ' + str(age)) 
    # Затем необходимо провести диагностику и выбрать подходящее лечение на основе симптомов 
    if symptoms == &#34;головная боль&#34;: 
        print('диагноз - мигрень') 
        print('лечение - принять анальгин') 
    elif symptoms == &#34;кашель&#34;: 
        print('диагноз - простуда') 
        print('лечение - пить чай с медом') 
    elif symptoms == &#34;животная боль&#34;: 
        print('диагноз - гастрит') 
        print('лечение - лечь в больницу') 
    else: 
        print('диагноз - неизвестен') 
        print('требуется дальнейшее изучение') 
def treatment(): 
    print(&#34;Проводится лечение.&#34;) 
    print(&#34;Лечение прошло успешно.&#34;) 
def main(): 
    print(&#34;Добро пожаловать в симулятор врача!&#34;) 
    while True: 
        choice = input(&#34;Выберите действие (1 - осмотр, 2 - лечение, q - выход): &#34;) 
        if choice == '1': 
            name = input(&#34;Введите ваше имя: &#34;) 
            age = int(input(&#34;Введите ваш возраст: &#34;)) 
            symptoms = input(&#34;Опишите ваши симптомы: &#34;) 
            examination(name, age, symptoms) 
        elif choice == '2': 
            treatment() 
        elif choice == 'q': 
            break 
        else: 
            print(&#34;Некорректный выбор. Пожалуйста, повторите.&#34;) 
main()

у меня в выборе 2 пишет проблема устранена, а я хочу чтобы там писалось как ее вылечить
как это можно исправить?

S.H.I. · Answer

diagnosis = "" 
 
def examination(name, age, symptoms):  
    global diagnosis  # объявляем переменную как глобальную, чтобы можно было её изменять 
     
    print("Пациент пришел на осмотр.")  
    print('Имя: ' + name + '. Возраст: ' + str(age))  
     
    if symptoms == "головная боль":  
        diagnosis = 'мигрень' 
        print('диагноз - ' + diagnosis)  
        print('лечение - принять анальгин')  
    elif symptoms == "кашель":  
        diagnosis = 'простуда' 
        print('диагноз - ' + diagnosis)  
        print('лечение - пить чай с медом')  
    elif symptoms == "животная боль":  
        diagnosis = 'гастрит' 
        print('диагноз - ' + diagnosis)  
        print('лечение - лечь в больницу')  
    else:  
        diagnosis = 'неизвестен' 
        print('диагноз - ' + diagnosis)  
        print('требуется дальнейшее изучение')  
 
def treatment():  
    if diagnosis == "мигрень": 
        print('Рекомендуется принять анальгин.')  
    elif diagnosis == "простуда": 
        print('Рекомендуется пить чай с медом.')  
    elif diagnosis == "гастрит": 
        print('Рекомендуется лечь в больницу.')  
    elif diagnosis == "неизвестен": 
        print('Требуется дальнейшее изучение.')  
    else: 
        print("Диагноз не установлен. Необходим осмотр.")  
 
    print("Лечение прошло успешно.")  
 
def main():  
    global diagnosis  # глобальная переменная для хранения диагноза 
 
    print("Добро пожаловать в симулятор врача!")  
    while True:  
        choice = input("Выберите действие (1 - осмотр, 2 - лечение, q - выход): ")  
        if choice == '1':  
            name = input("Введите ваше имя: ")  
            age = int(input("Введите ваш возраст: "))  
            symptoms = input("Опишите ваши симптомы: ")  
            examination(name, age, symptoms)  
        elif choice == '2':  
            treatment()  
        elif choice == 'q':  
            break  
        else:  
            print("Некорректный выбор. Пожалуйста, повторите.")  
 
main()