|
import gradio as gr |
|
import subprocess |
|
import os |
|
|
|
def run_workflow(): |
|
try: |
|
print("π Checking if hr.py exists...") |
|
script_path = "ComfyUI/hr.py" |
|
if not os.path.exists(script_path): |
|
print("β Error: hr.py not found!") |
|
return "Error: hr.py not found!" |
|
|
|
print("β
Running hr.py...") |
|
result = subprocess.run(["python", script_path], capture_output=True, text=True) |
|
|
|
if result.returncode == 0: |
|
print("β
Execution Successful!") |
|
return result.stdout |
|
else: |
|
print(f"β Execution Failed:\n{result.stderr}") |
|
return f"Error:\n{result.stderr}" |
|
|
|
except Exception as e: |
|
print(f"β Exception: {str(e)}") |
|
return f"Exception: {str(e)}" |
|
|
|
ui = gr.Interface( |
|
fn=run_workflow, |
|
inputs=[], |
|
outputs="text", |
|
title="Run ComfyUI Workflow", |
|
description="Click the button to execute hr.py and generate an image." |
|
) |
|
|
|
print("π Launching Gradio UI...") |
|
if __name__ == "__main__": |
|
ui.launch(share=True) |
|
|