import matplotlib.pyplot as plt
from M2 import main

X, X1 = main()

if X1:
    plt.figure(figsize=(10, 5))
    plt.plot(X, 'b-', label='Исходный', alpha=0.7)
    plt.plot(X1, 'r-', label='Исправленный', linewidth=2)
    plt.legend()
    plt.grid(True)
    plt.show()
else:
    plt.plot(X, 'b-')
    plt.title("Нет аномалий для исправления")
    plt.grid(True)
    plt.show()