|
import gradio as gr
|
|
from dartrs.v2 import AspectRatioTag, LengthTag, RatingTag, IdentityTag
|
|
|
|
|
|
V2_ASPECT_RATIO_OPTIONS: list[AspectRatioTag] = [
|
|
"ultra_wide",
|
|
"wide",
|
|
"square",
|
|
"tall",
|
|
"ultra_tall",
|
|
]
|
|
V2_RATING_OPTIONS: list[RatingTag] = [
|
|
"sfw",
|
|
"general",
|
|
"sensitive",
|
|
"nsfw",
|
|
"questionable",
|
|
"explicit",
|
|
]
|
|
V2_LENGTH_OPTIONS: list[LengthTag] = [
|
|
"very_short",
|
|
"short",
|
|
"medium",
|
|
"long",
|
|
"very_long",
|
|
]
|
|
V2_IDENTITY_OPTIONS: list[IdentityTag] = [
|
|
"none",
|
|
"lax",
|
|
"strict",
|
|
]
|
|
|
|
|
|
|
|
def gradio_copy_text(_text: None):
|
|
gr.Info("Copied!")
|
|
|
|
|
|
COPY_ACTION_JS = """\
|
|
(inputs, _outputs) => {
|
|
// inputs is the string value of the input_text
|
|
if (inputs.trim() !== "") {
|
|
navigator.clipboard.writeText(inputs);
|
|
}
|
|
}"""
|
|
|
|
|
|
def gradio_copy_prompt(prompt: str):
|
|
gr.Info("Copied!")
|
|
return prompt
|
|
|
