Upload 18 files

.gitattributes

@@ -25,7 +25,6 @@
 *.safetensors filter=lfs diff=lfs merge=lfs -text
 saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
 *.tflite filter=lfs diff=lfs merge=lfs -text
 *.tgz filter=lfs diff=lfs merge=lfs -text
 *.wasm filter=lfs diff=lfs merge=lfs -text
@@ -33,3 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+driver/chromedriver filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -0,0 +1,38 @@
+FROM python:3.10
+
+# Install Google Chrome
+RUN apt-get update && apt-get install -y wget gnupg2
+RUN wget -q -O - https://dl-ssl.google.com/linux/linux_signing_key.pub | apt-key add -
+RUN echo "deb [arch=amd64] http://dl.google.com/linux/chrome/deb/ stable main" >> /etc/apt/sources.list.d/google-chrome.list
+RUN apt-get update && apt-get install -y google-chrome-stable
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Set up a new user named "user" with user ID 1000
+RUN useradd -m -u 1000 user
+# Switch to the "user" user
+USER user
+# Set home to the user's home directory
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+
+# Set the working directory to the user's home directory
+WORKDIR $HOME/app
+
+# Copy the current directory contents into the container at $HOME/app setting the owner to the user
+COPY --chown=user . $HOME/app
+
+# Install the ML models
+RUN mkdir -p /home/user/app/cv_files/ && \
+    curl -L https://huggingface.co/datasets/Kyo-Kai/Fsg_pp_files/resolve/main/AniClassifier.pt?download=true -o /home/user/app/cv_files/AniClassifier.pt && \
+    curl -L https://huggingface.co/datasets/Kyo-Kai/Fsg_pp_files/resolve/main/AniFaceDet.pt?download=true -o /home/user/app/cv_files/AniFaceDet.pt
+
+# Give executable permissions to chromedriver
+RUN chmod +x $HOME/app/driver/chromedriver
+
+# Run the python file
+CMD ["uvicorn", "Fsg_pp:app", "--host", "0.0.0.0", "--port", "7860"]

Fsg_pp.py

@@ -0,0 +1,235 @@
+import numpy as np
+import gradio as gr
+import os
+import commands.exec_path as exec_path
+import commands.driver_instance as driver_instance
+import glob
+import logging
+
+from commands.universal import searchQuery
+from ai.autocrop import autoCropImages
+from sites.pixiv import getOrderedPixivImages
+from sites.danbooru import getOrderedDanbooruImages
+from sites.zerochan import getOrderedZerochanImages
+from sites.yandex import getOrderedYandexImages
+
+
+
+logging.basicConfig(level=logging.INFO)
+class ImageGallery:
+    def __init__(self):
+        self.imgz = []
+        self.selected = 0
+        self.search_counter = 0
+
+    def return_images(self, image_locs):
+        self.imgz = image_locs
+
+        total_images = len(glob.glob('./Images/*'))
+        if total_images >= 20:
+            os.system("rm -r ./Images")
+            os.makedirs("./Images")
+
+        # Log the return value
+        logging.info(self.imgz)
+        return self.imgz if self.imgz else []
+
+    def get_select_index(self, evt: gr.SelectData):
+        self.selected = evt.index
+        return self.selected
+
+    def send_number(self): 
+        return self.imgz[int(self.selected)], gr.Tabs(selected=0)
+
+# Create an instance of ImageGallery for each tab
+pixiv_gallery = ImageGallery()
+danbooru_gallery = ImageGallery()
+zerochan_gallery = ImageGallery()
+yandex_gallery = ImageGallery()
+
+
+# Helper functions
+def pix_imgs(searchQuery, num_pics, num_pages,searchTypes,viewRestriction,imageControl,n_likes, n_bookmarks, n_views, 
+             start_date, end_date, user_name, pass_word):
+    driver = driver_instance.create_driver(profile=1)
+    return pixiv_gallery.return_images(getOrderedPixivImages(driver=driver, exec_path=exec_path, user_search=searchQuery, num_pics=num_pics, num_pages=num_pages,searchTypes=searchTypes,viewRestriction=viewRestriction,imageControl=imageControl, n_likes=n_likes, n_bookmarks=n_bookmarks,
+                                n_views=n_views, start_date=start_date,end_date=end_date, user_name=user_name, pass_word=pass_word))
+
+def danb_imgs(searchQuery, num_pics, num_pages, filters, bl_tags, inc_tags,imageControl):
+    driver = driver_instance.create_driver()
+    return danbooru_gallery.return_images(getOrderedDanbooruImages(driver=driver, user_search=searchQuery, num_pics=num_pics, num_pages=num_pages, filters=filters, bl_tags=bl_tags, inc_tags=inc_tags, exec_path=exec_path,imageControl=imageControl))
+
+def zero_imgs(searchQuery, num_pics, num_pages, n_likes, filters,imageControl):
+    driver = driver_instance.create_driver()
+    return zerochan_gallery.return_images(getOrderedZerochanImages(driver=driver, exec_path=exec_path, user_search=searchQuery, num_pics=num_pics, num_pages=num_pages, n_likes=n_likes, filters=filters,imageControl=imageControl))
+
+def yandex_imgs(searchQuery, num_pics, filters,imageOrientation):
+    driver = driver_instance.create_driver()
+    return yandex_gallery.return_images(getOrderedYandexImages(driver=driver, exec_path=exec_path, user_search=searchQuery, num_pics=num_pics, filters=filters,imageOrientation=imageOrientation))
+
+
+
+# Feature Functions
+def open_folder(folder_path, mode=0):
+    folder_opened = os.path.abspath(folder_path)
+    if mode:
+        folder_opened = os.path.abspath(os.path.join(folder_path, "cropped"))
+    os.system(f'open "{folder_opened}"' if os.name == 'posix' else f'explorer "{folder_opened}"')
+
+def cropImages(image,crop_scale_factor):
+    return autoCropImages(image,crop_scale_factor)
+
+def create_gallery_tab(tab_name, search_fn, search_inputs, gallery_instance, fn_on_click):
+    with gr.Column():
+        gallery=gr.Gallery(label="Image Preview", preview=True, object_fit="cover", container=True, columns=5)
+
+        with gr.Row():
+            crop_btn = gr.Button(value="Crop Selected Image",variant='secondary')
+            crop_btn.click(fn=fn_on_click, outputs=[image,tabs])
+            open_btn = gr.Button(value="Open 📁",variant='secondary')
+            open_btn.click(fn=open_folder, inputs=folder_input)
+        
+        with gr.Row():
+                        gr.HTML('''<div>
+                            <p style="margin-top: 20px; font-size: 1.25rem;">For testing purposes only. AI mode is set to be always on even if unchecked. You will experience lag due to hosting limitations, also due to extra throttling imposed</p>
+                            <p style="font-size: 1.25rem;">For the full experience, please check out the GitHub page:</p>
+                            <p style="font-size: 1.25rem;"><a href="https://github.com/EngMarchG/Fsg-Pp">Fsg-Pp - Finally Some Good Profile Pictures</a></p>
+                        </div>''')
+
+    gallery.select(gallery_instance.get_select_index, None)
+    green_btn.click(search_fn, search_inputs, outputs=gallery)
+
+
+# Main Layout of the GUI
+with gr.Blocks(css='style.css') as demo:
+    with gr.Tabs(selected=1) as tabs:
+        folder_input = gr.Textbox(value="./Images/", label="Enter Folder Path", visible=False)
+        
+        # Automatic Crop Tab
+        with gr.TabItem("Automatic Crop", id=0):
+            with gr.Row():
+                with gr.Column():
+                    image = gr.Image(type="filepath")
+                    crop_scale_factor = gr.Slider(0.5,3, value=1.2,step=0.1, label="Crop Scale Factor")
+                with gr.Column():
+                    outputImages = gr.Gallery(label="Cropped Image Preview", preview=True, object_fit="cover", container=True)
+
+                    with gr.Row():
+                        green_btn = gr.Button(value="Crop Image", size='sm')
+                        green_btn.click(cropImages, [image,crop_scale_factor],outputs=outputImages)
+                        open_btn = gr.Button(value="Open 📁",variant='secondary', size='sm')
+                        open_btn.click(fn=open_folder, inputs=[folder_input,crop_scale_factor])
+                    with gr.Row():
+                        gr.HTML('''<div>
+                            <p style="margin-top: 20px; font-size: 1.25rem;">For testing purposes only. AI mode is set to be always on even if unchecked. You will experience lag due to hosting limitations, also due to extra throttling imposed</p>
+                            <p style="font-size: 1.25rem;">For the full experience, please check out the GitHub page:</p>
+                            <p style="font-size: 1.25rem;"><a href="https://github.com/EngMarchG/Fsg-Pp">Fsg-Pp - Finally Some Good Profile Pictures</a></p>
+                        </div>''')
+
+
+        # Pixiv Tab
+        with gr.TabItem("Pixiv", id=1):
+            with gr.Row():
+                with gr.Column():
+                    searchQuery = gr.Textbox(label="Search Query", placeholder="Suggested to use the char's full name")
+                    with gr.Row():
+                        num_pics = gr.Slider(1,4, value=2, step=int, label="Number of Pictures")
+                    with gr.Row():
+                        num_pages = gr.Slider(1,3, value=1, step=int, label="Number of Pages")
+                    with gr.Row():
+                        with gr.Column():
+                            with gr.Row():
+                                searchTypes = gr.CheckboxGroup(["Premium Search","Freemium"], value=["Freemium"], label="Search Type", type="index", elem_id="pixiv")
+                            with gr.Row():
+                                viewRestriction = gr.CheckboxGroup(["PG","R-18"],label="Viewing Restriction (Default: Account Settings)",type="index",elem_id="viewing-restrictions")
+                        with gr.Row(elem_id='button-row'):
+                            green_btn = gr.Button(value="Search")
+                    with gr.Row():
+                        imageControl = gr.CheckboxGroup(["Full Res", "Continue Search","Search by Oldest", "AI Classifier"], value=["Full Res"], label="Image Control", type="index",elem_id="pixiv-filters")
+                    with gr.Row():
+                        with gr.Row():
+                            n_likes = gr.Number(value=0, label="Filter by Likes")
+                        with gr.Row():
+                            n_bookmarks = gr.Number(value=0, label="Filter by Bookmarks")
+                        with gr.Row():
+                            n_views = gr.Number(value=0, label="Filter by Views")
+                    with gr.Row():
+                            start_date = gr.Textbox(label="Start date", placeholder=("2016-01-22  YEAR-MONTH-DAY"))
+                    with gr.Row():
+                            end_date = gr.Textbox(label="End date", placeholder=("2022-09-22  YEAR-MONTH-DAY"))
+                    with gr.Row():
+                            user_name = gr.Textbox(label="Email", type="email")
+                    with gr.Row():
+                            pass_word = gr.Textbox(label="Password", type="password")
+                    
+                pixiv_inputs = [searchQuery, num_pics, num_pages,searchTypes,viewRestriction,imageControl,n_likes, n_bookmarks, n_views, 
+                                start_date,end_date, user_name, pass_word]
+                create_gallery_tab("Pixiv", pix_imgs, pixiv_inputs, pixiv_gallery, pixiv_gallery.send_number)
+
+
+        # Danbooru Tab
+        with gr.TabItem("Danbooru", id=2):
+            with gr.Row():
+                with gr.Column():
+                    searchQuery = gr.Textbox(label="Search Query", placeholder="Suggested to use the char's full name")
+                    with gr.Row():
+                        num_pics = gr.Slider(1,4, value=2, step=int, label="Number of Pictures")
+                    with gr.Row():
+                        num_pages = gr.Slider(1,3, value=1, step=int, label="Number of Pages")
+                    with gr.Row():
+                        filters = gr.CheckboxGroup(["Score", "Exact Match", "More PG", "Sensitive", "Strictly PG", "AI Classifier"], label="Filters", type="index", elem_id="filtering")
+                    with gr.Row():
+                        imageControl = gr.CheckboxGroup(["Continue Search"], label="Image Control", type="index", elem_id="imageControl")
+                    with gr.Row():
+                        bl_tags = gr.Textbox(label="Tags to Filter", placeholder=("Add stuff like typical undergarments etc to ensure complete pg friendliness"),lines=2)
+                    with gr.Row():
+                        inc_tags = gr.Textbox(label="Tags to Include", placeholder=("1girl, 1boy for profile pictures"))
+                    green_btn = gr.Button(value="Search")
+            
+                danbooru_inputs = [searchQuery, num_pics, num_pages, filters, bl_tags, inc_tags,imageControl]
+                create_gallery_tab("Danbooru", danb_imgs, danbooru_inputs, danbooru_gallery, danbooru_gallery.send_number)
+
+
+        # Zerochan Tab
+        with gr.TabItem("Zerochan", id=3):
+            with gr.Row():
+                with gr.Column():
+                    searchQuery = gr.Textbox(label="Search Query", placeholder="Suggested to use the char's full name")
+                    with gr.Row():
+                        num_pics = gr.Slider(1,4, value=2, step=int, label="Number of Pictures")
+                    with gr.Row():
+                        num_pages = gr.Slider(1,3, value=1, step=int, label="Number of Pages")
+                    with gr.Row():
+                        with gr.Row():
+                            n_likes = gr.Number(value=0, label="Filter by Likes")
+                            with gr.Row():
+                                filters = gr.CheckboxGroup(["AI Classifier"], label="Filters", type="index",elem_id="zeroAIhover")
+                        with gr.Column():
+                            imageControl = gr.CheckboxGroup(["Continue Search"], label="Image Control", type="index", elem_id="imageControl")   
+                    green_btn = gr.Button(value="Search")
+                
+                with gr.Column():
+                    zerochan_inputs = [searchQuery, num_pics, num_pages, n_likes, filters,imageControl]
+                    create_gallery_tab("Zerochan", zero_imgs, zerochan_inputs, zerochan_gallery, zerochan_gallery.send_number)
+
+
+    # Yandex Tab
+        with gr.TabItem("Yandex", id=4):
+            with gr.Row():
+                with gr.Column():
+                    searchQuery = gr.Textbox(label="Search Query", placeholder="Suggested to use the char's full name")
+                    with gr.Row():
+                        num_pics = gr.Slider(1,10, value=2, step=int, label="Number of Pictures")
+                    with gr.Row():
+                        with gr.Row():
+                                filters = gr.CheckboxGroup(["AI Classifier","Search By Recent"], label="Filters", type="index",elem_id="zeroAIhover")
+                        with gr.Column():
+                            imageOrientation = gr.Radio(["Landscape","Portrait","Square"], label="Image Orientation", type="index", elem_id="imageControl")   
+                    green_btn = gr.Button(value="Search")
+                
+                yandex_inputs = [searchQuery, num_pics, filters,imageOrientation]
+                create_gallery_tab("Yandex", yandex_imgs, yandex_inputs, yandex_gallery, yandex_gallery.send_number)
+
+
+
+demo.launch(demo.launch(server_name="0.0.0.0", server_port=7860))

