Create app.py

app.py

@@ -0,0 +1,82 @@
+import gradio as gr
+import random
+
+def generate_prompt(scene_tags, num_people, position_tags, characters, outfit_tags, camera_tags, concept_tags, lora_tags, num_tags=3):
+    all_tags = {
+        "scene": scene_tags.split(','),
+        "position": position_tags.split(','),
+        "outfit": outfit_tags.split(','),
+        "camera": camera_tags.split(','),
+        "concept": concept_tags.split(','),
+    }
+    
+    # Remove empty strings and strip whitespace
+    for category in all_tags:
+        all_tags[category] = [tag.strip() for tag in all_tags[category] if tag.strip()]
+    
+    # Process characters
+    character_list = [char.strip() for char in characters.split(';') if char.strip()]
+    character_prompts = []
+    for i, character in enumerate(character_list):
+        char_tags = character.split(',')
+        char_name = char_tags[0].strip()
+        char_traits = [tag.strip() for tag in char_tags[1:] if tag.strip()]
+        char_prompt = f"{char_name}, " + ", ".join(random.sample(char_traits, min(num_tags, len(char_traits))))
+        character_prompts.append(char_prompt)
+    
+    # Generate main prompt
+    selected_tags = []
+    for category, tags in all_tags.items():
+        if tags:
+            selected_tags.extend(f"{tag}:{random.uniform(0.8, 1.2):.2f}" for tag in random.sample(tags, min(num_tags, len(tags))))
+    
+    # Add number of people
+    if num_people.strip():
+        selected_tags.append(f"{num_people} people:1.1")
+    
+    # Combine all parts of the prompt
+    prompt_parts = character_prompts + selected_tags
+    random.shuffle(prompt_parts)
+    main_prompt = ", ".join(prompt_parts)
+    
+    # Add LORA tags
+    lora_list = [lora.strip() for lora in lora_tags.split(',') if lora.strip()]
+    lora_prompt = " ".join(f"<lora:{lora}:1>" for lora in lora_list)
+    
+    # Add fixed tags at the end
+    fixed_tags = "source_anime, score_9, score_8_up, score_7_up, masterpiece, best quality, very aesthetic, absurdres, anime artwork, anime style, vibrant, studio anime, highly detailed"
+    
+    full_prompt = f"{main_prompt}, {fixed_tags} {lora_prompt}".strip()
+    return full_prompt
+
+def create_ui():
+    with gr.Blocks() as demo:
+        gr.Markdown("# Advanced Pony SDXL Prompt Generator")
+        
+        with gr.Row():
+            with gr.Column():
+                scene_input = gr.Textbox(label="Scene Tags (comma-separated)")
+                num_people_input = gr.Textbox(label="Number of People")
+                position_input = gr.Textbox(label="Position Tags (comma-separated)")
+                characters_input = gr.Textbox(label="Characters (Format: Name, Trait1, Trait2; Name2, Trait1, Trait2)")
+                outfit_input = gr.Textbox(label="Outfit Tags (comma-separated)")
+                camera_input = gr.Textbox(label="Camera View/Angle Tags (comma-separated)")
+                concept_input = gr.Textbox(label="Concept Tags (comma-separated)")
+                lora_input = gr.Textbox(label="LORA Tags (comma-separated)")
+                num_tags_slider = gr.Slider(minimum=1, maximum=5, step=1, value=3, label="Number of tags per category")
+                generate_button = gr.Button("Generate Prompt")
+            
+            with gr.Column():
+                output = gr.Textbox(label="Generated Prompt", lines=5)
+        
+        generate_button.click(
+            generate_prompt,
+            inputs=[scene_input, num_people_input, position_input, characters_input, outfit_input, camera_input, concept_input, lora_input, num_tags_slider],
+            outputs=[output]
+        )
+    
+    return demo
+
+if __name__ == "__main__":
+    demo = create_ui()
+    demo.launch()