Create app.py

app.py

@@ -0,0 +1,23 @@
+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()