LICENSE

@@ -0,0 +1,661 @@
+                    GNU AFFERO GENERAL PUBLIC LICENSE
+                       Version 3, 19 November 2007
+
+ Copyright (C) 2007 Free Software Foundation, Inc. <https://fsf.org/>
+ Everyone is permitted to copy and distribute verbatim copies
+ of this license document, but changing it is not allowed.
+
+                            Preamble
+
+  The GNU Affero General Public License is a free, copyleft license for
+software and other kinds of works, specifically designed to ensure
+cooperation with the community in the case of network server software.
+
+  The licenses for most software and other practical works are designed
+to take away your freedom to share and change the works.  By contrast,
+our General Public Licenses are intended to guarantee your freedom to
+share and change all versions of a program--to make sure it remains free
+software for all its users.
+
+  When we speak of free software, we are referring to freedom, not
+price.  Our General Public Licenses are designed to make sure that you
+have the freedom to distribute copies of free software (and charge for
+them if you wish), that you receive source code or can get it if you
+want it, that you can change the software or use pieces of it in new
+free programs, and that you know you can do these things.
+
+  Developers that use our General Public Licenses protect your rights
+with two steps: (1) assert copyright on the software, and (2) offer
+you this License which gives you legal permission to copy, distribute
+and/or modify the software.
+
+  A secondary benefit of defending all users' freedom is that
+improvements made in alternate versions of the program, if they
+receive widespread use, become available for other developers to
+incorporate.  Many developers of free software are heartened and
+encouraged by the resulting cooperation.  However, in the case of
+software used on network servers, this result may fail to come about.
+The GNU General Public License permits making a modified version and
+letting the public access it on a server without ever releasing its
+source code to the public.
+
+  The GNU Affero General Public License is designed specifically to
+ensure that, in such cases, the modified source code becomes available
+to the community.  It requires the operator of a network server to
+provide the source code of the modified version running there to the
+users of that server.  Therefore, public use of a modified version, on
+a publicly accessible server, gives the public access to the source
+code of the modified version.
+
+  An older license, called the Affero General Public License and
+published by Affero, was designed to accomplish similar goals.  This is
+a different license, not a version of the Affero GPL, but Affero has
+released a new version of the Affero GPL which permits relicensing under
+this license.
+
+  The precise terms and conditions for copying, distribution and
+modification follow.
+
+                       TERMS AND CONDITIONS
+
+  0. Definitions.
+
+  "This License" refers to version 3 of the GNU Affero General Public License.
+
+  "Copyright" also means copyright-like laws that apply to other kinds of
+works, such as semiconductor masks.
+
+  "The Program" refers to any copyrightable work licensed under this
+License.  Each licensee is addressed as "you".  "Licensees" and
+"recipients" may be individuals or organizations.
+
+  To "modify" a work means to copy from or adapt all or part of the work
+in a fashion requiring copyright permission, other than the making of an
+exact copy.  The resulting work is called a "modified version" of the
+earlier work or a work "based on" the earlier work.
+
+  A "covered work" means either the unmodified Program or a work based
+on the Program.
+
+  To "propagate" a work means to do anything with it that, without
+permission, would make you directly or secondarily liable for
+infringement under applicable copyright law, except executing it on a
+computer or modifying a private copy.  Propagation includes copying,
+distribution (with or without modification), making available to the
+public, and in some countries other activities as well.
+
+  To "convey" a work means any kind of propagation that enables other
+parties to make or receive copies.  Mere interaction with a user through
+a computer network, with no transfer of a copy, is not conveying.
+
+  An interactive user interface displays "Appropriate Legal Notices"
+to the extent that it includes a convenient and prominently visible
+feature that (1) displays an appropriate copyright notice, and (2)
+tells the user that there is no warranty for the work (except to the
+extent that warranties are provided), that licensees may convey the
+work under this License, and how to view a copy of this License.  If
+the interface presents a list of user commands or options, such as a
+menu, a prominent item in the list meets this criterion.
+
+  1. Source Code.
+
+  The "source code" for a work means the preferred form of the work
+for making modifications to it.  "Object code" means any non-source
+form of a work.
+
+  A "Standard Interface" means an interface that either is an official
+standard defined by a recognized standards body, or, in the case of
+interfaces specified for a particular programming language, one that
+is widely used among developers working in that language.
+
+  The "System Libraries" of an executable work include anything, other
+than the work as a whole, that (a) is included in the normal form of
+packaging a Major Component, but which is not part of that Major
+Component, and (b) serves only to enable use of the work with that
+Major Component, or to implement a Standard Interface for which an
+implementation is available to the public in source code form.  A
+"Major Component", in this context, means a major essential component
+(kernel, window system, and so on) of the specific operating system
+(if any) on which the executable work runs, or a compiler used to
+produce the work, or an object code interpreter used to run it.
+
+  The "Corresponding Source" for a work in object code form means all
+the source code needed to generate, install, and (for an executable
+work) run the object code and to modify the work, including scripts to
+control those activities.  However, it does not include the work's
+System Libraries, or general-purpose tools or generally available free
+programs which are used unmodified in performing those activities but
+which are not part of the work.  For example, Corresponding Source
+includes interface definition files associated with source files for
+the work, and the source code for shared libraries and dynamically
+linked subprograms that the work is specifically designed to require,
+such as by intimate data communication or control flow between those
+subprograms and other parts of the work.
+
+  The Corresponding Source need not include anything that users
+can regenerate automatically from other parts of the Corresponding
+Source.
+
+  The Corresponding Source for a work in source code form is that
+same work.
+
+  2. Basic Permissions.
+
+  All rights granted under this License are granted for the term of
+copyright on the Program, and are irrevocable provided the stated
+conditions are met.  This License explicitly affirms your unlimited
+permission to run the unmodified Program.  The output from running a
+covered work is covered by this License only if the output, given its
+content, constitutes a covered work.  This License acknowledges your
+rights of fair use or other equivalent, as provided by copyright law.
+
+  You may make, run and propagate covered works that you do not
+convey, without conditions so long as your license otherwise remains
+in force.  You may convey covered works to others for the sole purpose
+of having them make modifications exclusively for you, or provide you
+with facilities for running those works, provided that you comply with
+the terms of this License in conveying all material for which you do
+not control copyright.  Those thus making or running the covered works
+for you must do so exclusively on your behalf, under your direction
+and control, on terms that prohibit them from making any copies of
+your copyrighted material outside their relationship with you.
+
+  Conveying under any other circumstances is permitted solely under
+the conditions stated below.  Sublicensing is not allowed; section 10
+makes it unnecessary.
+
+  3. Protecting Users' Legal Rights From Anti-Circumvention Law.
+
+  No covered work shall be deemed part of an effective technological
+measure under any applicable law fulfilling obligations under article
+11 of the WIPO copyright treaty adopted on 20 December 1996, or
+similar laws prohibiting or restricting circumvention of such
+measures.
+
+  When you convey a covered work, you waive any legal power to forbid
+circumvention of technological measures to the extent such circumvention
+is effected by exercising rights under this License with respect to
+the covered work, and you disclaim any intention to limit operation or
+modification of the work as a means of enforcing, against the work's
+users, your or third parties' legal rights to forbid circumvention of
+technological measures.
+
+  4. Conveying Verbatim Copies.
+
+  You may convey verbatim copies of the Program's source code as you
+receive it, in any medium, provided that you conspicuously and
+appropriately publish on each copy an appropriate copyright notice;
+keep intact all notices stating that this License and any
+non-permissive terms added in accord with section 7 apply to the code;
+keep intact all notices of the absence of any warranty; and give all
+recipients a copy of this License along with the Program.
+
+  You may charge any price or no price for each copy that you convey,
+and you may offer support or warranty protection for a fee.
+
+  5. Conveying Modified Source Versions.
+
+  You may convey a work based on the Program, or the modifications to
+produce it from the Program, in the form of source code under the
+terms of section 4, provided that you also meet all of these conditions:
+
+    a) The work must carry prominent notices stating that you modified
+    it, and giving a relevant date.
+
+    b) The work must carry prominent notices stating that it is
+    released under this License and any conditions added under section
+    7.  This requirement modifies the requirement in section 4 to
+    "keep intact all notices".
+
+    c) You must license the entire work, as a whole, under this
+    License to anyone who comes into possession of a copy.  This
+    License will therefore apply, along with any applicable section 7
+    additional terms, to the whole of the work, and all its parts,
+    regardless of how they are packaged.  This License gives no
+    permission to license the work in any other way, but it does not
+    invalidate such permission if you have separately received it.
+
+    d) If the work has interactive user interfaces, each must display
+    Appropriate Legal Notices; however, if the Program has interactive
+    interfaces that do not display Appropriate Legal Notices, your
+    work need not make them do so.
+
+  A compilation of a covered work with other separate and independent
+works, which are not by their nature extensions of the covered work,
+and which are not combined with it such as to form a larger program,
+in or on a volume of a storage or distribution medium, is called an
+"aggregate" if the compilation and its resulting copyright are not
+used to limit the access or legal rights of the compilation's users
+beyond what the individual works permit.  Inclusion of a covered work
+in an aggregate does not cause this License to apply to the other
+parts of the aggregate.
+
+  6. Conveying Non-Source Forms.
+
+  You may convey a covered work in object code form under the terms
+of sections 4 and 5, provided that you also convey the
+machine-readable Corresponding Source under the terms of this License,
+in one of these ways:
+
+    a) Convey the object code in, or embodied in, a physical product
+    (including a physical distribution medium), accompanied by the
+    Corresponding Source fixed on a durable physical medium
+    customarily used for software interchange.
+
+    b) Convey the object code in, or embodied in, a physical product
+    (including a physical distribution medium), accompanied by a
+    written offer, valid for at least three years and valid for as
+    long as you offer spare parts or customer support for that product
+    model, to give anyone who possesses the object code either (1) a
+    copy of the Corresponding Source for all the software in the
+    product that is covered by this License, on a durable physical
+    medium customarily used for software interchange, for a price no
+    more than your reasonable cost of physically performing this
+    conveying of source, or (2) access to copy the
+    Corresponding Source from a network server at no charge.
+
+    c) Convey individual copies of the object code with a copy of the
+    written offer to provide the Corresponding Source.  This
+    alternative is allowed only occasionally and noncommercially, and
+    only if you received the object code with such an offer, in accord
+    with subsection 6b.
+
+    d) Convey the object code by offering access from a designated
+    place (gratis or for a charge), and offer equivalent access to the
+    Corresponding Source in the same way through the same place at no
+    further charge.  You need not require recipients to copy the
+    Corresponding Source along with the object code.  If the place to
+    copy the object code is a network server, the Corresponding Source
+    may be on a different server (operated by you or a third party)
+    that supports equivalent copying facilities, provided you maintain
+    clear directions next to the object code saying where to find the
+    Corresponding Source.  Regardless of what server hosts the
+    Corresponding Source, you remain obligated to ensure that it is
+    available for as long as needed to satisfy these requirements.
+
+    e) Convey the object code using peer-to-peer transmission, provided
+    you inform other peers where the object code and Corresponding
+    Source of the work are being offered to the general public at no
+    charge under subsection 6d.
+
+  A separable portion of the object code, whose source code is excluded
+from the Corresponding Source as a System Library, need not be
+included in conveying the object code work.
+
+  A "User Product" is either (1) a "consumer product", which means any
+tangible personal property which is normally used for personal, family,
+or household purposes, or (2) anything designed or sold for incorporation
+into a dwelling.  In determining whether a product is a consumer product,
+doubtful cases shall be resolved in favor of coverage.  For a particular
+product received by a particular user, "normally used" refers to a
+typical or common use of that class of product, regardless of the status
+of the particular user or of the way in which the particular user
+actually uses, or expects or is expected to use, the product.  A product
+is a consumer product regardless of whether the product has substantial
+commercial, industrial or non-consumer uses, unless such uses represent
+the only significant mode of use of the product.
+
+  "Installation Information" for a User Product means any methods,
+procedures, authorization keys, or other information required to install
+and execute modified versions of a covered work in that User Product from
+a modified version of its Corresponding Source.  The information must
+suffice to ensure that the continued functioning of the modified object
+code is in no case prevented or interfered with solely because
+modification has been made.
+
+  If you convey an object code work under this section in, or with, or
+specifically for use in, a User Product, and the conveying occurs as
+part of a transaction in which the right of possession and use of the
+User Product is transferred to the recipient in perpetuity or for a
+fixed term (regardless of how the transaction is characterized), the
+Corresponding Source conveyed under this section must be accompanied
+by the Installation Information.  But this requirement does not apply
+if neither you nor any third party retains the ability to install
+modified object code on the User Product (for example, the work has
+been installed in ROM).
+
+  The requirement to provide Installation Information does not include a
+requirement to continue to provide support service, warranty, or updates
+for a work that has been modified or installed by the recipient, or for
+the User Product in which it has been modified or installed.  Access to a
+network may be denied when the modification itself materially and
+adversely affects the operation of the network or violates the rules and
+protocols for communication across the network.
+
+  Corresponding Source conveyed, and Installation Information provided,
+in accord with this section must be in a format that is publicly
+documented (and with an implementation available to the public in
+source code form), and must require no special password or key for
+unpacking, reading or copying.
+
+  7. Additional Terms.
+
+  "Additional permissions" are terms that supplement the terms of this
+License by making exceptions from one or more of its conditions.
+Additional permissions that are applicable to the entire Program shall
+be treated as though they were included in this License, to the extent
+that they are valid under applicable law.  If additional permissions
+apply only to part of the Program, that part may be used separately
+under those permissions, but the entire Program remains governed by
+this License without regard to the additional permissions.
+
+  When you convey a copy of a covered work, you may at your option
+remove any additional permissions from that copy, or from any part of
+it.  (Additional permissions may be written to require their own
+removal in certain cases when you modify the work.)  You may place
+additional permissions on material, added by you to a covered work,
+for which you have or can give appropriate copyright permission.
+
+  Notwithstanding any other provision of this License, for material you
+add to a covered work, you may (if authorized by the copyright holders of
+that material) supplement the terms of this License with terms:
+
+    a) Disclaiming warranty or limiting liability differently from the
+    terms of sections 15 and 16 of this License; or
+
+    b) Requiring preservation of specified reasonable legal notices or
+    author attributions in that material or in the Appropriate Legal
+    Notices displayed by works containing it; or
+
+    c) Prohibiting misrepresentation of the origin of that material, or
+    requiring that modified versions of such material be marked in
+    reasonable ways as different from the original version; or
+
+    d) Limiting the use for publicity purposes of names of licensors or
+    authors of the material; or
+
+    e) Declining to grant rights under trademark law for use of some
+    trade names, trademarks, or service marks; or
+
+    f) Requiring indemnification of licensors and authors of that
+    material by anyone who conveys the material (or modified versions of
+    it) with contractual assumptions of liability to the recipient, for
+    any liability that these contractual assumptions directly impose on
+    those licensors and authors.
+
+  All other non-permissive additional terms are considered "further
+restrictions" within the meaning of section 10.  If the Program as you
+received it, or any part of it, contains a notice stating that it is
+governed by this License along with a term that is a further
+restriction, you may remove that term.  If a license document contains
+a further restriction but permits relicensing or conveying under this
+License, you may add to a covered work material governed by the terms
+of that license document, provided that the further restriction does
+not survive such relicensing or conveying.
+
+  If you add terms to a covered work in accord with this section, you
+must place, in the relevant source files, a statement of the
+additional terms that apply to those files, or a notice indicating
+where to find the applicable terms.
+
+  Additional terms, permissive or non-permissive, may be stated in the
+form of a separately written license, or stated as exceptions;
+the above requirements apply either way.
+
+  8. Termination.
+
+  You may not propagate or modify a covered work except as expressly
+provided under this License.  Any attempt otherwise to propagate or
+modify it is void, and will automatically terminate your rights under
+this License (including any patent licenses granted under the third
+paragraph of section 11).
+
+  However, if you cease all violation of this License, then your
+license from a particular copyright holder is reinstated (a)
+provisionally, unless and until the copyright holder explicitly and
+finally terminates your license, and (b) permanently, if the copyright
+holder fails to notify you of the violation by some reasonable means
+prior to 60 days after the cessation.
+
+  Moreover, your license from a particular copyright holder is
+reinstated permanently if the copyright holder notifies you of the
+violation by some reasonable means, this is the first time you have
+received notice of violation of this License (for any work) from that
+copyright holder, and you cure the violation prior to 30 days after
+your receipt of the notice.
+
+  Termination of your rights under this section does not terminate the
+licenses of parties who have received copies or rights from you under
+this License.  If your rights have been terminated and not permanently
+reinstated, you do not qualify to receive new licenses for the same
+material under section 10.
+
+  9. Acceptance Not Required for Having Copies.
+
+  You are not required to accept this License in order to receive or
+run a copy of the Program.  Ancillary propagation of a covered work
+occurring solely as a consequence of using peer-to-peer transmission
+to receive a copy likewise does not require acceptance.  However,
+nothing other than this License grants you permission to propagate or
+modify any covered work.  These actions infringe copyright if you do
+not accept this License.  Therefore, by modifying or propagating a
+covered work, you indicate your acceptance of this License to do so.
+
+  10. Automatic Licensing of Downstream Recipients.
+
+  Each time you convey a covered work, the recipient automatically
+receives a license from the original licensors, to run, modify and
+propagate that work, subject to this License.  You are not responsible
+for enforcing compliance by third parties with this License.
+
+  An "entity transaction" is a transaction transferring control of an
+organization, or substantially all assets of one, or subdividing an
+organization, or merging organizations.  If propagation of a covered
+work results from an entity transaction, each party to that
+transaction who receives a copy of the work also receives whatever
+licenses to the work the party's predecessor in interest had or could
+give under the previous paragraph, plus a right to possession of the
+Corresponding Source of the work from the predecessor in interest, if
+the predecessor has it or can get it with reasonable efforts.
+
+  You may not impose any further restrictions on the exercise of the
+rights granted or affirmed under this License.  For example, you may
+not impose a license fee, royalty, or other charge for exercise of
+rights granted under this License, and you may not initiate litigation
+(including a cross-claim or counterclaim in a lawsuit) alleging that
+any patent claim is infringed by making, using, selling, offering for
+sale, or importing the Program or any portion of it.
+
+  11. Patents.
+
+  A "contributor" is a copyright holder who authorizes use under this
+License of the Program or a work on which the Program is based.  The
+work thus licensed is called the contributor's "contributor version".
+
+  A contributor's "essential patent claims" are all patent claims
+owned or controlled by the contributor, whether already acquired or
+hereafter acquired, that would be infringed by some manner, permitted
+by this License, of making, using, or selling its contributor version,
+but do not include claims that would be infringed only as a
+consequence of further modification of the contributor version.  For
+purposes of this definition, "control" includes the right to grant
+patent sublicenses in a manner consistent with the requirements of
+this License.
+
+  Each contributor grants you a non-exclusive, worldwide, royalty-free
+patent license under the contributor's essential patent claims, to
+make, use, sell, offer for sale, import and otherwise run, modify and
+propagate the contents of its contributor version.
+
+  In the following three paragraphs, a "patent license" is any express
+agreement or commitment, however denominated, not to enforce a patent
+(such as an express permission to practice a patent or covenant not to
+sue for patent infringement).  To "grant" such a patent license to a
+party means to make such an agreement or commitment not to enforce a
+patent against the party.
+
+  If you convey a covered work, knowingly relying on a patent license,
+and the Corresponding Source of the work is not available for anyone
+to copy, free of charge and under the terms of this License, through a
+publicly available network server or other readily accessible means,
+then you must either (1) cause the Corresponding Source to be so
+available, or (2) arrange to deprive yourself of the benefit of the
+patent license for this particular work, or (3) arrange, in a manner
+consistent with the requirements of this License, to extend the patent
+license to downstream recipients.  "Knowingly relying" means you have
+actual knowledge that, but for the patent license, your conveying the
+covered work in a country, or your recipient's use of the covered work
+in a country, would infringe one or more identifiable patents in that
+country that you have reason to believe are valid.
+
+  If, pursuant to or in connection with a single transaction or
+arrangement, you convey, or propagate by procuring conveyance of, a
+covered work, and grant a patent license to some of the parties
+receiving the covered work authorizing them to use, propagate, modify
+or convey a specific copy of the covered work, then the patent license
+you grant is automatically extended to all recipients of the covered
+work and works based on it.
+
+  A patent license is "discriminatory" if it does not include within
+the scope of its coverage, prohibits the exercise of, or is
+conditioned on the non-exercise of one or more of the rights that are
+specifically granted under this License.  You may not convey a covered
+work if you are a party to an arrangement with a third party that is
+in the business of distributing software, under which you make payment
+to the third party based on the extent of your activity of conveying
+the work, and under which the third party grants, to any of the
+parties who would receive the covered work from you, a discriminatory
+patent license (a) in connection with copies of the covered work
+conveyed by you (or copies made from those copies), or (b) primarily
+for and in connection with specific products or compilations that
+contain the covered work, unless you entered into that arrangement,
+or that patent license was granted, prior to 28 March 2007.
+
+  Nothing in this License shall be construed as excluding or limiting
+any implied license or other defenses to infringement that may
+otherwise be available to you under applicable patent law.
+
+  12. No Surrender of Others' Freedom.
+
+  If conditions are imposed on you (whether by court order, agreement or
+otherwise) that contradict the conditions of this License, they do not
+excuse you from the conditions of this License.  If you cannot convey a
+covered work so as to satisfy simultaneously your obligations under this
+License and any other pertinent obligations, then as a consequence you may
+not convey it at all.  For example, if you agree to terms that obligate you
+to collect a royalty for further conveying from those to whom you convey
+the Program, the only way you could satisfy both those terms and this
+License would be to refrain entirely from conveying the Program.
+
+  13. Remote Network Interaction; Use with the GNU General Public License.
+
+  Notwithstanding any other provision of this License, if you modify the
+Program, your modified version must prominently offer all users
+interacting with it remotely through a computer network (if your version
+supports such interaction) an opportunity to receive the Corresponding
+Source of your version by providing access to the Corresponding Source
+from a network server at no charge, through some standard or customary
+means of facilitating copying of software.  This Corresponding Source
+shall include the Corresponding Source for any work covered by version 3
+of the GNU General Public License that is incorporated pursuant to the
+following paragraph.
+
+  Notwithstanding any other provision of this License, you have
+permission to link or combine any covered work with a work licensed
+under version 3 of the GNU General Public License into a single
+combined work, and to convey the resulting work.  The terms of this
+License will continue to apply to the part which is the covered work,
+but the work with which it is combined will remain governed by version
+3 of the GNU General Public License.
+
+  14. Revised Versions of this License.
+
+  The Free Software Foundation may publish revised and/or new versions of
+the GNU Affero General Public License from time to time.  Such new versions
+will be similar in spirit to the present version, but may differ in detail to
+address new problems or concerns.
+
+  Each version is given a distinguishing version number.  If the
+Program specifies that a certain numbered version of the GNU Affero General
+Public License "or any later version" applies to it, you have the
+option of following the terms and conditions either of that numbered
+version or of any later version published by the Free Software
+Foundation.  If the Program does not specify a version number of the
+GNU Affero General Public License, you may choose any version ever published
+by the Free Software Foundation.
+
+  If the Program specifies that a proxy can decide which future
+versions of the GNU Affero General Public License can be used, that proxy's
+public statement of acceptance of a version permanently authorizes you
+to choose that version for the Program.
+
+  Later license versions may give you additional or different
+permissions.  However, no additional obligations are imposed on any
+author or copyright holder as a result of your choosing to follow a
+later version.
+
+  15. Disclaimer of Warranty.
+
+  THERE IS NO WARRANTY FOR THE PROGRAM, TO THE EXTENT PERMITTED BY
+APPLICABLE LAW.  EXCEPT WHEN OTHERWISE STATED IN WRITING THE COPYRIGHT
+HOLDERS AND/OR OTHER PARTIES PROVIDE THE PROGRAM "AS IS" WITHOUT WARRANTY
+OF ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING, BUT NOT LIMITED TO,
+THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+PURPOSE.  THE ENTIRE RISK AS TO THE QUALITY AND PERFORMANCE OF THE PROGRAM
+IS WITH YOU.  SHOULD THE PROGRAM PROVE DEFECTIVE, YOU ASSUME THE COST OF
+ALL NECESSARY SERVICING, REPAIR OR CORRECTION.
+
+  16. Limitation of Liability.
+
+  IN NO EVENT UNLESS REQUIRED BY APPLICABLE LAW OR AGREED TO IN WRITING
+WILL ANY COPYRIGHT HOLDER, OR ANY OTHER PARTY WHO MODIFIES AND/OR CONVEYS
+THE PROGRAM AS PERMITTED ABOVE, BE LIABLE TO YOU FOR DAMAGES, INCLUDING ANY
+GENERAL, SPECIAL, INCIDENTAL OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE
+USE OR INABILITY TO USE THE PROGRAM (INCLUDING BUT NOT LIMITED TO LOSS OF
+DATA OR DATA BEING RENDERED INACCURATE OR LOSSES SUSTAINED BY YOU OR THIRD
+PARTIES OR A FAILURE OF THE PROGRAM TO OPERATE WITH ANY OTHER PROGRAMS),
+EVEN IF SUCH HOLDER OR OTHER PARTY HAS BEEN ADVISED OF THE POSSIBILITY OF
+SUCH DAMAGES.
+
+  17. Interpretation of Sections 15 and 16.
+
+  If the disclaimer of warranty and limitation of liability provided
+above cannot be given local legal effect according to their terms,
+reviewing courts shall apply local law that most closely approximates
+an absolute waiver of all civil liability in connection with the
+Program, unless a warranty or assumption of liability accompanies a
+copy of the Program in return for a fee.
+
+                     END OF TERMS AND CONDITIONS
+
+            How to Apply These Terms to Your New Programs
+
+  If you develop a new program, and you want it to be of the greatest
+possible use to the public, the best way to achieve this is to make it
+free software which everyone can redistribute and change under these terms.
+
+  To do so, attach the following notices to the program.  It is safest
+to attach them to the start of each source file to most effectively
+state the exclusion of warranty; and each file should have at least
+the "copyright" line and a pointer to where the full notice is found.
+
+    <one line to give the program's name and a brief idea of what it does.>
+    Copyright (C) <year>  <name of author>
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU Affero General Public License as published
+    by the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU Affero General Public License for more details.
+
+    You should have received a copy of the GNU Affero General Public License
+    along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+Also add information on how to contact you by electronic and paper mail.
+
+  If your software can interact with users remotely through a computer
+network, you should also make sure that it provides a way for users to
+get its source.  For example, if your program is a web application, its
+interface could display a "Source" link that leads users to an archive
+of the code.  There are many ways you could offer source, and different
+solutions will be better for different programs; see section 13 for the
+specific requirements.
+
+  You should also get your employer (if you work as a programmer) or school,
+if any, to sign a "copyright disclaimer" for the program, if necessary.
+For more information on this, and how to apply and follow the GNU AGPL, see
+<https://www.gnu.org/licenses/>.

