compleate the basice confirm data tui :D

Author: 0x61nas

solve.py

@@ -63,7 +63,10 @@
 discription = main_div.find("div", {"class": "_1l1MA"})
 
 # Show the tui for confirm the data and choose the language to solve the problem
-data = tui.confirm_data(Data(title, level, problem_path))
+data = Data(title, level, problem_path)
+tui.confirm_data(data)
+print(data)
+exit(0)
 
 # Download the images if there are any
 for img in discription.find_all("img"):

utils/pytermgui_tui.py

@@ -61,7 +61,21 @@ def get_the_url(base_url: str) -> str:
         wm.run()
     return problem_url
 
-def confirm_data(data: Data) -> Data:
+def parse_data(manager: ptg.WindowManager, window: ptg.Window, data: Data) -> None:
+    """ Parse the data and choose the language to solve the problem """
+    for widget in window:
+        if isinstance(widget, ptg.InputField):
+            if widget.prompt == "Title: ":
+                data.title = widget.value
+            elif widget.prompt == "Level: ":
+                data.level = widget.value
+            elif widget.prompt == "Base path:":
+                data.problem_path = widget.value
+            elif widget.prompt == "Solve with:":
+                data.solve_with = [str(s).strip() for s in widget.value.split(",")]
+    manager.stop()
+
+def confirm_data(data: Data) -> None:
     """ Show the tui using the pytermgui for confirm the data and choose the language to solve the problem """
     with ptg.WindowManager() as wm:
         wm.layout.add_slot("Body")
@@ -71,16 +85,14 @@ def confirm_data(data: Data) -> Data:
                 ptg.InputField(data.title, prompt="Title: "),
                 ptg.InputField(data.level, prompt="Level: "),
                 ptg.InputField(data.problem_path, prompt="Base path:"),
+                ptg.InputField("rust,", prompt="Solve with:"),
                 "",
-                ["Confirm", lambda *_: wm.stop()],
+                ["Confirm", lambda *_: parse_data(wm, window, data)],
                 width=60,
                 box="DOUBLE",
             )
             .set_title("[green bold]Confirm the data")
             .center()
         )
         wm.add(window)
-        wm.run()
-    
-    data.solve_with = ["python", "java", "c++", "c", "go"]
-    return data
+        wm.run()