Image Generation. Thats all I can do today sorry

Author: MF366-Coding

main.py

@@ -324,6 +324,7 @@ class __InvalidButtonAction(Exception): ...
 PH_DATA: list[str] | None = None
 POINTS: int = 0
 GEN_SF: int = 0
+CUR_IMAGE = None
 
 
 class Database:
@@ -831,6 +832,29 @@ def generate_new_screenshot(**kw) -> None:
     return
 
 
+def generate_new_image(**kw) -> str:
+    """
+    # generate_new_image
+
+    Get an image from the selected data.
+    
+    :param database: overwrite for CUR_DB *(should be of type Database)*
+
+    Returns:
+        str: the link to the new image *(it also saves it as a global so it might not be necessary to get the returned value)*
+    """
+    
+    global CUR_IMAGE
+    
+    db: Database = kw.get('database', CUR_DB)
+
+    x = random.choice(db.structure[CUR_DATA[0]][CUR_DATA[1]][CUR_DATA[2]]['screenshots'].remove(CUR_IMAGE))
+    CUR_IMAGE = x
+    del x
+    
+    return CUR_IMAGE
+
+
 def setup_play_screen():
     PLAY_ITEMS.heading = ttk.Label(PLAY_ITEMS, text='Play', font=HEADING1)
     PLAY_ITEMS.f1 = ttk.Frame(PLAY_ITEMS)