Spaces:
Running
Running
| import gradio as gr | |
| import os | |
| # os.environ["CUDA_VISIBLE_DEVICES"] = "-1" | |
| from categories import categories, font_list | |
| import pdb | |
| from PIL import Image | |
| import random | |
| try: | |
| import pygsheets | |
| except Exception as e: | |
| print("pygsheets not found", e) | |
| os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:128" | |
| global_index = 0 | |
| mode_global = "DS-Fusion-Express" | |
| prompt_global = "" | |
| def log_data_to_sheet(pre_style, style_custom, glyph, attribute): | |
| try: | |
| #authorization | |
| gc = pygsheets.authorize(service_file='./huggingface-connector-02adcb4cdf00.json') | |
| #open the google spreadsheet (where 'PY to Gsheet Test' is the name of my sheet) | |
| sh = gc.open('HuggingFace Logs') | |
| #select the first sheet | |
| wks = sh[0] | |
| # Send fields list | |
| wks.append_table(values=[pre_style, style_custom, glyph, attribute]) | |
| except: | |
| pass | |
| def change_font(evt: gr.SelectData): | |
| global global_index | |
| global_index = evt.index | |
| def my_main(pre_style, style_custom, glyph, attribute): | |
| log_data_to_sheet(pre_style, style_custom, glyph, attribute) | |
| global prompt_global | |
| glyph = glyph[0] | |
| command = "rm -r out_cur/*" | |
| os.system(command) | |
| for i in range(1,5): | |
| command = "cp initial_show/" + str(i) +".png out_cur/"+str(i)+".png" | |
| os.system(command) | |
| style = pre_style | |
| command = "rm -r data_style/" | |
| os.system(command) | |
| if style_custom != "": | |
| style = style_custom | |
| if len(glyph) != 1: | |
| prompt_global = f" {style}" | |
| else: | |
| prompt_global = f" {style} {glyph}" | |
| li = "ckpt/all2.ckpt" | |
| output_path = f"out/express" | |
| if attribute == "": | |
| prompt = f" '{style} {glyph}'" | |
| else: | |
| prompt = f" '{attribute} {style} {glyph}'" | |
| command = "rm -r out/" | |
| os.system(command) | |
| print(prompt) | |
| command = "python txt2img.py --ddim_eta 1.0 \ | |
| --n_samples 4 \ | |
| --n_iter 1\ | |
| --ddim_steps 50 \ | |
| --scale 5.0\ | |
| --H 256\ | |
| --W 256\ | |
| --outdir " + output_path + " --ckpt " +li +" --prompt " + prompt | |
| os.system(command) | |
| command = "rm -r out_cur/*" | |
| os.system(command) | |
| path = [] | |
| final_imgs = os.listdir(output_path+"/samples") | |
| for i in range(4): | |
| path.append(os.path.join(output_path+"/samples", final_imgs[i])) | |
| path_in = os.path.join(output_path+"/samples", final_imgs[i]) | |
| command = "cp " + path_in + " " + "out_cur/"+final_imgs[i] | |
| os.system(command) | |
| return gr.update(value=path) | |
| def rem_bg(): | |
| command= "rm -r out_bg/*" | |
| os.system(command) | |
| files = os.listdir("out_cur") | |
| if len(files)>0: | |
| command_3 = f"python script_step3.py --input_dir out_cur --method 'rembg'" | |
| os.system(command_3) | |
| for file in files: | |
| command = "cp out_bg/"+file +" out_cur/" | |
| os.system(command) | |
| path = [] | |
| for file in files: | |
| file_path = os.path.join("out_cur", file) | |
| image = Image.open(file_path) | |
| new_image = Image.new("RGBA", image.size, "WHITE") | |
| new_image.paste(image, (0, 0), image) | |
| new_image.save(file_path, "PNG") | |
| path.append(file_path) | |
| return gr.update(value = path) | |
| font_list_express = [ | |
| 'Caladea-Regular', #works good | |
| # 'Garuda-Bold', #works poorlyß | |
| 'FreeSansOblique', #works average | |
| "Purisa", #works good | |
| "Uroob" #worksaverage | |
| ] | |
| path_fonts_express = [] | |
| for font in font_list_express: | |
| path_in = "font_list/fonts/"+font+".png" | |
| path_fonts_express.append(path_in) | |
| def make_upper(value): | |
| if value == "": | |
| return "" | |
| return value[0].upper() | |
| def get_out_cur(): | |
| path = [] | |
| pth = "log_view" | |
| for file in os.listdir(pth): | |
| file_final = os.path.join(pth, file) | |
| path.append(file_final) | |
| return gr.update(value=path) | |
| def update_time_mode(value): | |
| if value == 'DS-Fusion': | |
| return gr.update(value="Generation Time: ~5 mins") | |
| else: | |
| return gr.update(value="Generation Time: ~30 seconds") | |
| def update_time_cb(value): | |
| if value: | |
| if mode_global == "DS-Fusion": | |
| return gr.update(value="Generation Time: ~8 mins") | |
| return gr.update(value="Generation Time: ~30 seconds") | |
| else: | |
| if mode_global == "DS-Fusion": | |
| return gr.update(value="Generation Time: ~5 mins") | |
| return gr.update(value="Generation Time: ~30 seconds") | |
| def load_img(): | |
| path = [] | |
| dir = "out_cur" | |
| for file in os.listdir(dir): | |
| file_full = os.path.join(dir, file) | |
| path.append(file_full) | |
| return path | |
| css = ''' | |
| <!-- this makes the items inside the gallery to shrink --> | |
| div#gallery_64 div.grid { | |
| height: 64px; | |
| width: 180px; | |
| } | |
| <!-- this makes the gallery's height to shrink --> | |
| div#gallery_64 > div:nth-child(3) { | |
| min-height: 172px !important; | |
| } | |
| <!-- this makes the gallery's height to shrink when you click one image to view it bigger --> | |
| div#gallery_64 > div:nth-child(4) { | |
| min-height: 172px !important; | |
| } | |
| ''' | |
| with gr.Blocks(css=css) as demo: | |
| with gr.Column(): | |
| with gr.Row(): | |
| with gr.Column(): | |
| with gr.Row(): | |
| in4 = gr.Text(label="Character (A-Z, 0-9) to Stylize", info = "Only works with capitals. Will pick first letter if more than one", value = "R", interactive = True) | |
| in2 = gr.Dropdown(categories, label="Pre-Defined Style Categories", info = "Categories used to train Express", value = "DRAGON", interactive = True) | |
| with gr.Row(): | |
| in3 = gr.Text(label="Override Style Category ", info="This will replace the pre-defined style value", value = "", interactive = True) | |
| in5 = gr.Text(label="Additional Style Attribute ",info= "e.g. pixel, grayscale, etc", value = "", interactive = True) | |
| # with gr.Row(): | |
| # with gr.Column(): | |
| # in8 = gr.Checkbox(label="MULTI FONT INPUT - font selection below is over-ridden", info="Select for more abstract results", value = False, interactive = True).style(container=True) | |
| # gallery = gr.Gallery([], label="Select Font", show_label=True, elem_id="gallery_64").style(grid=[2,6], preview=True, height="auto") | |
| with gr.Row(): | |
| btn = gr.Button("Let's Stylize It - Generation Time: ~60 seconds", interactive = True) | |
| # btn_bg = gr.Button("Remove Background", interactive = True) | |
| with gr.Column(): | |
| gallery_out = gr.Gallery(label="Generated images", elem_id="gallery_out").style(grid=[2,2], height="full") | |
| outputs = [gallery_out] | |
| # gallery.select(change_font, None, None) | |
| inputs = [in2,in3,in4,in5] | |
| btn.click(my_main, inputs, outputs) | |
| # btn_bg.click(rem_bg, None, outputs) | |
| if __name__ == "__main__": | |
| command = "rm -r out_cur/*" | |
| os.system(command) | |
| for i in range(1,5): | |
| command = "cp initial_show/" + str(i) +".png out_cur/"+str(i)+".png" | |
| os.system(command) | |
| demo.queue() | |
| demo.launch(share=False) | |