from gradio_client import Client

import random
import time

def public_sdxl_gen(prompt : str, neg_prompt : str, progress : callable, n, poll_time):
    seed = random.randrange(1000000000, 1999999999)
    client = Client("https://victor-sdxl-0-9.hf.space/")
    job = client.submit(prompt, neg_prompt, 10, 50, seed, api_name="/predict")
    while not job.done():
        job_status = job.status()
        progress_str = "{r}/{q}, eta = {eta}".format(r=job_status.rank,q=job_status.queue_size,eta=job_status.eta)
        progress(n, desc=progress_str)
        time.sleep(poll_time)
    return (seed, job.result())