# M2.py
import os
from M1 import func1, func2

# Запрос имени файла с проверкой существования
filename = input("Введите имя файла с исходными данными: ")
while not os.path.isfile(filename):
    print("Файл не найден.")
    filename = input("Введите имя файла с исходными данными: ")

# Запрос порога
KK = float(input("Введите пороговое значение KK: "))

# Вызов func1
data = func1(filename, KK)

# Имена новых файлов
base, ext = os.path.splitext(filename)
file1 = base + '1' + ext
file2 = base + '2' + ext

# Три вызова func2
files = [filename, file1, file2]
labels = ["Исходный", "Больше KK", "Не больше KK"]

print("\nРезультаты статистики:")
print("-" * 50)
for i in range(3):
    mean, median, minv, maxv, std = func2(files[i])
    print(f"\n{labels[i]} файл:")
    print(f"  Среднее:        {mean:.4f}")
    print(f"  Медиана:        {median:.4f}")
    print(f"  Минимум:        {minv:.4f}")
    print(f"  Максимум:       {maxv:.4f}")
    print(f"  Ст. откл.:      {std:.4f}")

# Возвращаем исходные данные для M0
original_data = data