app.py

import gradio as gr
import requests
import json
import os

URL = "http://127.0.0.1:8188/prompt"

def get_latest_image(folder):
    files = os.listdir(folder)
    image_files = [f for f in files if f.lower().endswith(('.png','.jpg','.jpeg'))]

def start_queue(prompt_workflow):
    p = {"prompt": prompt_workflow}
    data = json.dumps(p).encode('utf-8')
    requests.post(URL, data=data)

def generate_image():
    with open("/home/ec2-user/SageMaker/truss-examples/comfyui-truss/data/comfy_ui_workflow.json", "r") as file_json:
        prompt = json.load(file_json)
    start_queue(prompt)

demo = gr.Interface(fn = generate_image, inputs = [], outputs = ["image"])
demo. launch()