From a69f05e1dacf92f0c6e08212b214738cd86e3b00 Mon Sep 17 00:00:00 2001
From: cheykrym <cheykrym@MACOS-CATALINA.local>
Date: Mon, 14 Oct 2024 23:04:09 +0300
Subject: [PATCH] menu update

---
 app/menu_load.py | 27 ++++++++++++++++++++++++---
 main.py          | 29 +++--------------------------
 2 files changed, 27 insertions(+), 29 deletions(-)

diff --git a/app/menu_load.py b/app/menu_load.py
index 854b324..685d0ac 100644
--- a/app/menu_load.py
+++ b/app/menu_load.py
@@ -2,12 +2,33 @@ from app.utils.system import clear_console
 from app.zadachi.zadacha1 import zadacha1
 from app.zadachi.zadacha2 import zadacha2
 
-class pod_menu:
+class menu:
     
     def __init__(self):
         pass  # Конструктор пока пустой, при необходимости можно добавить параметры
 
-    def zadacha1_menu():
+    def main_menu(self):
+        clear_console()
+        while True:
+            print('\nMain Menu:')
+            print('1. Задача 1')
+            print('2. Задача 2')
+            print('0. Exit')
+
+            choice = input('Select an option: ')
+
+            clear_console()
+            if choice == '1':
+                self.zadacha1_menu()  # Переход к подменю Задачи 1
+            elif choice == '2':
+                self.zadacha2_menu()  # Переход к подменю Задачи 2
+            elif choice == '0':
+                print('Exiting program...')
+                break
+            else:
+                print('Invalid option')
+
+    def zadacha1_menu(self):
         """Меню для задачи 1 с подкатегориями."""
         clear_console()
         while True:
@@ -32,7 +53,7 @@ class pod_menu:
             else:
                 print('Неверный выбор, попробуйте снова.')
 
-    def zadacha2_menu():
+    def zadacha2_menu(self):
         """Меню для задачи 2 (расчет коэффициентов методом наименьших квадратов)."""
         clear_console()
         z1 = zadacha1()  # Используем объект задачи 1 для генерации данных
diff --git a/main.py b/main.py
index f25480c..613f6fc 100644
--- a/main.py
+++ b/main.py
@@ -1,29 +1,6 @@
 
-from app.utils.config import debug
-from app.utils.system import clear_console
-from app.menu_load import pod_menu
-
-
-def main_menu():
-    clear_console()
-    while True:
-        print('\nMain Menu:')
-        print('1. Задача 1')
-        print('2. Задача 2')
-        print('0. Exit')
-
-        choice = input('Select an option: ')
-
-        clear_console()
-        if choice == '1':
-            pod_menu.zadacha1_menu()  # Переход к подменю Задачи 1
-        elif choice == '2':
-            pod_menu.zadacha2_menu()  # Переход к подменю Задачи 2
-        elif choice == '0':
-            print('Exiting program...')
-            break
-        else:
-            print('Invalid option')
+from app.menu_load import menu
 
 if __name__ == "__main__":
-    main_menu()
+    mn = menu()
+    mn.main_menu()