Пишу Бывалым программистам на Python . Отцените код, подскажите что изменить и как улучшить код. Код писал на телефоне

Вот улучшенный

import random
import math

def simulate_battle(country):
"""Симулирует битву между Северогерманским союзом и Францией."""

if country == "1":
print("Отличный выбор\nПрусская пехота славится своей выучкой и отвагой.\n+10% бонус организации\n\nВаши войска встретились с армией Франции у фр.города 'S'")
pruss_bonus = get_army_strength("Северогерманский союз", 8000, 9000, 0.1)
fra_bonus = get_army_strength("Франция", 9000, 10500, -0.1) # Отрицательный бонус
loss_threshold = 4000
elif country == "2":
print("Отличный выбор\nОтсутствие хорошо проработанного плана, нежелание народа служить в армии.\n-10% бонус организации\n\nВаши войска встретились с армией Пруссии у прусс.города 'B'")
fra_bonus = get_army_strength("Франция", 9000, 10500, -0.1) # Отрицательный бонус
pruss_bonus = get_army_strength("Северогерманский союз", 8000, 9000, 0.1)
loss_threshold = 4000
else:
print("Некорректный выбор страны. Игра окончена.")
return

print(f"Ваша армия: {int(pruss_bonus if country == '1' else fra_bonus)} ?\nВражеская армия: {int(fra_bonus if country == '1' else pruss_bonus)} ?")

battle = input("Начать бой? ⚔️ ").lower()
if battle == "да":
if pruss_bonus > fra_bonus if country == '1' else fra_bonus > pruss_bonus:
print("Наша армия победила!")
else:
print(f"Мы проиграли, {'французы взяли Берлин' if country == '1' else 'немцы взяли Париж'}\nИгра окончена...")
else:
print("Наша армия отступила")
losses = calculate_retreat_losses(pruss_bonus if country == '1' else fra_bonus, 0.3 if country == '1' else 0.4) #Разные потери в зависимости от страны
print(f"Наши потери: {int(losses)} ?")
if losses > loss_threshold:
print(f"Мы проиграли, {'французы взяли Берлин' if country == '1' else 'немцы взяли Париж'}\nИгра окончена...")

def get_army_strength(country_name, min_army_size, max_army_size, bonus_percentage):
"""Генерирует численность армии с учетом бонуса."""
army_size = random.randint(min_army_size, max_army_size)
bonus = army_size * bonus_percentage
return army_size + bonus

def calculate_retreat_losses(army_strength, loss_percentage):
"""Вычисляет потери при отступлении."""
return army_strength - (army_strength * loss_percentage)

game = input("Это симулятор сражений, хотите начать? ").lower()
if game == 'да':
print("Выберите сторону:\n1. Северогерманский союз ??\n2. Франция ??\nЧтобы выбрать страну напишите её номер!")
country = input("").lower()
simulate_battle(country)
else:
print("Выход из игры...")