Fsg-Pp / Fsg_pp.py
Kyo-Kai's picture
Upload 18 files
f4d52c1 verified
raw history blame
No virus
13.2 kB
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))