README.md

@@ -1,11 +1,9 @@
 ---
-title: Fsg Pp
-emoji: 🌖
+title: Finally Some Good Profile Pictures 
+emoji: 😉
 colorFrom: purple
-colorTo: pink
+colorTo: blue
 sdk: docker
 pinned: false
-license: apache-2.0
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+license: gpl-3.0
+---

ai/autocrop.py

@@ -0,0 +1,88 @@
+import commands.exec_path 
+from ultralytics import YOLO
+from PIL import Image, ImageDraw, ImageFont
+import os
+import random
+from pathlib import Path
+
+def autoCropImages(image,scale_factor):
+    # Load a model
+    model = YOLO("./cv_files/AniFaceDet.pt")
+
+    test_images = []
+    test_images.append(image)
+
+    # Create a directory for saving cropped images
+    relative_dir = './Images/cropped'
+    cropped_dir = os.path.abspath(relative_dir)
+    if not os.path.exists(cropped_dir):
+        os.makedirs(cropped_dir)
+
+    imagesToReturn = []
+    # Load the test image
+    for img in test_images:
+        if img.split(".")[-1] not in ["jpg", "jpeg", "png"]:
+            continue
+        image_path = Path(image) / img
+        image = Image.open(image_path)
+
+        # Get the size of the image
+        image_width, image_height = image.size
+
+        # Calculate the scaling factor based on the image size for font size
+        scaling_factor = max(image_width, image_height) / 200
+        # Calculate the final font size by scaling the base font size
+        base_font_size = 10
+        font_size = int(base_font_size * scaling_factor)
+        font = ImageFont.load_default()
+
+        # Predict the bounding boxes #Defaults conf=0.25, iou=0.7
+        pred = model.predict(image, conf=0.65, iou=0.7)
+
+        # Extract the bounding box coordinates, class labels, and confidence scores
+        boxes = pred[0].boxes.xyxy.tolist()  
+        classes = pred[0].boxes.cls.tolist()  
+        scores = pred[0].boxes.conf.tolist()
+
+        # Choose a scale factor for the cropped image
+        scale_factor = scale_factor
+
+        # Loop over all detected faces and draw bounding boxes, crop, and save
+        for i in range(len(boxes)):
+            box = boxes[i]
+            score = scores[i]
+
+            x1, y1, x2, y2 = box
+
+            # Calculate the width and height of the bounding box and apply the scale factor
+            box_width = x2 - x1
+            box_height = y2 - y1
+            scaled_width = int(box_width * scale_factor)
+            scaled_height = int(box_height * scale_factor)
+
+            # Calculate the top-left corner coordinates of the cropped region
+            cropped_x1 = max(0, int(x1 - (scaled_width - box_width) / 2))
+            cropped_y1 = max(0, int(y1 - (scaled_height - box_height) / 2))
+
+            # Calculate the bottom-right corner coordinates of the cropped region
+            cropped_x2 = min(int(x2 + (scaled_width - box_width) / 2), image_width)
+            cropped_y2 = min(int(y2 + (scaled_height - box_height) / 2), image_height)
+
+            # Crop the image based on the detected face
+            cropped_image = image.crop((cropped_x1, cropped_y1, cropped_x2, cropped_y2))
+
+            # Save the cropped image with the original filename and an index
+            cropped_image_name = '{}_cropped_{}_scale_{}.jpg'.format(os.path.splitext(os.path.split(img)[-1])[0], i, scale_factor)
+            cropped_image_path = os.path.join(cropped_dir, cropped_image_name)
+            cropped_image.save(fp=cropped_image_path)
+
+            # Appending Cropped images in an array to display in gradio for end-user
+            imagesToReturn.append(cropped_image_path)
+
+            print('Cropped image saved:', cropped_image_path)
+
+            # Draw bounding boxes on the original image
+            draw = ImageDraw.Draw(image)
+
+    return imagesToReturn
+    

