Spaces:
Running
Running
import gradio as gr | |
import requests | |
from pydantic import BaseModel | |
from diffusers.utils import load_image | |
from io import BytesIO | |
sdxl_inference_endpoint = 'https://vikramsingh178-picpilot-server.hf.space/api/v1/product-diffusion/sdxl_v0_lora_inference' | |
kandinsky_inpainting_inference = 'https://vikramsingh178-picpilot-server.hf.space/api/v1/product-diffusion/inpainting' | |
# Define the InpaintingRequest model | |
class InputRequest(BaseModel): | |
prompt: str | |
num_inference_steps: int | |
guidance_scale: float | |
negative_prompt: str | |
num_images: int | |
mode: str | |
class InpaintingRequest(BaseModel): | |
prompt: str | |
negative_prompt: str | |
num_inference_steps: int | |
strength: float | |
guidance_scale: float | |
mode: str | |
async def generate_sdxl_lora_image(prompt, negative_prompt, num_inference_steps, guidance_scale, num_images, mode): | |
# Prepare the payload for SDXL LORA API | |
payload = InputRequest( | |
prompt=prompt, | |
negative_prompt=negative_prompt, | |
num_inference_steps=num_inference_steps, | |
guidance_scale=guidance_scale, | |
num_images=num_images, | |
mode=mode | |
).dict() | |
response = requests.post(sdxl_inference_endpoint, json=payload) | |
response = response.json() | |
url = response['url'] | |
image = load_image(url) | |
return image | |
def generate_outpainting(prompt, negative_prompt, num_inference_steps, strength, guidance_scale, mode, image): | |
# Convert the image to bytes | |
img_byte_arr = BytesIO() | |
image.save(img_byte_arr, format='PNG') | |
img_byte_arr = img_byte_arr.getvalue() | |
# Prepare the payload for multipart/form-data | |
files = { | |
'image': ('image.png', img_byte_arr, 'image/png'), | |
'prompt': (None, prompt), | |
'negative_prompt': (None, negative_prompt), | |
'num_inference_steps': (None, str(num_inference_steps)), | |
'strength': (None, str(strength)), | |
'guidance_scale': (None, str(guidance_scale)), | |
'mode': (None, mode) | |
} | |
response = requests.post(kandinsky_inpainting_inference, files=files) | |
response.raise_for_status() | |
response = response.json() | |
url = response['url'] | |
image = load_image(url) | |
return image | |
with gr.Blocks(theme='VikramSingh178/Webui-Theme') as demo: | |
with gr.Tab("SdxL-Lora"): | |
with gr.Row(): | |
with gr.Column(): | |
with gr.Group(): | |
prompt = gr.Textbox(label="Prompt", placeholder="Enter your prompt here") | |
negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="Enter negative prompt here") | |
num_inference_steps = gr.Slider(minimum=1, maximum=1000, step=1, value=20, label="Inference Steps") | |
guidance_scale = gr.Slider(minimum=1.0, maximum=10.0, step=0.1, value=7.5, label="Guidance Scale") | |
num_images = gr.Slider(minimum=1, maximum=10, step=1, value=1, label="Number of Images") | |
mode = gr.Dropdown(choices=["s3_json", "b64_json"], value="s3_json", label="Mode") | |
generate_button = gr.Button("Generate Image",variant='primary') | |
with gr.Column(scale=1): | |
image_preview = gr.Image(label="Generated Image",show_download_button=True,show_share_button=True,container=True) | |
generate_button.click(generate_sdxl_lora_image, inputs=[prompt, negative_prompt, num_inference_steps, guidance_scale, num_images, mode], outputs=[image_preview]) | |
with gr.Tab("Generate AI Background"): | |
with gr.Row(): | |
with gr.Column(): | |
with gr.Group(): | |
image_input = gr.Image(type="pil", label="Upload Image") | |
prompt = gr.Textbox(label="Prompt", placeholder="Enter your prompt here") | |
negative_prompt = gr.Textbox(label="Negative Prompt", placeholder="Enter negative prompt here") | |
num_inference_steps = gr.Slider(minimum=1, maximum=500, step=1, value=20, label="Inference Steps") | |
guidance_scale = gr.Slider(minimum=1.0, maximum=10.0, step=0.1, value=7.5, label="Guidance Scale") | |
strength = gr.Slider(minimum=0.1, maximum=1, step=0.1, value=1, label="Strength") | |
mode = gr.Dropdown(choices=["s3_json", "b64_json"], value="s3_json", label="Mode") | |
generate_button = gr.Button("Generate Background", variant='primary') | |
with gr.Column(scale=1): | |
image_preview = gr.Image(label="Image", show_download_button=True, show_share_button=True, container=True) | |
generate_button.click(generate_outpainting, inputs=[prompt, negative_prompt, num_inference_steps, strength, guidance_scale, mode, image_input], outputs=[image_preview]) | |
demo.launch() | |