from dataclasses import dataclass
from PIL import Image
from typing import List
import torch

@dataclass
class VyroParams:
    latents: torch.Tensor
    user_prompt: str
    user_neg_prompt: str
    mode: str
    cfg: float
    batch_size: int
    steps: int
    width: int
    height: int
    seed: int
    denoise: float

    stage1_strength: float
    stage2_strength: float
    efficiency_multiplier: float
    style: str = ''
    final_positive_prompt: str = ''
    final_negative_prompt: str = ''
    is_raw: bool = False
  
    
    MULTIPLIER = [1.0,1.33, 2.0]
    STATES = ["disabled", "enabled"]
    ALLOWED = ["allow", "deny"]

    MODE = ["t2i", "i2i"]

    STYLES = [
        "3d render",
        "abstract art",
        "anime",
        "architecture",
        "cinematic",
        "conceptual art",
        "dark fantasy",
        "fantasy realism",
        "fashion",
        "graffiti",
        "illustration",
        "interior design",
        "logo",
        "painting",
        "photography",
        "portrait photography",
        "poster",
        "product",
        "sticker",
        "surrealism",
        "typography",
        "ukiyo-e",
        "vector design",
        "vibrant digital artwork",
        "watercolor",
        "wildlife photography"
    ]
    PARAMS = [
        "latents",
        "user_prompt",
        "user_neg_prompt",
        "mode",
        "cfg",
        "steps",
        "width",
        "height",
        "seed",
        "denoise",
     
        "stage1_strength",
        "stage2_strength",
        "efficiency_multiplier",
        "style",
        "final_positive_prompt",
        "final_negative_prompt",
        "is_raw"
       
    ]