import os
import M1

def main():
    fname = input("Файл (data.txt): ")
    if not os.path.isfile(fname):
        print("Файл не найден")
        return
    
    # Чтение данных
    data = {}
    with open(fname, 'r', encoding='utf-8') as f:
        for line in f:
            c, p, r = line.strip().split()
            data[c] = [float(p), float(r)]
    
    print("Данные:", data)
    
    # Расчёт
    c, p, o = M1.calc(data)
    
    print("\nРесурс на душу:")
    for country, val in zip(c, p):
        print(f"{country}: {val:.3f}")
    print(f"\nОбщее: {o:.3f}")
    
    # График
    M1.plot(data)
    
    return c, p, o