ai/classifying_ai.py

@@ -0,0 +1,44 @@
+import commands.exec_path 
+from ultralytics import YOLO
+from PIL import Image, ImageDraw, ImageFont, ImageFile
+import os
+import random
+
+model_path = os.path.join(os.getcwd(), 'cv_files/AniClassifier.pt')
+model = YOLO(model_path)
+
+
+def img_classifier(image, classifer_type=0):
+
+    test_images = []
+    test_images.append(image)
+    imagesToReturn = []
+
+    # Create a directory for saving classified images
+    folder_dir = './Images'
+    if not os.path.exists(folder_dir):
+        os.makedirs(folder_dir) 
+
+    # Classify images with "good" class in the images folder and save them in the image directory
+    for img in test_images:
+        img_loc = img
+        img_class = model(img_loc, verbose=False)
+
+        # If the first index is higher than the second index, the image is classified as "good"
+        if img_class[0].probs.data[0] < img_class[0].probs.data[1]:
+            
+            # Save the image in the classified directory
+            if classifer_type:
+                image = Image.open(img_loc)
+                image.save(folder_dir + img)
+
+            # Appending Cropped images in an array to display in gradio for end-user
+                imagesToReturn.append(folder_dir + img)
+                return imagesToReturn
+            
+            # Downloading Thumbnail images so don't save them in the image directory
+            else:
+                return True
+
+        else:
+            return False

commands/driver_instance.py

@@ -0,0 +1,72 @@
+from selenium import webdriver
+import urllib.request
+import os
+import logging
+
+def create_driver(profile=0):
+    # Options to make it more human-like
+    options = webdriver.ChromeOptions()
+    options.add_argument("start-maximized")
+    options.add_argument("--disable-blink-features=AutomationControlled")
+    options.add_argument("--disable-notifications")
+    options.add_argument("--disable-popup-blocking")
+    options.add_argument("--disable-extensions")
+    options.add_argument("--disable-gpu")
+    options.add_argument("--disable-infobars")
+    options.add_argument("--disable-dev-shm-usage")
+    options.add_argument("--no-sandbox")
+
+    # May be required to change this later on
+    options.add_argument("user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/118.0.0.0 Safari/537.36")
+
+
+    user_data_dir = os.path.abspath(os.getcwd())+"/commands/profile"
+    if profile:
+        options.add_argument(f"user-data-dir={user_data_dir}")
+    prefs = {"credentials_enable_service": False,
+             "profile.password_manager_enabled": False}
+    options.add_experimental_option("prefs", prefs)
+    options.add_argument("--headless")
+
+    # to supress the error messages/logs?
+    options.add_argument("--log-level=3")
+    options.add_experimental_option("excludeSwitches", ["enable-logging"])
+    options.add_experimental_option("excludeSwitches", ["enable-automation"])
+    
+    # Not recommended to change from default value (decides how the page is loaded)
+    options.page_load_strategy = 'normal' 
+
+    # Remove logs from console
+    selenium_logger = logging.getLogger('selenium')
+    selenium_logger.setLevel(logging.ERROR)
+
+    # Create driver instance (using service is not required)
+    driver = webdriver.Chrome(options=options)
+    return driver
+
+def create_url_headers(tempImg, site=0):
+    opener = urllib.request.build_opener()
+    if not site:
+        site = tempImg
+    opener.addheaders = [
+            ('Accept', 'application/json, text/javascript, */*; q=0.01'),
+            ('X-Requested-With', 'XMLHttpRequest'),
+            ('Referer', f'{site}'),
+            ('Host', f'https//{tempImg.split("/")[2]}'),
+            ('Content-Type', 'application/json; charset=UTF-8'),
+            ('Connection', 'keep-alive'),
+            ('user-agent','Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36')
+        ]
+    return opener
+
+def tab_handler(driver, image=0):
+    if image:
+        tempImg = image.get_attribute("href")
+        driver.execute_script("window.open('');")
+        driver.switch_to.window(driver.window_handles[1])
+        driver.get(f"{tempImg}")
+        return driver, tempImg
+
+    driver.close()
+    driver.switch_to.window(driver.window_handles[0])
+    return driver

commands/exec_path.py

