código feito por mim para resolver contas de física quando eu estava no 1o bimestre

maquina termica fisica.py

from time import sleep

print('Responda abaixo utilizando 1, 2 ou 3 o para demonstrar qual será. 1 para q1 e q2 inclusos. 2 para apenas q1 e T \n'
      'e 3 para quando pedir potência.')
print()

formula = int(input('Digite o número: '))

if formula == 1:
    print('Ok, agora informe o que se pede no trabalho: ')
    q1 = int(input('Digite a fonte quente: '))
    q2 = int(input('Digite a fonte fria: '))
    t = q1 - q2

    rendimento = t / q1
    rendimentop = rendimento * 100

    print()
    print(f'A quantidade de fonte quente é {q1} a de fonte fria {q2} e a quantidade de trabalho gerado é de {t}')
    print(f'O rendimento foi de {rendimentop}%')

elif formula == 2:
    print('Ok, agora informe o que se pede no trabalho: ')
    q1 = int(input('Digite a fonte quente: '))
    t = int(input('Digite a quantidade de trabalho: '))
    q2 = q1 - t

    rendimento = t / q1
    rendimentop = rendimento * 100

    print()
    print(f'A quantidade de fonte quente é {q1} a de fonte fria {q2} e a quantidade de trabalho gerado é de {t}')
    print(f'O rendimento foi de {rendimentop}%')

elif formula == 3:
    print('Ok, agora informe o que se pede no trabalho: ')
    q1 = int(input('Digite a fonte quente: '))
    q2 = int(input('Digite a fonte fria: '))
    ciclo = int(input('O ciclo dura quantos segundos: '))
    t = q1 - q2

    rendimento = t / q1
    rendimentop = rendimento * 100

    potencia = t / ciclo

    print()
    print(f'A quantidade de fonte quente é {q1} a de fonte fria {q2} e a quantidade de trabalho gerado é de {t}')
    print(f'O rendimento foi de {rendimentop}%')
    print(f'A potência foi de {potencia}W')

else:
    print('Digite algo válido.')

sleep(120)