artist / app.py
jin-nin's picture
Update app.py
81694d1
raw
history blame
5.16 kB
import gradio as gr
import os
import sys
from pathlib import Path
import random
import string
import time
from queue import Queue
from threading import Thread
import emoji
def pipe(val):
return val
ru2en = gr.Interface.load("huggingface/Helsinki-NLP/opus-mt-ru-en")
text_gen=gr.Interface.load("spaces/phenomenon1981/MagicPrompt-Stable-Diffusion")
def get_prompts(prompt_text):
return text_gen("photo, " + prompt_text + ", high details, harmony, ideal proportions")
proc1=gr.Interface.load("models/dreamlike-art/dreamlike-photoreal-2.0")
def restart_script_periodically():
while True:
time.sleep(600) # 10 minutes
try:
os.execl(sys.executable, sys.executable, *sys.argv)
except:
pass
restart_thread = Thread(target=restart_script_periodically, daemon=True)
restart_thread.start()
queue = Queue()
queue_threshold = 800
def add_random_noise(prompt, noise_level=0.00):
if noise_level == 0:
noise_level = 0.00
# Get the percentage of characters to add as noise
percentage_noise = noise_level * 5
# Get the number of characters to add as noise
num_noise_chars = int(len(prompt) * (percentage_noise/100))
# Get the indices of the characters to add noise to
noise_indices = random.sample(range(len(prompt)), num_noise_chars)
# Add noise to the selected characters
prompt_list = list(prompt)
# Add numbers, special characters, and all emojis to the list of characters used to add noise
noise_chars = string.ascii_letters + string.punctuation + ' ' + string.digits + emoji.emojize(":all:")
for index in noise_indices:
prompt_list[index] = random.choice(noise_chars)
return "".join(prompt_list)
def send_it(inputs, noise_level, proc1=proc1):
prompt_with_noise = add_random_noise(inputs, noise_level)
output = proc1(prompt_with_noise)
return output
with gr.Blocks(analytics_enabled=False, css='style.css') as demo:
with gr.Column(elem_id="col-container"):
with gr.Row(elem_id="shorts"):
input_text_ru = gr.Textbox(
label="Short Ru",
placeholder="девушка",
)
Translate = gr.Button("Ru ▶ En", elem_id="translate").style(full_width=False)
input_text = gr.Textbox(
label="Short En",
placeholder="girl",
)
with gr.Row(elem_id="longs"):
with gr.Column(scale=1000):
prompt = gr.Textbox(
label="Prompt",
placeholder="proto, girl, high details",
)
with gr.Column(scale=1, elem_id="longs-fillers"):
Literally = gr.Button("◀ Literally")
Explain = gr.Button("◀ Explain")
with gr.Row(elem_id="params"):
with gr.Column():
noise_level = gr.Slider(minimum=0.0, value=1.0, maximum=3, step=0.1, label="Noise Level")
with gr.Row(elem_id="paints"):
with gr.Column():
run1 = gr.Button("🔻 Paint")
output1=gr.Image()
with gr.Column():
run2 = gr.Button("🔻 Paint")
output2=gr.Image()
with gr.Column():
run3 = gr.Button("🔻 Paint")
output3=gr.Image()
# with gr.Column():
# run4 = gr.Button("🔻 Paint")
# output4=gr.Image()
gr.Markdown("Honor **Artists**: Yayoi Kusama, Pablo Picasso, Leonardo da Vinci, Banksy, Rembrandt, Frida Kahlo, Vincent van Gogh, Henri Matisse, Salvador Dali, Claude Monet, Andy Warhol, Georgia O'Keeffe, Jackson Pollock, Marcel Duchamp, Edward Hopper, Willem de Kooning, Mark Rothko, David Hockney")
gr.Markdown("Popuar **Styles**: Anime, Abstract, Minimalist, Cyberpunk, Steampunk, Organic, Geometric, Sci-Fi, Futuristic, Vaporwave, Gothic")
gr.Markdown("Known **Moods**: joyful, light-hearted, exciting, calming, soothing, playful, fun, bright, colourful, dynamic, energetic, passionate, romantic, vibrant, vivid")
gr.Markdown("Typical **Techniques**: oil on canvas, chinese painting, graffiti, watercolour, graphite, cinematic, film noir, fluorescent, moody lighting, silhouette, ultraviolet, x-ray, olaroid, double exposure, fisheye lens, bokeh")
Literally.click(pipe, inputs=[input_text], outputs=[prompt], queue=False)
Translate.click(ru2en, inputs=[input_text_ru], outputs=[input_text], queue=False, api_name="ru2en")
Explain.click(text_gen, inputs=[input_text], outputs=[prompt], queue=False, api_name="explain")
run1.click(send_it, inputs=[prompt, noise_level], outputs=[output1], api_name="paint/0")
run2.click(send_it, inputs=[prompt, noise_level], outputs=[output2], api_name="paint/1")
run3.click(send_it, inputs=[prompt, noise_level], outputs=[output3], api_name="paint/2")
# run4.click(send_it, inputs=[prompt, noise_level], outputs=[output4])
demo.launch(enable_queue=True, inline=True, show_api=False)
block.queue(concurrency_count=2)