@@ -0,0 +1,25 @@
+import os
+
+driver_path = "driver"
+folder_path = "Images"
+driverPath = os.path.join(os.getcwd(), driver_path)
+imagePath = os.path.join(os.getcwd(), folder_path)
+
+try:
+    if len(os.listdir(driverPath)) > 1:
+        raise Exception("Put 1 driver only")
+    elif os.listdir(driverPath)[0] != "driver.exe":
+        os.rename(driverPath+"/"+"driver.exe")
+except:
+    pass
+finally:
+    executable_path = os.path.join(driverPath, 'driver.exe')
+
+
+def imgList(mode=0):
+    if mode==0: # Danbooru
+        return [image.split(" ")[-1].split(".")[0] for image in os.listdir(imagePath) if image.split(".")[-1] in ["jpg","png","jpeg"]]
+    if mode==1: # Pixiv
+        return [image.split("_")[0] for image in os.listdir(imagePath) if image.split(".")[-1] in ["jpg","png","jpeg"]]
+    if mode==2: # Zerochan
+        return os.listdir(imagePath)

commands/universal.py

@@ -0,0 +1,137 @@
+from selenium.webdriver.common.by import By
+from selenium.webdriver.common.keys import Keys
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+import time
+
+def searchQuery(user_search, driver, elem, mode=0, score="", isLoggedIn=True):
+    if isLoggedIn == False:
+        anchors = driver.find_elements(By.XPATH, '//*[@class="sc-93qi7v-2 hbGpVM"]//a')
+        for n_iter,anchor in enumerate(anchors):
+            if anchor.get_attribute("lang") =="en":
+                driver.execute_script("arguments[0].click();", anchor)
+        
+    user_search = user_search.lower()
+    try:
+        WebDriverWait(driver, timeout=5).until(EC.presence_of_element_located((By.XPATH, elem)))
+    except:
+        driver.refresh()
+        WebDriverWait(driver, timeout=20).until(EC.presence_of_element_located((By.XPATH, elem)))
+    search_bar = driver.find_element(By.XPATH, elem)
+
+    try:
+        search_bar.click()
+        if not mode:
+            time.sleep(1.8)
+        driver.execute_script('arguments[0].value=arguments[1]', search_bar, user_search)
+    except:
+        time.sleep(3)
+        driver.execute_script('arguments[0].value=arguments[1]', search_bar, user_search)
+        
+    if not mode:
+        specific_Query(driver=driver, search_bar=search_bar, user_search=user_search)
+    else:
+        search_bar.send_keys(Keys.ARROW_DOWN)
+        time.sleep(2)
+    
+    if mode:
+        search_bar.send_keys(Keys.ARROW_DOWN)
+        time.sleep(1.2)
+    search_bar.send_keys(Keys.ENTER)
+    time.sleep(0.3)
+
+    try:
+        driver.execute_script("arguments[0].value+=arguments[1]", search_bar, score)
+        search_bar.send_keys(Keys.ENTER)
+    except:
+        pass
+
+
+def save_Search(driver, mode=0):
+    try:
+        with open("./commands/url.txt", "r+") as file:
+            lines = file.readlines()
+
+            line_to_modify = mode + 1  # Line number to modify based on mode
+            new_line_content = driver.current_url  # Content to write to the line
+
+            if line_to_modify > len(lines):
+                lines.extend(['\n'] * (line_to_modify - len(lines)))
+            elif lines[line_to_modify - 1].strip() == new_line_content:
+                # If the line already has the same content, no action is needed
+                return
+
+            lines[line_to_modify - 1] = new_line_content + '\n'
+
+            file.seek(0)
+            file.writelines(lines)
+            file.truncate()
+    except:
+        line_to_modify = mode + 1  # Line number to modify based on mode
+        new_line_content = driver.current_url  # Content to write to the line
+
+        with open("./commands/url.txt", "w") as file:
+            lines = ['\n'] * line_to_modify
+            lines[line_to_modify - 1] = new_line_content + '\n'
+            file.writelines(lines)
+
+
+
+def continue_Search(driver, link, mode=0):
+    try:
+        with open("./commands/url.txt", "r") as file:
+            lines = file.readlines()
+
+        line_to_read = mode + 1  # Line number to read based on mode
+
+        if len(lines) >= line_to_read and lines[line_to_read - 1].strip() != '':
+            url = lines[line_to_read - 1].strip()
+            driver.get(url)
+        else:
+            driver.get(link)
+    except:
+        driver.get(link)
+
+
+def specific_Query(driver, search_bar, user_search):
+    user_search = user_search.replace("_"," ").split()
+    try:
+        for user_word in user_search:
+            time.sleep(1.2)
+            driver.execute_script('arguments[0].value=arguments[1]', search_bar, user_word)
+            time.sleep(0.6)
+            search_bar.send_keys(Keys.SPACE)
+            time.sleep(2.2)
+
+            checker = 0
+            tag_pos = 0
+
+            search_bar_queries = driver.find_element(By.XPATH, '//*[@class="sc-1974j38-2 kjWkkt"]').find_elements(By.XPATH, './/*[@class="sc-d98f2c-0"]')
+            for n_iter, query in enumerate(search_bar_queries):
+                query_text = query.find_element(By.XPATH, './/div//div[2]').text.lower()
+
+                for text in user_search:
+                    if query_text.find(text) != -1:
+                        checker += 1
+
+                if checker == len(user_search):
+                    tag_pos = n_iter
+                    break
+                else:
+                    checker = 0
+
+            if checker == len(user_search):
+                for i in range(tag_pos+1):
+                    time.sleep(0.5)
+                    search_bar.send_keys(Keys.ARROW_DOWN)
+                break
+    except:
+        driver.execute_script('arguments[0].value=arguments[1]', search_bar, user_search)
+        search_bar.send_keys(Keys.ENTER)
+
+def contains_works(driver, elem):
+    try:
+        WebDriverWait(driver, timeout=9).until(EC.presence_of_element_located((By.XPATH, elem)))
+        return True
+    except:
+        return False

driver/chromedriver

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c0e4f5a35fad9c8d70c796a02c92f040a2a8e84a85833e0b0ea92a529fb98f48
+size 13768328

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+selenium
+ultralytics==8.0.228

sites/danbooru.py

@@ -0,0 +1,212 @@
+import time
+import urllib.request
+import os
+from random import randint
+from selenium.webdriver.common.by import By
+from selenium.webdriver.common.keys import Keys
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+from commands.driver_instance import create_url_headers, tab_handler
+from commands.exec_path import imgList
+from commands.universal import searchQuery, save_Search, continue_Search, contains_works
+from ai.classifying_ai import img_classifier
+
+def getOrderedDanbooruImages(driver, exec_path, user_search, num_pics, num_pages, filters, bl_tags, inc_tags,  imageControl):
+    global image_locations, bl_tags_list, inc_tags_list, image_names, ai_mode,rating_filters
+    image_names = imgList(mode=0)
+    image_locations = []
+    link = "https://danbooru.donmai.us/"
+
+    if 0 in imageControl:
+        continue_Search(driver, link, mode=1)
+    else:
+        driver.get(link)
+
+    # Rating Filter Creation
+    rating_filters = ["s","e"] 
+    rating_filters = ["q","s","e"] if 4 in filters else []
+
+    # Tag list creation
+    score = 1 if 0 in filters else 0
+    match_type = 1 if 1 in filters else 0
+    r_18 = pg_lenient() 
+    r_18 = pg_strict() if 3 in filters else r_18
+    ai_mode = 1 
+
+    continue_search = 1 if imageControl else 0
+
+    # Replace spaces to make spaces feasible by the user
+    user_search = user_search.replace(" ", "_")
+    score = filter_score(score)
+
+    bl_tags_list = create_filter_tag_list(bl_tags, r_18)
+    inc_tags_list = create_tag_list(inc_tags, match_type) if inc_tags else []
+    
+    if 0 not in imageControl:
+        searchQuery(user_search, driver, '//*[@name="tags"]', mode=1, score=score)
+    
+    if not contains_works(driver, '//*[@class="posts-container gap-2"]'):
+        print("No works found...")
+        return []
+    
+    if ai_mode:
+        WebDriverWait(driver, timeout=11).until(EC.presence_of_element_located((By.XPATH, '//*[@class="popup-menu-content"]')))
+        driver.get(driver.find_element(By.XPATH, '(//*[@class="popup-menu-content"]//li)[6]//a').get_attribute("href"))
+
+    curr_page = driver.current_url
+    while len(image_locations) < num_pics*num_pages:
+        pages_to_search(driver, num_pages, num_pics, exec_path)
+        if curr_page == driver.current_url and len(image_locations) < num_pics*num_pages:
+            print("Reached end of search results")
+            break
+        curr_page = driver.current_url
+    driver.close()
+
+    return image_locations
+
+def filter_score(score):
+    if score:
+        return " order:score"
+    return ""
+
+def pages_to_search(driver, num_pages, num_pics, exec_path):
+    for i in range(num_pages):
+        WebDriverWait(driver, timeout=11).until(EC.presence_of_element_located((By.XPATH, '//*[@class="posts-container gap-2"]')))
+        # Selects the picture grids
+        images = driver.find_element(
+            By.XPATH, '//*[@class="posts-container gap-2"]'
+        ).find_elements(By.CLASS_NAME, "post-preview-link")
+        grid_search(driver, num_pics, images, exec_path, num_pages)
+        save_Search(driver, mode=1)
+        if not valid_page(driver) or len(image_locations) >= num_pics*num_pages:
+            break
+
+def grid_search(driver, num_pics, images, exec_path, num_pages):
+    time.sleep(2)
+    temp_img_len = len(image_locations)
+    for n_iter, image in enumerate(images):
+        if len(image_locations) >= num_pics*num_pages or len(image_locations) - temp_img_len >= num_pics:
+            break
+
+        try:
+            if image.find_element(By.XPATH, ".//img").get_attribute('src').split("/")[-1].split(".")[0].encode("ascii", "ignore").decode("ascii") in image_names:
+                print("\nImage already exists, moving to another image...")
+                continue
+
+            # Has to be checked this way otherwise tags are not visible in headless mode
+            img_tags = driver.find_elements(By.CLASS_NAME, "post-preview")[n_iter].get_attribute('data-tags')
+            img_rating = driver.find_elements(By.CLASS_NAME, "post-preview")[n_iter].get_attribute('data-rating')
+
+            if filter_ratings(img_rating,rating_filters) and filter_tags(bl_tags_list, inc_tags_list, img_tags):
+                
+                
+                if ai_mode:
+                    checker = 0
+                    image_loc = download_image(exec_path=exec_path, driver=driver, image=image)
+                    if img_classifier(image_loc):
+                        print("AI Mode: I approve this image")
+                    else:
+                        print("AI Mode: Skipping this image")
+                        checker = 1
+                    os.remove(image_loc)
+                    if checker:
+                        continue
+
+                driver, tempImg = tab_handler(driver=driver,image=image)
+                WebDriverWait(driver, timeout=15).until(EC.presence_of_element_located((By.XPATH, '//*[@id="post-option-download"]/a')))
+                download_image(exec_path=exec_path, driver=driver)
+                driver = tab_handler(driver=driver)
+
+            else:
+                print("\nFilters did not match/Not an image, moving to another image...")
+
+        except:
+            print("\nI ran into an error, closing the tab and moving on...")
+            if driver.window_handles[-1] != driver.window_handles[0]:
+                driver = tab_handler(driver=driver)
+            time.sleep(randint(0,2) + randint(0,9)/10)
+
+def filter_ratings(img_rating,rating_filters):
+    if img_rating not in rating_filters:
+        return True
+    return False
+
+def filter_tags(bl_tags_list, inc_tags_list, img_tags):
+    # Hashmap of picture's tags for O(1) time searching
+    img_hash = {}
+    for img_tag in img_tags.split(" "):
+        img_hash[img_tag] = 1
+
+    # Included tags (exact match or not exact)
+    if inc_tags_list and inc_tags_list[-1] == 1:
+        inc_tags_list.pop()
+        for tag in inc_tags_list:
+            if not img_hash.get(tag, 0):
+                return False
+    elif inc_tags_list:
+        cond = False
+        for tag in inc_tags_list:
+            if img_hash.get(tag, 0):
+                cond = True
+                break
+        if not cond:
+            return False
+
+    # Note that bl_tags_list is never empty since it filters videos
+    for tag in bl_tags_list:
+        if img_hash.get(tag,0):
+            return False
+    return True
+
+def create_tag_list(inc_tags, match_type):
+    temp_tags = [tag.lstrip().replace(" ","_") for tag in inc_tags.split(",")]
+    if match_type:
+        temp_tags.append(1)
+    return temp_tags
+
+def create_filter_tag_list(bl_tags, r_18):
+    temp_tags = ["animated", "video", "sound"]
+    if bl_tags:
+        temp_tags += [tag.lstrip().replace(" ","_") for tag in bl_tags.split(",")]
+    if r_18:
+        temp_tags += r_18
+    return temp_tags
+
+# Find the next page and ensure it isn't the last page
+def valid_page(driver):
+    cur_url = driver.current_url
+    driver.find_element(By.CLASS_NAME, "paginator-next").click()
+    if cur_url == driver.current_url:
+        return 0
+    return 1
+
+def pg_lenient():
+    return ["sex","penis","vaginal","completely_nude","nude","exposed_boobs","ahegao","cum","no_panties","no_bra", 
+            "nipple_piercing", "anal_fluid","uncensored", "see-through", "pussy", "cunnilingus", "oral", "ass_focus",
+            "anal", "sex_from_behind", "cum_on_clothes", "cum_on_face", "nipple","nipples", "missionary"
+            "fellatio", "rape", "breasts_out","cum_in_pussy", "condom", "dildo", "sex_toy", "cum_in_mouth", "heavy_breathing", "cum_on_tongue"
+            "panties", "panty_pull", "nude_cover", "underwear_only","grabbing_own_breast","ass_grab","censored","areola_slip","areolae","torn_pantyhose","micro_bikini","steaming_body"]
+
+def pg_strict():
+    return pg_lenient() + ["piercings", "cleavage","boobs","thongs","fellatio_gesture", "mosaic_censoring", "ass", "mosaic_censoring", 
+                            "covered_nipples", "thigh_focus", "thighs", "bikini", "swimsuit", "grabbing_another's_breast", "huge_breasts", 
+                            "foot_focus", "licking_foot", "foot_worship", "shirt_lift","clothes_lift", "underwear", "panties_under_pantyhose"]
+
+def download_image(exec_path, driver, image=0):
+    if not image:
+        tempDL = driver.find_element(By.XPATH, '//*[@id="post-option-download"]/a')
+        tempDLAttr = tempDL.get_attribute("href")
+        tempDLName = tempDL.get_attribute("download").encode('ascii', 'ignore').decode('ascii')
+    else:
+        tempDLAttr = image.find_element(By.XPATH, ".//img").get_attribute('src')
+        tempDLName = tempDLAttr.split("/")[-1].encode("ascii", "ignore").decode("ascii")
+    print(f"\n{tempDLAttr.split('?')[0]}")
+    
+    img_loc = f"./{exec_path.folder_path}/{tempDLName}"
+    urllib.request.urlretrieve(
+                    tempDLAttr, f"./{exec_path.folder_path}/{tempDLName}"
+                )
+    if not image:
+        image_locations.append(f"./{exec_path.folder_path}/{tempDLName}")
+        image_names.append(f"{tempDLName.split('.')[0]}")
+    return img_loc

