♻️ Refactor generators and add an about window

src/window/app.py

@@ -1,20 +1,15 @@
-import io
-import os
-import tempfile
 import tkinter
 from pathlib import Path
 from tkinter import *
 from tkinter import filedialog
-
-from datetime import datetime
 import customtkinter
-import requests
-from PIL import Image
-
 from src.entity.config import Config
+from src.entity.generator import Generator
 from src.util import Util
+from src.window.about import AboutWindow
 from src.window.image import ImageWindow
 from src.window.session import SessionWindow
+from CTkMenuBar import *
 
 
 class App:
@@ -30,13 +25,20 @@ class App:
         self.selected_folder = ""
         self.found_images = []
         self.timer = 0
+
+        menu = CTkMenuBar(master=root)
+        generate_button = menu.add_cascade("Generate")
+        help_button = menu.add_cascade("Help")
+
+        dropdown = CustomDropdownMenu(widget=generate_button)
+        dropdown.add_option(option="Poses", command=lambda:self.download_poses('poses'))
+
+        about = CustomDropdownMenu(widget=help_button)
+        about.add_option(option="À propos", command=self.show_help)
+
         self.title = customtkinter.CTkLabel(root, text="Drawing Training")
         self.title.pack()
 
-        self.download_from_reference = customtkinter.CTkButton(root, text="Download 3 images from sketch daily",
-                                                               command=self.download_from_reference)
-        self.download_from_reference.pack()
-
         self.folder_selector = customtkinter.CTkButton(root, text="Select a folder", command=self.select_folder)
         self.folder_selector.pack()
 
@@ -79,6 +81,9 @@ class App:
         except Exception:
             pass
 
+    def show_help(self):
+        AboutWindow(self)
+
     def lets_draw(self):
         self.image_window = ImageWindow(self)
         self.image_window.lets_draw(self.found_images.copy(), self.timer)
@@ -96,30 +101,9 @@ class App:
             self.find_images_in_folder()
             self.check_lets_draw()
 
-    def download_from_reference(self):
-        dt = datetime.now()
-        ts = int(datetime.timestamp(dt))
-        url = 'https://quickposes.com/ajax/getPosesForSlider/Chiaroscuro?_=' + str(ts)
-
-        resp_json = requests.get(url=url)
-        if resp_json.status_code == 200:
-            data = resp_json.json()
-            c = 0
-            for src in data:
-                if c >= 3:
-                    break
-                buffer = tempfile.SpooledTemporaryFile(max_size=1e9)
-                r = requests.get('https://quickposes.com/assets/poses/' + src['src'], stream=True)
-                if r.status_code == 200:
-                    downloaded = 0
-                    for chunk in r.iter_content(chunk_size=1024):
-                        downloaded += len(chunk)
-                        buffer.write(chunk)
-                    buffer.seek(0)
-                    i = Image.open(io.BytesIO(buffer.read()))
-                    i.save(os.path.join(self.selected_folder, src['src']), quality=85)
-                buffer.close()
-                c += 1
+    def download_poses(self, generator=None):
+        Generator(generator=generator, folder=self.selected_folder)
+        self.find_images_in_folder()
 
     def find_images_in_folder(self):
         if Path(self.selected_folder).exists():