import spaces
from dataclasses import dataclass
import json
import logging
import os
import random
import re
import sys
import warnings

from PIL import Image
from diffusers import AutoencoderKL, FlowMatchEulerDiscreteScheduler
import gradio as gr
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

from prompt_check import is_unsafe_prompt

sys.path.append(os.path.dirname(os.path.abspath(__file__)))

from diffusers import ZImagePipeline
from diffusers.models.transformers.transformer_z_image import ZImageTransformer2DModel

from pe import prompt_template

#