sites/pixiv.py

@@ -0,0 +1,394 @@
+import sys
+sys.path.append("..")
+
+import time
+import urllib.request
+import os
+import re
+from selenium.webdriver.common.action_chains import ActionChains
+from selenium.webdriver.common.by import By
+from selenium.webdriver.common.keys import Keys
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.common.exceptions import TimeoutException
+from datetime import date, datetime
+from random import randint
+from commands.driver_instance import create_url_headers, tab_handler
+from commands.exec_path import imgList
+from commands.universal import searchQuery, save_Search, continue_Search, contains_works
+from ai.classifying_ai import img_classifier
+
+
+def getOrderedPixivImages(driver,exec_path,user_search,num_pics,num_pages,searchTypes,viewRestriction,imageControl,
+                          n_likes,n_bookmarks,n_views, start_date=0,end_date=0, user_name=0, pass_word=0):
+    global image_locations, image_names, ultimatium, ai_mode, prev_search
+    image_names = imgList(mode=1)
+    image_locations = []
+    prev_search = 0
+    link = "https://www.pixiv.net/tags/illustration"
+    success_login = False
+
+    filters = {
+        "likes": 0 if not n_likes else n_likes,
+        "bookmarks": 0 if not n_bookmarks else n_bookmarks,
+        "viewcount": 0 if not n_views else n_views,
+    }
+    searchLimit = {"pagecount": num_pages, "imagecount": num_pics}
+
+    start_date = start_date if date_handler(start_date) else ""
+    end_date = date.today() if not date_handler(end_date) else end_date
+
+    if 1 in imageControl:
+        continue_Search(driver, link, mode=0)
+    else:
+        driver.get(link)
+
+    # Will use those when not logged in
+    bar_search = '//input[@placeholder="Search works"]'
+    li_search = "//h3[contains(text(), 'Works') or contains(text(), 'Illustrations and Manga') or contains(text(), 'Illustrations')]/ancestor::section[1]/div[2]//li"
+    premium_search = "//h3[contains(text(), 'Popular works')]/ancestor::section[1]/div[2]//li"
+    search_param = {
+        "bar_search": bar_search,
+        "li_search": li_search,
+        "premium_search": premium_search,
+    }
+
+    # Check if logged in otherwise log in with credentials
+    try:
+        # Explicit wait to check for favorite button (only appears for logged in users)
+        WebDriverWait(driver, timeout=6).until(
+            EC.presence_of_element_located((By.XPATH, "//button[contains(text(), 'Add to your favorites')]")))
+        
+        if driver.find_elements(By.XPATH, "//button[contains(text(), 'Add to your favorites')]"):
+            success_login = True
+
+        if not success_login:
+            print("Failed! You are not logged in...")
+
+    except:
+        print("Failed! You are not logged in...")
+        pass
+    
+    if 1 not in imageControl:
+        searchQuery(user_search, driver, search_param["bar_search"], isLoggedIn=success_login)
+    time.sleep(2)
+
+    if start_date and not success_login:
+        driver.get(driver.current_url + f"?scd={start_date}&ecd={end_date}")
+        time.sleep(2)
+    elif start_date and success_login:
+        cur_url = driver.current_url.split("?")
+        driver.get(cur_url[0] + f"?scd={start_date}&ecd={end_date}&" + cur_url[1])
+        time.sleep(2)
+
+    premiumSearch = 1 if 0 in searchTypes else 0
+    freemiumSearch = 1 if 1 in searchTypes else 0
+    pg_friendly = 1 if 0 in viewRestriction else 0
+    r_18 = 1 if 1 in viewRestriction else 0
+    ultimatium = 1 if 0 in imageControl else 0
+    order_by_oldest = 1 if 2 in imageControl else 0
+    ai_mode = 1 
+
+    if not contains_works(driver, search_param["li_search"]):
+        print("No works found...")
+        return []
+    
+    if premiumSearch == 1:
+        search_image(driver, exec_path, filters, search_param)
+
+    # Switch to english
+    try:
+        english_span = driver.find_element(By.XPATH, "//span[contains(text(), 'English')]")
+        driver.execute_script("arguments[0].click();", english_span)
+    except:
+        pass
+
+    # Apply filters if logged in
+    if success_login:
+        try:
+            driver.find_element(By.XPATH, "/html/body/div[1]/div[2]/div/div[3]/div/div[5]/nav/a[2]").click()
+            print("Illustrations only")
+            time.sleep(1)
+
+            mode = ""
+            order = ""
+
+            if pg_friendly == 1 and r_18 == 1:
+                print("PG Friendly and r-18")
+            elif pg_friendly == 1:
+                mode = "mode=safe&"
+                print("PG Friendly")
+            elif r_18 == 1:
+                mode = "mode=r18&"
+                print("r-18")
+            if order_by_oldest == 1:
+                order = "order=date&"
+                print("Order by oldest")
+
+            cur_url = driver.current_url.split("?")
+            driver.get(cur_url[0] + f"?{order}{mode}" + cur_url[1])
+        except:
+            pass
+    
+    # Click show all results
+    try:
+        time.sleep(1)
+        show_all_div = driver.find_element(By.XPATH, "//div[contains(text(), 'Show all')]")
+        if show_all_div:
+            driver.find_element(By.XPATH, '//*[@class="sc-d98f2c-0 sc-s46o24-1 dAXqaU"]').click()
+    except:
+        pass
+    
+    prev_search = len(image_locations)
+    curr_page = driver.current_url
+
+    if freemiumSearch:
+        while len(image_locations) < num_pics*num_pages:
+            search_image(driver,exec_path,filters,search_param=search_param,searchLimit=searchLimit)
+            if not valid_page(driver) and len(image_locations) < num_pics*num_pages:
+                print("Reached end of search results")
+                break
+    driver.quit()
+
+    return image_locations
+
+
+def search_image(driver,exec_path,filters,search_param,searchLimit={"pagecount": 1, "imagecount": 99}):
+    # Searches using premium or freemium
+    search_type = awaitPageLoad(driver=driver,searchLimit=searchLimit,search_param=search_param)
+    if search_type == -1:
+        return
+    
+    # The main image searcher
+    for page in range(searchLimit["pagecount"]):
+        
+        temp_img_len = len(image_locations)
+        WebDriverWait(driver, timeout=9).until(
+            EC.presence_of_element_located(
+                (By.XPATH, search_param["li_search"] + "//a")))  
+        images = search_image_type(search_type, driver, search_param=search_param)
+
+        for image in images:
+            time.sleep(2)
+            if len(image_locations) - prev_search >= searchLimit["imagecount"]*searchLimit["pagecount"] or len(image_locations) - temp_img_len >= searchLimit["imagecount"]:
+                break
+            image = image.find_element(By.XPATH, "." + "/" + "/a")
+            imageLink = image.find_elements(By.XPATH, ".//img")
+
+            if image.get_attribute("href").rsplit("/", 1)[-1] not in image_names:
+                if ai_mode == 1 and process_ai_mode(imageLink, image, driver, exec_path):
+                    continue
+
+                try:
+                    if sum(filters.values()) == 0 and len(imageLink): # Dl the image directly from the grid
+                        thumbnailDownloader(imageLink=imageLink, image=image, driver=driver, exec_path=exec_path)
+                    
+                    else: # Dl the image from the image page (opens a new tab)
+                        driver, tempImg = tab_handler(driver=driver, image=image)
+                        WebDriverWait(driver, timeout=11).until(EC.presence_of_element_located((By.XPATH, "//div[@role='presentation']")))
+                        tempDL = driver.find_element(By.XPATH, "//div[@role='presentation']//img")
+
+                        imagePopularity = parseImageData(filters=filters, 
+                                                         Data=driver.find_elements(By.TAG_NAME, "dd"))
+                        time.sleep(1)
+
+                        if filterOptions(filters, imagePopularity=imagePopularity): # Check if image filters are satisfied
+                            tempDLLink = tempDL.get_attribute("src")
+
+                            # Dl the original rez image
+                            if ultimatium:
+                                tempDLLink = tempDLLink.replace("img-master", "img-original"
+                                ).replace("_master1200", "")
+
+                            download_image(imageLink=tempDLLink, exec_path=exec_path, driver=driver)
+                        else:
+                            print("\nImage filters not satisfied...")
+                        driver = tab_handler(driver=driver)
+                        time.sleep(0.3)
+
+                # In case of stale element or any other errors
+                except:
+                    if driver.window_handles[-1] != driver.window_handles[0]:
+                        print("\nI ran into an error, moving on...")
+                        driver = tab_handler(driver=driver)
+                    time.sleep(randint(1, 3) + randint(0, 9) / 10)
+                    continue
+
+            else:
+                print("\nImage already exists, moving to another image...")
+            save_Search(driver, mode=0)
+        if not valid_page(driver):
+            break
+
+
+######## FUNCTIONS PRONE TO CHANGE ########
+def login_handler(driver, exec_path, user_name, pass_word):
+    time.sleep(5)
+    login_btn = driver.find_elements(By.XPATH, "//*[@class='sc-oh3a2p-4 gHKmNu']//a")[1]
+    login_btn.click()
+
+    WebDriverWait(driver, timeout=11).until(
+        EC.presence_of_element_located((By.XPATH, "//*[@class='sc-2o1uwj-0 elngKN']"))
+    )
+    user_btn = driver.find_element(
+        By.XPATH, "//*[@class='sc-2o1uwj-0 elngKN']"
+    ).find_elements(By.TAG_NAME, "fieldset")
+    user_btn[0]
+
+    actions = ActionChains(driver)
+    actions.click(user_btn[0]).send_keys(user_name).perform()
+    time.sleep(0.5)
+    actions.click(user_btn[1]).send_keys(pass_word).perform()
+
+    # Log in button
+    driver.find_element(By.XPATH,"//button[contains(text(), 'Log In')]").click()
+
+    return True
+
+
+def download_image(imageLink, exec_path, driver, mode=1):
+    tempDLName = imageLink.rsplit("/", 1)[-1]
+    img_loc = f"./{exec_path.folder_path}/{tempDLName}"
+    if not ultimatium or not mode:
+        installUrlOpeners(driver=driver,mode=0)
+    else:
+        installUrlOpeners(imageLink)
+    try:
+        requestUrlretrieve(imageLink=imageLink, img_loc=img_loc)
+    except:
+        imageLink = imageLink.rsplit(".",1)[0]+".png"
+        requestUrlretrieve(imageLink, img_loc=img_loc)
+    
+    print(f"\n{imageLink}")
+    if mode:
+        image_locations.append(f"./{exec_path.folder_path}/{tempDLName}")
+        image_names.append(f"{tempDLName.split('.')[0]}")
+    else:
+        return img_loc
+
+
+def thumbnailDownloader(imageLink, image, driver, exec_path, mode=1):
+    imageLink = image_type(imageLink=imageLink, mode=mode)
+
+    action = ActionChains(driver=driver)
+    action.move_to_element(image.find_element(By.XPATH, ".//img")).perform()
+
+    return download_image(imageLink=imageLink, exec_path=exec_path, driver=driver, mode=mode)
+
+
+######## URLLIB LIBRARY ########
+def installUrlOpeners(driver,mode=1): # Mode 0 means its a thumbnail
+    if ultimatium and mode:
+        urllib.request.install_opener(create_url_headers(driver))
+    else:
+        urllib.request.install_opener(create_url_headers(driver.current_url))
+
+def requestUrlretrieve(imageLink, img_loc): # Download the image
+    urllib.request.urlretrieve(imageLink, img_loc)
+
+
+######## HELPER FUNCTIONS (UNLIKELY TO CHANGE) ########
+# Handles the search type (premium or freemium)
+def search_image_type(search_type, driver, search_param):
+    if search_type == 0:
+        return driver.find_elements(By.XPATH, search_param["premium_search"])
+    elif search_type == 1:
+        return driver.find_elements(By.XPATH, search_param["li_search"])
+
+
+# Handles the image type (if mode then it is not a thumbnail, so switch it to view res else Max res)
+def image_type(imageLink, mode=0):
+    imageLink = imageLink[0].get_attribute("src")
+    if mode: # View res
+        imageLink = re.sub(r"c/.*?/.*?/", "img-master/", imageLink)
+        imageLink = imageLink.replace("square", "master").replace("custom", "master")
+
+        if ultimatium: # Max res
+            imageLink = imageLink.replace("img-master", "img-original").replace("_master1200", "")
+    return imageLink
+
+
+# Handles finding the popular or freemium section
+def awaitPageLoad(driver, searchLimit, search_param, search_type=0):
+    # Waits on the page to load (for popular or freemium)
+    if searchLimit["imagecount"] == 99:
+        try:
+            WebDriverWait(driver, timeout=12).until(
+                EC.presence_of_element_located(
+                    (By.XPATH, search_param["premium_search"])
+                )
+            )
+            print("Premium section found, searching for images...")
+        except:
+            print("No popular section")
+            search_type = -1
+            return search_type
+    else:
+        try:
+            WebDriverWait(driver, timeout=12).until(
+                EC.presence_of_element_located((By.XPATH, search_param["li_search"]))
+            )
+            print("\nFreemium section found, searching for images...")
+        except:
+            driver.refresh()
+            time.sleep(12)
+        if not driver.find_elements(By.XPATH, search_param["li_search"]):
+            return
+        search_type = 1
+    return search_type
+
+
+def filterOptions(filters, imagePopularity):
+    for key in filters.keys():
+        if filters[key] > imagePopularity[key]:
+            return False
+    return True
+
+
+def parseImageData(Data, filters):
+    parsedData = {}
+    for iter, key in enumerate(filters.keys()):
+        parsedData[key] = int(Data[iter].text.replace(",", ""))
+    return parsedData
+
+
+def valid_page(driver):
+    cur_url = driver.current_url
+    try:
+        next_page = (
+            driver.find_element(By.XPATH, '//*[@class="sc-xhhh7v-0 kYtoqc"]')
+            .find_elements(By.XPATH, ".//a")[-1]
+            .get_attribute("href")
+        )
+        if cur_url == next_page:
+            return 0
+        if next_page:
+            driver.get(next_page)
+        return 1
+    except:
+        return 0
+
+
+def date_handler(sel_date):
+    temp = sel_date.split("-")
+    try:
+        datetime(int(temp[0]), int(temp[1]), int(temp[2]))
+    except ValueError:
+        return 0
+    return 1
+
+
+def process_ai_mode(imageLink, image, driver, exec_path):
+    try:
+        # Dl the image thumbnail from the grid
+        img_loc = thumbnailDownloader(imageLink=imageLink, image=image, driver=driver, exec_path=exec_path, mode=0)
+
+        if img_classifier(img_loc):
+            print("AI Mode: I approve this image")
+            return False
+        else:
+            print("AI Mode: Skipping this image")
+            return True
+        os.remove(img_loc)
+    except:
+        return True

