|
import gradio as gr |
|
import diff |
|
import os |
|
from PIL import Image |
|
import uuid |
|
import random |
|
import bg |
|
from rembg import remove as rm |
|
proc1=diff |
|
proc2=diff |
|
|
|
|
|
|
|
|
|
|
|
sky = "https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/sky.png" |
|
platform = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/platform.png' |
|
star = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/star.png' |
|
bomb = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/bomb.png' |
|
dude = 'https://huggingface.co/spaces/Omnibus/game-test/resolve/main/assets/dude.png' |
|
|
|
|
|
|
|
def game_fn(sky=sky,star=star,bomb=bomb,platform=platform,dude=dude): |
|
html_mod=f""" |
|
<div id="my_game" style="height:600px"> |
|
<iframe |
|
id="myIframe" |
|
src="https://omnibus-game-test-static.static.hf.space/index.html?sky={sky}&platform={platform}&star={star}&bomb={bomb}&dude={dude}" |
|
frameborder="0" |
|
width="100%" |
|
height="100%" |
|
></iframe> |
|
</div>""" |
|
return html_mod |
|
|
|
def update_game(inp): |
|
|
|
return game_fn(sky=sky) |
|
|
|
def sky_fn(inp): |
|
rand = random.randint(1,200) |
|
for i in range(rand): |
|
inp+=" " |
|
uid=uuid.uuid4() |
|
output=proc1.send_it(inp,5,1) |
|
print(output) |
|
outp=Image.open(output[0]) |
|
width, height = outp.size |
|
rat = width/height |
|
if width > height: |
|
outp = outp.resize((600*rat,600)) |
|
elif width < height: |
|
outp = outp.resize((800,800*rat)) |
|
else: |
|
outp = outp.resize((800,536)) |
|
outp.save(f"{uid}_sky.png") |
|
out = os.path.abspath(f"{uid}_sky.png") |
|
|
|
out_url = f'https://omnibus-game-test.hf.space/file={out}' |
|
return output[0],out_url |
|
|
|
def star_fn(inp): |
|
rand = random.randint(1,200) |
|
for i in range(rand): |
|
inp+=" " |
|
uid=uuid.uuid4() |
|
output=proc2.send_it(inp,5,1) |
|
outp=Image.open(output[0]) |
|
out=rm(outp) |
|
outrs = out.resize((36,36)) |
|
outrs.save(f"{uid}_star.png") |
|
out_file = os.path.abspath(f"{uid}_star.png") |
|
out_url = f'https://omnibus-game-test.hf.space/file={out_file}' |
|
return out,out_url |
|
|
|
def enemy_fn(inp): |
|
rand = random.randint(1,200) |
|
for i in range(rand): |
|
inp+=" " |
|
uid=uuid.uuid4() |
|
output=proc2.send_it(inp,5,1) |
|
outp=Image.open(output[0]) |
|
out=rm(outp) |
|
outrs = out.resize((24,24)) |
|
outrs.save(f"{uid}_enemy.png") |
|
out_file = os.path.abspath(f"{uid}_enemy.png") |
|
out_url = f'https://omnibus-game-test.hf.space/file={out_file}' |
|
return out,out_url |
|
|
|
with gr.Blocks() as app: |
|
with gr.Row(): |
|
with gr.Column(): |
|
prompt_sky=gr.Textbox(label="Background",value="beautiful landscape, real, 8k") |
|
btn_sky=gr.Button("Make") |
|
out_im_sky=gr.Image(type='filepath') |
|
out_sky_url=gr.Textbox(visible=False) |
|
with gr.Column(): |
|
prompt_star=gr.Textbox(label="Star",value="Colorful Star, blank background") |
|
btn_star=gr.Button("Make") |
|
out_im_star=gr.Image(type='filepath') |
|
out_star_url=gr.Textbox(visible=False) |
|
with gr.Column(): |
|
prompt_enemy=gr.Textbox(label="Enemy",value="Ball on fire, blank background") |
|
btn_enemy=gr.Button("Make") |
|
out_im_enemy=gr.Image(type='filepath') |
|
out_enemy_url=gr.Textbox(visible=False) |
|
gr.Column() |
|
gr.Column() |
|
|
|
with gr.Row(): |
|
update_game=gr.Button("Make Game") |
|
|
|
html_game = gr.HTML() |
|
|
|
update_game.click(game_fn,[out_sky_url,out_star_url,out_enemy_url],html_game) |
|
btn_sky.click(sky_fn,prompt_sky,[out_im_sky,out_sky_url]) |
|
btn_star.click(star_fn,prompt_star,[out_im_star,out_star_url]) |
|
btn_enemy.click(enemy_fn,prompt_enemy,[out_im_enemy,out_enemy_url]) |
|
|
|
|
|
app.queue(concurrency_count=10).launch(show_api=False) |
