Spaces:
Sleeping
Sleeping
File size: 1,974 Bytes
31d3041 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
import json
import os
import gradio as gr
import requests
import time
import numpy as np
import random
from PIL import Image
URL = "http://127.0.0.1:8188/prompt"
OUTPUT_DIR = "E:\\SD\\ComfyUI-aki-v1.3\\ComfyUI-aki-v1.3\\output"
def get_latest_image(folder):
files = os.listdir(folder)
image_files = [f for f in files if f.lower().endswith(('.png','.jpg','.jpeg'))]
image_files.sort(key=lambda x: os.path.getmtime(os.path.join(folder,x)))
latest_image = os.path.join(folder,image_files[-1]) if image_files else None
return latest_image
def update_seed():
seed_value = random.randint(1, 1000000)
json_file_path = "workflow_api.json"
with open(json_file_path, 'r', encoding='utf-8') as file:
data = json.load(file)
data['10']['inputs']['noise_seed'] = seed_value
with open(json_file_path, 'w', encoding='utf-8') as file:
json.dump(data, file, indent=4)
return seed_value
def start_queue(prompt_workflow):
p = {"prompt":prompt_workflow}
data = json.dumps(p).encode('utf-8')
requests.post(URL,data=data)
def generate_image(prompt_text,prompt_text2,cfg_count):
with open("workflow_api.json","r",encoding="utf-8") as file_json:
prompt = json.load(file_json)
prompt["6"]["inputs"]["text"] = f"digital artwork of a {prompt_text}"
prompt["7"]["inputs"]["text"] = f"digital artwork of a {prompt_text2}"
prompt["10"]["inputs"]["cfg"] = cfg_count
previous_image = get_latest_image(OUTPUT_DIR)
random_seed = update_seed()
print(f"New seed value generated:{random_seed}")
start_queue(prompt)
while True:
latest_image = get_latest_image(OUTPUT_DIR)
if latest_image != previous_image:
return latest_image
time.sleep(1)
demo = gr.Interface(fn=generate_image,inputs=["text","text","text"],outputs=["image"])
demo.launch(share=True) |