sites/yandex.py

@@ -0,0 +1,166 @@
+import time
+import urllib.request
+import os
+import re
+from random import randint
+from selenium.webdriver.common.by import By
+from selenium.webdriver.common.keys import Keys
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+from commands.driver_instance import create_url_headers, tab_handler
+from commands.exec_path import imgList
+from commands.universal import searchQuery, save_Search, continue_Search, contains_works
+from ai.classifying_ai import img_classifier
+
+
+def getOrderedYandexImages(
+    driver, exec_path, user_search, num_pics, filters, imageOrientation):
+    global image_locations, image_names, ai_mode
+    image_names = imgList(mode=2)
+    ai_mode = True
+    recents = True if 1 in filters else False
+
+    image_locations = []
+    link = "https://yandex.com/images/search?isize=large&"
+    link = link + "text=" + user_search.replace(" ", "+").replace("_", "+")
+    if imageOrientation:
+        orientations = ["horizontal", "vertical", "square"]
+        link += f"&iorient={orientations[imageOrientation]}"
+    driver.get(link)
+
+    WebDriverWait(driver, timeout=11).until(
+        EC.presence_of_element_located((By.XPATH, '//*[@class="SerpList"]'))
+    )
+
+    driver.find_element(
+        By.XPATH,
+        "//*[@class='SimpleImage SimpleImage_showPlaceholderIcon SerpItem-Thumb']//a",
+    ).click()
+
+    grid_search(driver, num_pics, exec_path, user_search)
+
+    time.sleep(20)
+    driver.close()
+
+    return image_locations
+
+
+def grid_search(driver, num_pics, exec_path, user_search):
+    WebDriverWait(driver, timeout=11).until(
+        EC.presence_of_element_located(
+            (By.XPATH, "//*[contains(@class, 'MMGallery-Item')]")
+        )
+    )
+    images = driver.find_elements(By.XPATH, "//*[@class='MMGallery-Container']/*")
+
+    for image in images:
+        time.sleep(2)
+        # Navigate the webpage and filter the image link 
+        try:
+            if len(image_locations) >= num_pics:
+                break
+
+            driver.execute_script("arguments[0].click();", image)
+            time.sleep(0.5)
+            imageLink = driver.find_element(
+                By.XPATH,
+                '//*[@class="OpenImageButton OpenImageButton_text OpenImageButton_sizes MMViewerButtons-OpenImageSizes"]//a',
+            ).get_attribute("href")
+
+            if (imageLink.rsplit("/",1)[-1].encode("ascii", "ignore")
+                .decode("ascii")) in image_names:
+                print("Image already exists, moving to another image...\n")
+                continue
+        except:
+            print("I ran into an error finding the image, closing the tab and moving on...\n")
+            time.sleep(randint(0, 1) + randint(0, 9) / 10)
+            continue
+        
+        # Ai mode check
+        try:
+            if ai_mode:
+                checker = ai_dl(image, exec_path, driver, user_search)
+                if checker:
+                    continue
+        except Exception as e: # TODO: Implement proper exception handling in case of http error 404
+            time.sleep(randint(0, 1) + randint(0, 9) / 10)
+            checker = ai_dl(image, exec_path, driver, user_search, site="https://yandex.com/")
+        except:
+            time.sleep(randint(0, 1) + randint(0, 9) / 10)
+            print("AI mode failed to check the image, skipping...\n")
+            continue
+        
+        # Download the image
+        try:
+            download_image(
+                exec_path=exec_path,
+                driver=driver,
+                image=imageLink,
+                user_search=user_search,
+            )
+        except: # TODO: Use the same exception handling as above to try and redownload the image
+            print("I ran into an error downloading, closing the tab and moving on...\n")
+            time.sleep(randint(0, 1) + randint(0, 9) / 10)
+
+
+def download_image(exec_path, driver, image, user_search, mode=1, site=0):
+    tempDLAttr = image
+    matching = re.search(r"([^/]+\.(?:jpg|jpeg|png|webp))", image.rsplit("/", 1)[-1])
+
+    if not matching:
+        tempDLAttr += ".png"
+    if tempDLAttr.startswith('//'):
+        tempDLAttr = 'https:' + tempDLAttr
+
+    tempDLName = (
+    re.search(r"([^/]+\.(?:jpg|jpeg|png|webp))", tempDLAttr.rsplit("/", 1)[-1])
+    .group(1)
+    .encode("ascii", "ignore")
+    .decode("ascii"))
+    
+    if not mode:
+        tempDLName = re.sub(r'[\\/*?:"<>|]', "", tempDLName)
+    img_loc = f"./{exec_path.folder_path}/{tempDLName}"
+    
+    # User other site headers (to be implemented properly)
+    if site:
+        urllib.request.install_opener(create_url_headers(tempDLAttr, site=site))
+        urllib.request.urlretrieve(tempDLAttr, img_loc)
+
+    urllib.request.install_opener(create_url_headers(tempDLAttr))
+    urllib.request.urlretrieve(tempDLAttr, img_loc)
+
+    if mode:
+        print(f"{tempDLAttr}\n")
+        image_locations.append(img_loc)
+        image_names.append(f"{tempDLName.split('.')[0]}")
+    return img_loc
+
+
+def ai_dl(image, exec_path, driver, user_search, site=""):
+    checker = 0
+    image_thumbnail = image.find_element(
+        By.XPATH, './/*[@class="MMThumbImage-Image"]'
+    ).get_attribute("style")
+
+    # Filter url from the style attribute
+    image_thumbnail = re.findall(r'url\("(.+?)"\)', image_thumbnail)[0]
+
+    # Download the image thumbnail
+    image_loc = download_image(
+        exec_path=exec_path,
+        driver=driver,
+        image=image_thumbnail,
+        user_search=user_search,
+        mode=0,
+        site=site
+    )
+
+    # Check if the image is good or not and delete the image
+    if img_classifier(image_loc):
+        print("AI Mode: I approve this image")
+    else:
+        print("AI Mode: Skipping this image\n")
+        checker = 1
+    os.remove(image_loc)
+    return checker

sites/zerochan.py

