core/generation_logic.py

from typing import Any, Dict
import gradio as gr

from core.pipelines.sd_image_pipeline import SdImagePipeline
from core.pipelines.newbie_xml_pipeline import NewbieXmlPipeline

sd_image_pipeline = SdImagePipeline()
newbie_xml_pipeline = NewbieXmlPipeline()


def generate_image_wrapper(ui_inputs: dict, progress=gr.Progress(track_tqdm=True)):
    return sd_image_pipeline.run(ui_inputs=ui_inputs, progress=progress)

def generate_xml_wrapper(ui_inputs: dict, progress=gr.Progress(track_tqdm=True)):
    return newbie_xml_pipeline.run(ui_inputs=ui_inputs, progress=progress)