import pylab
from M2 import run_M2

def main():
    X, X1 = run_M2()
    
    # Построение графиков
    pylab.figure(figsize=(10, 5))
    pylab.subplot(1, 2, 1)
    pylab.plot(X, label='Исходный сигнал')
    pylab.title("Исходный сигнал")
    pylab.xlabel("Отсчёты")
    pylab.ylabel("Значение")
    pylab.grid(True)

    pylab.subplot(1, 2, 2)
    pylab.plot(X1, label='Исправленный сигнал', color='orange')
    pylab.title("Исправленный сигнал")
    pylab.xlabel("Отсчёты")
    pylab.ylabel("Значение")
    pylab.grid(True)

    pylab.tight_layout()
    pylab.show()

if __name__ == "__main__":
    main()