@@ -0,0 +1,128 @@
+import time
+import urllib.request
+import os
+from random import randint
+from selenium.webdriver.common.by import By
+from selenium.webdriver.common.keys import Keys
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.common.exceptions import TimeoutException
+from commands.driver_instance import create_url_headers, tab_handler
+from commands.exec_path import imgList
+from commands.universal import contains_works, save_Search, continue_Search
+from ai.classifying_ai import img_classifier
+
+def getOrderedZerochanImages(driver, exec_path, user_search, num_pics, num_pages, n_likes, filters, imageControl):
+    global image_locations, image_names, ultimatium, ai_mode
+    image_names = imgList(mode=1)
+    image_locations = []
+
+    ai_mode = 1 
+    filters={'likes': 0 if not n_likes else n_likes}
+    searchLimit={'pagecount': num_pages,'imagecount':num_pics}
+    user_search = user_search.replace(" ","+").capitalize()
+    link = "https://www.zerochan.net/" + user_search
+
+    if not imageControl:
+        driver.get(link)
+    if imageControl:
+        continue_Search(driver, link, mode=2)
+
+    if driver.current_url == "https://www.zerochan.net/":
+        print("You continued for the first time, but there was no previous search to continue from!")
+        driver.get(driver.current_url + 'angry')
+
+    
+    is_valid_search(driver)
+    if not contains_works(driver, '//*[@id="thumbs2"]'):
+        print("No works found...")
+        return []
+
+    curr_page = driver.current_url
+    while len(image_locations) < num_pics*num_pages:
+        search_image(driver,exec_path,filters,searchLimit=searchLimit)
+        if curr_page == driver.current_url and len(image_locations) < num_pics*num_pages or image_locations[-1]==-1:
+            image_locations.pop()
+            print("Reached end of search results")
+            break
+        curr_page = driver.current_url
+    driver.quit()
+
+    return image_locations
+
+def search_image(driver, exec_path, filters, searchLimit):
+    filter_link = "https://www.zerochan.net/register"
+
+    # The main image searcher
+    for page in range(searchLimit["pagecount"]):
+        temp_img_len = len(image_locations)
+        save_Search(driver=driver, mode=2)
+        WebDriverWait(driver, timeout=11).until(EC.presence_of_element_located((By.XPATH, "//*[@id='thumbs2']")))
+        images = driver.find_elements(By.XPATH, "//*[@id='thumbs2']//li")
+        if image_locations and image_locations[-1] == -1:
+            break
+
+        for curr_iter,image in enumerate(images):
+            tempImg = image.find_element(By.XPATH,".//a").get_attribute("href")
+            if len(image_locations) >= searchLimit['imagecount']*searchLimit['pagecount'] or len(image_locations) - temp_img_len >= searchLimit['imagecount']:
+                break
+            try:
+                tempDLLink = image.find_elements(By.XPATH, ".//p//a")[0].get_attribute("href")
+                if tempDLLink.split(".")[-1] not in ["jpg","png","jpeg"]:
+                    tempDLLink = image.find_elements(By.XPATH, ".//p//a")[1].get_attribute("href")
+                tempDLAttr = tempDLLink.split("/")[-1]
+                counts = tempDLAttr.count(".")-1
+                tempDLAttr = tempDLAttr.replace(".", " ", counts).encode("ascii", "ignore").decode("ascii")
+
+                if tempImg == filter_link or tempDLAttr in image_names:
+                    print("\nImage already exists, moving to another image...")
+                    continue
+
+                rand_time = randint(0,1) + randint(0,9)/10
+                time.sleep(rand_time)     
+                if int(image.find_element(By.XPATH, './/*[@class="fav"]').get_property("text"))>filters["likes"]:
+                    urllib.request.install_opener(create_url_headers(tempImg=tempImg))
+                    urllib.request.urlretrieve(
+                        tempDLLink, f"./{exec_path.folder_path}/{tempDLAttr}"
+                    )
+                    image_locations.append(f"./{exec_path.folder_path}/{tempDLAttr}")
+                    image_names.append(f"{tempDLAttr}")
+                    print(f"\n{tempDLLink}")
+                    if ai_mode:
+                        if img_classifier(image_locations[-1]):
+                            print("AI Mode: I approve this image")
+                        else:
+                            os.remove(image_locations[-1])
+                            image_locations.pop()
+                            image_names.pop()
+                            print("AI Mode: Skipping this image")
+
+                else:
+                    image_locations.append(-1)
+
+            # In case of stale element or any other errors
+            except:
+                if driver.window_handles[-1] != driver.window_handles[0]:
+                    print("I ran into an error, closing the tab and moving on...")
+                    driver = tab_handler(driver=driver)
+                time.sleep(randint(1,3) + randint(0,9)/10)
+                continue
+
+            
+        if not valid_page(driver):
+             break
+
+def valid_page(driver):
+    try:
+        driver.get(driver.find_elements(By.XPATH, "//*[@class='pagination']//a")[-1].get_attribute("href"))
+        return True
+    except:
+        return False
+    
+def is_valid_search(driver):
+    try:
+        titles = driver.find_element(By.XPATH, "//*[@id='children']//a").get_attribute("href")
+        if titles:
+            driver.get(titles)
+    except:
+        pass

style.css

@@ -0,0 +1,251 @@
+@import url("https://fonts.googleapis.com/css2?family=Lexend&display=swap");
+
+:root {
+  --font: "Lexend", sans-serif !important;
+  --dark-purple: #242038 !important;
+  --button-secondary-background-fill: #c5b0fc !important;
+  --button-secondary-background-fill-hover: #a288e3 !important;
+  --button-secondary-border-color: var(--dark-purple) !important;
+  --border-color-primary: #9067c6 !important;
+
+  --slider-color: #c5b0fc !important;
+  --block-title-text-color: #fafaff !important;
+  --input-background-fill: var(--dark-purple) !important;
+  --neutral-800: #fafaff !important;
+
+  --checkbox-label-background-fill: var(--dark-purple) !important;
+  --checkbox-label-background-fill-hover: var(--dark-purple) !important;
+  --checkbox-background-color: #a288e3 !important;
+  --checkbox-border-color: #a288e3 !important;
+  --checkbox-background-color-hover: #c5b0fc !important;
+  --checkbox-background-color-selected: var(--dark-purple) !important;
+  --checkbox-border-color-focus: var(--dark-purple) !important;
+  --checkbox-border-color-selected: #c5b0fc !important;
+  --shadow-spread: 3px !important;
+  --shadow-inset: rgba(0, 0, 0, 0.05) 0px 2px 4px 0px inset !important;
+  --shadow-drop: 0px !important;
+  --input-shadow-focus: 0 0 0 var(--shadow-spread) #c5b0fc, var(--shadow-inset) !important;
+  --input-border-color-focus: #c5b0fc !important;
+
+  --background-fill-primary: var(--dark-purple) !important;
+
+  --block-background-fill: var(--dark-purple) !important;
+}
+
+@-webkit-keyframes popUp {
+  from {
+    opacity: 0;
+  }
+  to {
+    opacity: 1;
+  }
+}
+
+@keyframes popUp {
+  from {
+    opacity: 0;
+  }
+  to {
+    opacity: 1;
+  }
+}
+
+.dark {
+  --font: "Lexend", sans-serif !important;
+  --dark-purple: #242038 !important;
+  --button-secondary-background-fill: #c5b0fc !important;
+  --button-secondary-background-fill-hover: #a288e3 !important;
+  --button-secondary-border-color: var(--dark-purple) !important;
+  --border-color-primary: #9067c6 !important;
+
+  --slider-color: #c5b0fc !important;
+  --block-title-text-color: #fafaff !important;
+  --input-background-fill: var(--dark-purple) !important;
+  --neutral-800: #fafaff !important;
+
+  --checkbox-label-background-fill: var(--dark-purple) !important;
+  --checkbox-label-background-fill-hover: var(--dark-purple) !important;
+  --checkbox-background-color: #a288e3 !important;
+  --checkbox-border-color: #a288e3 !important;
+  --checkbox-background-color-hover: #c5b0fc !important;
+  --checkbox-background-color-selected: var(--dark-purple) !important;
+  --checkbox-border-color-focus: var(--dark-purple) !important;
+  --checkbox-border-color-selected: #c5b0fc !important;
+  --shadow-spread: 3px !important;
+  --shadow-inset: rgba(0, 0, 0, 0.05) 0px 2px 4px 0px inset !important;
+  --shadow-drop: 0px !important;
+  --input-shadow-focus: 0 0 0 var(--shadow-spread) #c5b0fc, var(--shadow-inset) !important;
+  --input-border-color-focus: #c5b0fc !important;
+
+  --background-fill-primary: var(--dark-purple) !important;
+  --block-background-fill: var(--dark-purple) !important;
+}
+
+.gradio-container {
+  background-color: var(--dark-purple) !important;
+}
+
+.svelte-vt1mxs {
+  background-color: var(--dark-purple) !important;
+}
+
+.block.svelte-mppz8v {
+  background-color: var(--dark-purple) !important;
+}
+
+.selected.svelte-1g805jl {
+  background: var(--border-color-primary) !important;
+}
+
+.svelte-1g805jl button:focus {
+  background: var(--dark-purple) !important;
+}
+.svelte-1g805jl button {
+  background: #211a1d !important;
+}
+.svelte-1g805jl button:hover {
+  color: #ffff !important;
+}
+
+[class*="form"][class*="svelte"] {
+  border: none !important;
+  border-radius: 0px !important;
+}
+
+#filtering > span,
+#pixiv > span,
+#pixiv-filters > span,
+#imageControl > span,
+#zeroAIhover > span {
+  position: relative;
+  z-index: 0;
+}
+
+#filtering [data-testid="checkbox-group"] > label > span,
+#pixiv [data-testid="checkbox-group"] > label > span,
+#pixiv-filters [data-testid="checkbox-group"] > label > span,
+#imageControl [data-testid="checkbox-group"] > label > span,
+#zeroAIhover [data-testid="checkbox-group"] > label > span {
+  position: relative;
+}
+
+#filtering [data-testid="checkbox-group"] > label > span::after,
+#pixiv [data-testid="checkbox-group"] > label > span::after,
+#pixiv-filters [data-testid="checkbox-group"] > label > span::after,
+#imageControl [data-testid="checkbox-group"] > label > span::after,
+#zeroAIhover [data-testid="checkbox-group"] > label > span::after {
+  position: absolute;
+  z-index: 1;
+  visibility: hidden;
+  background-color: #c5b0fc;
+  color: #fff;
+  padding: 5px;
+  border-radius: 5px;
+  font-size: 12px;
+  bottom: 150%;
+  left: 50%;
+  transform: translateX(-50%);
+  white-space: nowrap;
+}
+
+#filtering [data-testid="checkbox-group"] > label > span::before,
+#pixiv [data-testid="checkbox-group"] > label > span::before,
+#pixiv-filters [data-testid="checkbox-group"] > label > span::before,
+#imageControl [data-testid="checkbox-group"] > label > span::before,
+#zeroAIhover [data-testid="checkbox-group"] > label > span::before {
+  content: "";
+  position: absolute;
+  visibility: hidden;
+  bottom: 105%;
+  left: 50%;
+  margin-left: -5px;
+  border-width: 5px;
+  border-style: solid;
+  border-color: #c5b0fc transparent transparent transparent;
+}
+
+#filtering [data-testid="checkbox-group"] > label:nth-child(6) > span::after,
+#zeroAIhover [data-testid="checkbox-group"] > label:nth-child(1) > span::after,
+#pixiv-filters [data-testid="checkbox-group"] > label:nth-child(4) > span::after {
+  content: "AI to download 'good' images";
+}
+
+#filtering [data-testid="checkbox-group"] > label:nth-child(5) > span::after {
+content: "General (SFW) images only";
+}
+
+#filtering [data-testid="checkbox-group"] > label:nth-child(4) > span::after {
+  content: "Allows questionable images";
+}
+
+#filtering [data-testid="checkbox-group"] > label:nth-child(3) > span::after {
+  content: "Removes explicit images only";
+}
+
+#filtering [data-testid="checkbox-group"] > label:nth-child(2) > span::after {
+  content: "Images must include all tags given";
+}
+
+#filtering [data-testid="checkbox-group"] > label:nth-child(1) > span::after {
+  content: "Order by score";
+}
+
+#filtering [data-testid="checkbox-group"] > label > span:hover::after,
+#filtering [data-testid="checkbox-group"] > label > span:hover::before,
+#pixiv [data-testid="checkbox-group"] > label > span:hover::after,
+#pixiv [data-testid="checkbox-group"] > label > span:hover::before,
+#pixiv-filters [data-testid="checkbox-group"] > label > span:hover::after,
+#pixiv-filters [data-testid="checkbox-group"] > label > span:hover::before,
+#imageControl [data-testid="checkbox-group"] > label > span:hover::after,
+#imageControl [data-testid="checkbox-group"] > label > span:hover::before,
+#zeroAIhover [data-testid="checkbox-group"] > label > span:hover::after,
+#zeroAIhover [data-testid="checkbox-group"] > label > span:hover::before {
+  visibility: visible;
+  -webkit-animation: popUp 500ms;
+  animation: popUp 500ms;
+}
+
+#filtering [data-testid="checkbox-group"] > label:nth-child(3):hover > span {
+  color: #e71d36;
+}
+#filtering [data-testid="checkbox-group"] > label:nth-child(4):hover > span {
+  color: #ffd151;
+}
+#filtering [data-testid="checkbox-group"] > label:nth-child(5):hover > span {
+  color: #21fa90;
+}
+
+#pixiv [data-testid="checkbox-group"] > label:nth-child(1) > span::after {
+  content: "Downloads the previews";
+}
+
+#pixiv [data-testid="checkbox-group"] > label:nth-child(2) > span::after {
+  content: "Downloads through the grid";
+}
+
+#pixiv-filters [data-testid="checkbox-group"] > label:nth-child(1) > span::after {
+  content: "Downloads native image resolution";
+  font-size: 0.5rem;
+}
+#pixiv-filters [data-testid="checkbox-group"] > label:nth-child(2) > span::after,
+#imageControl [data-testid="checkbox-group"] > label:nth-child(1) > span::after {
+  content: "Continues Search if hangup occurs";
+}
+
+#pixiv-filters [data-testid="checkbox-group"] > label:nth-child(3) > span::after {
+  content: "Starts from the oldest images";
+}
+
+#viewing-restrictions [data-testid="checkbox-group"] > label:nth-child(1):hover span {
+  color: #21fa90;
+}
+
+#viewing-restrictions [data-testid="checkbox-group"] > label:nth-child(2):hover span {
+  color: #e71d36;
+}
+
+#button-row {
+  display: flex;
+  justify-content: center;
+  align-items: center;
+}