menu update

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 для генерации данных

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()