Upload txt2gif-Breastgrowthfast.py

Lora/timelapseBreast_breastV1/txt2gif-Breastgrowthfast.py

@@ -0,0 +1,109 @@
+import modules.scripts as scripts
+import gradio as gr
+import os
+import uuid 
+
+from modules.processing import process_images, Processed, fix_seed
+from modules.shared import opts, cmd_opts, state
+
+def build_prompts(character_prompt, lora_name, lora_strength):
+  prompts = []
+
+  prompts.append(f"topless, (small breasts:1.2), nipples, {character_prompt}, ")
+  prompts.append(f"topless, (small breasts:1.1), nipples, {character_prompt}, ")
+  prompts.append(f"topless, (small breasts:1.0), nipples, {character_prompt}, ")
+  prompts.append(f"topless, (small breasts:0.9), nipples, {character_prompt}, ")
+  prompts.append(f"topless, (small breasts:0.8), nipples, {character_prompt}, ")
+  prompts.append(f"topless, medium breasts, nipples, {character_prompt}, ")
+  prompts.append(f"topless, medium breasts, nipples, {character_prompt}, ")
+  prompts.append(f"topless, (large breasts:0.9), nipples, {character_prompt}, ")
+  prompts.append(f"topless, (large breasts:1.0), nipples, {character_prompt}, ")
+  prompts.append(f"topless, (large breasts:1.1), nipples, {character_prompt}, ")
+  prompts.append(f"topless, (large breasts:1.2), nipples, {character_prompt}, ")
+  prompts.append(f"topless, (large breasts:1.3), nipples, {character_prompt}, ")
+  prompts.append(f"topless, (large breasts:1.4), nipples, {character_prompt}, ")
+  prompts.append(f"topless, (huge breasts:1.0), nipples, {character_prompt}, ")
+  prompts.append(f"topless, (huge breasts:1.1), nipples, {character_prompt}, ")
+  prompts.append(f"topless, (huge breasts:1.2), nipples, {character_prompt}, ")
+  prompts.append(f"topless, (huge breasts:1.3), nipples, {character_prompt}, ")
+  prompts.append(f"topless, (huge breasts:1.4), nipples, {character_prompt}, ")
+
+  
+
+
+
+  
+
+  return prompts
+
+def make_gif(frames, filename = None):
+    if filename is None:
+      filename = str(uuid.uuid4())
+
+    outpath = "outputs/txt2img-images/txt2gif-Breastgrowthfast"
+    if not os.path.exists(outpath):
+        os.makedirs(outpath)
+
+    first_frame, append_frames = frames[0], frames[1:]
+
+    first_frame.save(f"{outpath}/{filename}.gif", format="GIF", append_images=append_frames,
+               save_all=True, duration=100, loop=0)
+
+    return first_frame
+
+def main(p, lora_name, lora_strength):
+  character_prompt = p.prompt.strip().rstrip(',')
+  frame_prompts = build_prompts(character_prompt, lora_name, lora_strength)
+
+  fix_seed(p)
+
+  state.job_count = len(frame_prompts)
+
+  imgs = []
+  all_prompts = []
+  infotexts = []
+
+  for i in range(len(frame_prompts)):
+    if state.interrupted:
+      break
+
+    p.prompt = frame_prompts[i]
+    proc = process_images(p)
+
+    if state.interrupted:
+      break
+
+    imgs += proc.images
+    all_prompts += proc.all_prompts
+    infotexts += proc.infotexts
+
+  gif = [make_gif(imgs)]
+
+  imgs += gif
+
+  return Processed(p, imgs, p.seed, "", all_prompts=all_prompts, infotexts=infotexts)
+
+class Script(scripts.Script):
+    is_txt2img = False
+
+    # Function to set title
+    def title(self):
+        return "txt2gif-Breastgrowthfast"
+
+    def ui(self, is_img2img):
+        with gr.Row():
+          lora_name = gr.Textbox(label="lora name", value = "")
+        with gr.Row():
+          lora_strength = gr.Number(label="lora strength", value = 0)
+
+
+        return [lora_name, lora_strength]
+
+    # Function to show the script
+    def show(self, is_img2img):
+        return True
+
+    # Function to run the script
+    def run(self, p, lora_name, lora_strength):
+        # Make a process_images Object
+        return main(p, lora_name, lora_strength)