refactor: modularize codebase and reorganize utilities

- Split monolithic app.py into modules/config.py, modules/integrations.py, modules/core_logic.py, and modules/ui_layout.py.
- Consolited ComfyUI utilities into a dedicated comfy/ directory.
- Updated imports to use the new package structure.
- Enhanced .gitignore to include __pycache__ and temporary character saves.
- Documented new architecture and safer port management in design.md.

.gitignore

@@ -1,7 +1,21 @@
 venv/
-__pycache__/
-*.pyc
 .env
 .vscode/
 .idea/
+
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# Temporary character saves
+rpg_character_config.json
+temp_character_*.json
+
+# Generated images (if they should be ignored)
 *.png
+!comfy_workflow_visual.png # Exception example if needed
+
+# OS generated files
+.DS_Store
+Thumbs.db

app.py

@@ -1,643 +1,5 @@
 import gradio as gr
-import yaml
-import random
-import os
-from dotenv import load_dotenv
-from google import genai
-from google.genai import types
-from PIL import Image
-import io
-import json
-import tempfile
-import traceback
-import requests
-import uuid
-import time
-
-# Load environment variables
-load_dotenv()
-
-# Setup Gemini
-api_key = os.getenv("GEMINI_API_KEY")
-client = None
-gemini_active = False
-if api_key:
-    try:
-        client = genai.Client(api_key=api_key)
-        gemini_active = True
-    except Exception as e:
-        print(f"Error initializing Gemini: {e}")
-
-EXAMPLES_DIR = "examples"
-COMFY_HOST = os.getenv("COMFY_HOST", "127.0.0.1")
-COMFY_PORT = os.getenv("COMFY_PORT", "8188")
-COMFY_URL = f"http://{COMFY_HOST}:{COMFY_PORT}"
-
-COMFY_WORKFLOW_FILE = "comfy_rpg_char_gen.json"
-OLLAMA_HOST = os.getenv("OLLAMA_HOST", "127.0.0.1")
-OLLAMA_PORT = os.getenv("OLLAMA_PORT", "11434")
-OLLAMA_MODEL = os.getenv("OLLAMA_MODEL", "llama3")
-
-def load_system_prompt(key="refinement"):
-    """Loads a system prompt from prompts.yaml."""
-    try:
-        with open("prompts.yaml", "r") as f:
-            prompts = yaml.safe_load(f)
-            return prompts.get(key, {}).get("system_instructions", "")
-    except Exception as e:
-        print(f"Error loading system prompt: {e}")
-        return ""
-
-def get_ollama_models():
-    """Fetches available models from Ollama server and checks if it's running."""
-    url = f"http://{OLLAMA_HOST}:{OLLAMA_PORT}/api/tags"
-    try:
-        response = requests.get(url, timeout=2)
-        if response.status_code == 200:
-            models = response.json().get("models", [])
-            return [m["name"] for m in models]
-        return []
-    except Exception:
-        return []
-
-# Load features from YAML
-def load_features():
-    with open("features.yaml", "r") as f:
-        return yaml.safe_load(f)
-
-features_data = load_features()
-
-# Define segments for prompt building
-# (Category, Subcategory, Template Key)
-FEATURE_SEQUENCE = [
-    ('identity', 'race', 'race'), 
-    ('identity', 'class', 'class'), 
-    ('identity', 'gender', 'gender'), 
-    ('identity', 'age', 'age'),
-    ('expression_pose', 'expression', 'expression'), 
-    ('expression_pose', 'pose', 'pose'),
-    ('appearance', 'hair_color', 'hair_color'), 
-    ('appearance', 'hair_style', 'hair_style'), 
-    ('appearance', 'eye_color', 'eye_color'), 
-    ('appearance', 'build', 'build'), 
-    ('appearance', 'skin_tone', 'skin_tone'), 
-    ('appearance', 'distinguishing_feature', 'distinguishing_feature'),
-    ('equipment', 'armor', 'armor'), 
-    ('equipment', 'weapon', 'weapon'), 
-    ('equipment', 'accessory', 'accessory'), 
-    ('equipment', 'accessory', 'accessory2'), 
-    ('equipment', 'material', 'material'),
-    ('environment', 'background', 'background'), 
-    ('environment', 'lighting', 'lighting'), 
-    ('environment', 'atmosphere', 'atmosphere'),
-    ('vfx_style', 'vfx', 'vfx'), 
-    ('vfx_style', 'style', 'style'), 
-    ('vfx_style', 'mood', 'mood'), 
-    ('vfx_style', 'camera', 'camera'),
-    ('technical', 'aspect_ratio', 'aspect_ratio')
-]
-
-# Section names for extra info
-SECTIONS = ['Identity', 'Appearance', 'Equipment', 'Environment', 'Style']
-
-def get_detail(category, subcategory, key):
-    """Retrieves the detailed description for a given key in a category/subcategory."""
-    return features_data.get(category, {}).get(subcategory, {}).get(key, key)
-
-def generate_prompt(*args):
-    """
-    Assembles the prompt based on dropdown selections and extra text info.
-    """
-    num_features = len(FEATURE_SEQUENCE)
-    feature_keys = args[:num_features]
-    extra_infos = args[num_features : num_features + len(SECTIONS)]
-
-    template = features_data.get("templates", {}).get("default", "")
-    
-    context = {}
-    for i, (cat, subcat, t_key) in enumerate(FEATURE_SEQUENCE):
-        key = feature_keys[i]
-        context[t_key] = get_detail(cat, subcat, key)
-    
-    # Handle multiple accessories dynamically
-    acc_list = []
-    # Identify accessory keys in context
-    for i, (cat, subcat, t_key) in enumerate(FEATURE_SEQUENCE):
-        if subcat == 'accessory' and feature_keys[i] != "None":
-            acc_list.append(context[t_key])
-    
-    if acc_list:
-        context['accessories'] = ", and has " + " as well as ".join(acc_list)
-    else:
-        context['accessories'] = ""
-    
-    # Inject extra info into the context
-    if extra_infos[0]: # Identity
-        context['age'] += f", {extra_infos[0]}"
-    if extra_infos[1]: # Appearance
-        context['distinguishing_feature'] += f", also {extra_infos[1]}"
-    if extra_infos[2]: # Equipment
-        if context['accessories']:
-            context['accessories'] += f", further complemented by {extra_infos[2]}"
-        else:
-            context['accessories'] = f", complemented by {extra_infos[2]}"
-    if extra_infos[3]: # Environment
-        context['atmosphere'] += f", additionally {extra_infos[3]}"
-    if extra_infos[4]: # Style
-        context['camera'] += f", art style notes: {extra_infos[4]}"
-    
-    try:
-        return template.format(**context)
-    except Exception as e:
-        return f"Error building prompt: {e}"
-
-def handle_regeneration(*args):
-    """Randomizes checkboxes and returns new values for dropdowns."""
-    num_features = len(FEATURE_SEQUENCE)
-    current_values = list(args[:num_features])
-    checkboxes = args[num_features : num_features*2]
-    
-    new_values = []
-    for i, (is_random, (cat, subcat, t_key)) in enumerate(zip(checkboxes, FEATURE_SEQUENCE)):
-        if is_random:
-            choices = list(features_data[cat][subcat].keys())
-            new_values.append(random.choice(choices))
-        else:
-            new_values.append(current_values[i])
-            
-    return new_values
-
-def save_character(*args):
-    """Saves all current UI values to a JSON file."""
-    num_features = len(FEATURE_SEQUENCE)
-    feature_keys = args[:num_features]
-    checkboxes = args[num_features : num_features*2]
-    extra_infos = args[num_features*2 : num_features*2 + len(SECTIONS)]
-    
-    data = {
-        "features": {FEATURE_SEQUENCE[i][2]: feature_keys[i] for i in range(num_features)},
-        "randomization": {FEATURE_SEQUENCE[i][2]: checkboxes[i] for i in range(num_features)},
-        "extra_info": {SECTIONS[i].lower(): extra_infos[i] for i in range(len(SECTIONS))}
-    }
-    
-    # Save to a file with a friendly name in a temp directory
-    temp_dir = tempfile.mkdtemp()
-    path = os.path.join(temp_dir, "rpg_character_config.json")
-    with open(path, 'w') as f:
-        json.dump(data, f, indent=4)
-    return path
-
-def load_character(file):
-    """Loads UI values from a JSON file."""
-    if file is None:
-        return [gr.update()] * (len(FEATURE_SEQUENCE) * 2 + len(SECTIONS))
-    
-    try:
-        with open(file.name, 'r') as f:
-            data = json.load(f)
-        
-        updates = []
-        # Update dropdowns
-        for _, _, t_key in FEATURE_SEQUENCE:
-            updates.append(data.get("features", {}).get(t_key, gr.update()))
-        
-        # Update checkboxes (randomization flags)
-        for _, _, t_key in FEATURE_SEQUENCE:
-            updates.append(data.get("randomization", {}).get(t_key, gr.update()))
-        
-        # Update extra info textboxes
-        for section in SECTIONS:
-            updates.append(data.get("extra_info", {}).get(section.lower(), ""))
-            
-        return updates
-    except Exception as e:
-        print(f"Error loading character: {e}")
-        return [gr.update()] * (len(FEATURE_SEQUENCE) * 2 + len(SECTIONS))
-
-def get_example_list():
-    """Returns a list of JSON filenames in the examples directory."""
-    if not os.path.exists(EXAMPLES_DIR):
-        return []
-    return [f for f in os.listdir(EXAMPLES_DIR) if f.endswith('.json')]
-
-def load_example_character(filename):
-    """Loads a character config from the examples directory."""
-    if not filename:
-        return [gr.update()] * (len(FEATURE_SEQUENCE) * 2 + len(SECTIONS))
-    
-    path = os.path.join(EXAMPLES_DIR, filename)
-    # Wrap path in a mock file object with a .name attribute for load_character
-    class MockFile:
-        def __init__(self, name):
-            self.name = name
-    
-    return load_character(MockFile(path))
-
-def refine_with_gemini(prompt):
-    if not gemini_active:
-        return "Gemini API key not found in .env file."
-    
-    system_prompt = load_system_prompt("refinement")
-    if not system_prompt:
-        # Fallback if file is missing
-        system_prompt = (
-            "You are an expert prompt engineer for AI image generators. "
-            "Your task is to take the provided technical prompt and refine it into a more vivid, "
-            "artistic, and detailed description while maintaining all the core features."
-        )
-    
-    try:
-        response = client.models.generate_content(
-            model='gemini-3-pro-preview',
-            contents=f"{system_prompt}\n\nOriginal Prompt: {prompt}"
-        )
-        text = response.text.strip()
-        # Remove common markdown code block wrappings if present
-        if text.startswith("```"):
-            lines = text.splitlines()
-            if lines[0].startswith("```"):
-                lines = lines[1:]
-            if lines and lines[-1].startswith("```"):
-                lines = lines[:-1]
-            text = "\n".join(lines).strip()
-        return text
-    except Exception as e:
-        return f"Error refining prompt with Gemini: {e}"
-
-def refine_with_ollama(prompt, model):
-    """Refines the prompt using a local Ollama instance."""
-    system_prompt = load_system_prompt("refinement")
-    url = f"http://{OLLAMA_HOST}:{OLLAMA_PORT}/api/generate"
-    
-    payload = {
-        "model": model,
-        "prompt": f"{system_prompt}\n\nOriginal Prompt: {prompt}",
-        "stream": False
-    }
-    
-    try:
-        response = requests.post(url, json=payload)
-        response.raise_for_status()
-        text = response.json().get("response", "").strip()
-        # Clean up potential markdown
-        if text.startswith("```"):
-            lines = text.splitlines()
-            if lines[0].startswith("```"): lines = lines[1:]
-            if lines and lines[-1].startswith("```"): lines = lines[:-1]
-            text = "\n".join(lines).strip()
-        return text
-    except Exception as e:
-        return f"Error refining prompt with Ollama: {e}"
-
-def refine_master(prompt, backend, ollama_model):
-    """Routes prompt refinement to the selected backend."""
-    if not prompt.strip():
-        return ""
-    if backend == "Ollama (Local)":
-        return refine_with_ollama(prompt, ollama_model)
-    else:
-        return refine_with_gemini(prompt)
-
-def generate_image_with_gemini(refined_prompt, technical_prompt, aspect_ratio):
-    if not gemini_active:
-        return None, None, "Gemini API key not found in .env file."
-    
-    # Priority: Refined Prompt -> Technical Prompt
-    final_prompt = refined_prompt.strip() if refined_prompt and refined_prompt.strip() else technical_prompt.strip()
-    
-    if not final_prompt:
-        return None, None, "No prompt available for generation."
-    
-    try:
-        # Using the new SDK's generate_images method
-        response = client.models.generate_images(
-            model='imagen-4.0-generate-001',
-            prompt=final_prompt,
-            config=types.GenerateImagesConfig(
-                aspect_ratio=aspect_ratio,
-                output_mime_type='image/png'
-            )
-        )
-        if response.generated_images:
-            img = Image.open(io.BytesIO(response.generated_images[0].image.image_bytes))
-            # Save as PNG to a temp file for download
-            temp_dir = tempfile.mkdtemp()
-            img_path = os.path.join(temp_dir, "rpg_portrait.png")
-            img.save(img_path, "PNG")
-            return img, img_path, f"Image generated using {'refined' if refined_prompt.strip() else 'technical'} prompt!"
-        return None, None, "Gemini Image generation did not return any images."
-    except Exception as e:
-        traceback.print_exc()
-        return None, None, f"Image Generation Error: {e}"
-
-def generate_image_with_comfy(prompt, aspect_ratio):
-    """Generates an image using a local ComfyUI instance."""
-    if not os.path.exists(COMFY_WORKFLOW_FILE):
-        return None, None, f"Error: Worklfow file {COMFY_WORKFLOW_FILE} not found."
-    
-    try:
-        with open(COMFY_WORKFLOW_FILE, 'r') as f:
-            workflow = json.load(f)
-        
-        # Node Mapping based on comfy_rpg_char_gen.json
-        # Node 6: Positive Prompt (CLIPTextEncode)
-        # Node 7: Negative Prompt (CLIPTextEncode) - we keep original for now
-        # Node 13: Aspect Ratio (EmptySD3LatentImage)
-        # Node 38: Seed (KSampler)
-        
-        # 1. Update Prompt
-        workflow["6"]["inputs"]["text"] = prompt
-        
-        # 2. Update Resolution based on Aspect Ratio
-        res_map = {
-            "1:1": (1024, 1024),
-            "16:9": (1344, 768),
-            "9:16": (768, 1344),
-            "4:3": (1152, 864),
-            "3:4": (864, 1152)
-        }
-        width, height = res_map.get(aspect_ratio, (1024, 1024))
-        workflow["13"]["inputs"]["width"] = width
-        workflow["13"]["inputs"]["height"] = height
-        
-        # 3. Randomize Seed
-        workflow["38"]["inputs"]["seed"] = random.randint(1, 1125899906842624)
-        
-        # 4. Queue Prompt
-        client_id = str(uuid.uuid4())
-        payload = {"prompt": workflow, "client_id": client_id}
-        
-        print(f"Queueing ComfyUI prompt (client_id: {client_id})")
-        response = requests.post(f"{COMFY_URL}/prompt", json=payload)
-        response.raise_for_status()
-        prompt_id = response.json().get("prompt_id")
-        
-        # 5. Poll for completion (simple approach)
-        print(f"Waiting for prompt {prompt_id} to complete...")
-        max_retries = 60
-        for _ in range(max_retries):
-            # Check history
-            hist_resp = requests.get(f"{COMFY_URL}/history/{prompt_id}")
-            if hist_resp.status_code == 200:
-                history = hist_resp.json()
-                if prompt_id in history:
-                    # Completed!
-                    outputs = history[prompt_id].get("outputs", {})
-                    # Find the first image in any output node
-                    for node_id in outputs:
-                        if "images" in outputs[node_id]:
-                            image_data = outputs[node_id]["images"][0]
-                            filename = image_data["filename"]
-                            subfolder = image_data["subfolder"]
-                            folder_type = image_data["type"]
-                            
-                            # Download the image
-                            img_url = f"{COMFY_URL}/view?filename={filename}&subfolder={subfolder}&type={folder_type}"
-                            img_resp = requests.get(img_url)
-                            img_resp.raise_for_status()
-                            
-                            img = Image.open(io.BytesIO(img_resp.content))
-                            # Save to temp
-                            temp_dir = tempfile.mkdtemp()
-                            img_path = os.path.join(temp_dir, "rpg_portrait_comfy.png")
-                            img.save(img_path, "PNG")
-                            
-                            return img, img_path, f"Image generated via ComfyUI (Node {node_id})!"
-            time.sleep(1)
-            
-        return None, None, "ComfyUI generation timed out."
-        
-    except Exception as e:
-        traceback.print_exc()
-        return None, None, f"ComfyUI Error: {e}"
-
-def generate_image_master(refined_prompt, technical_prompt, aspect_ratio, backend):
-    """Routes image generation to the selected backend."""
-    # Priority: Refined Prompt -> Technical Prompt
-    final_prompt = refined_prompt.strip() if refined_prompt and refined_prompt.strip() else technical_prompt.strip()
-    
-    if not final_prompt:
-        return None, None, "No prompt available for generation."
-        
-    if backend == "ComfyUI (Local)":
-        return generate_image_with_comfy(final_prompt, aspect_ratio)
-    else:
-        return generate_image_with_gemini(refined_prompt, technical_prompt, aspect_ratio)
-
-def build_ui():
-    with gr.Blocks(title="RPGPortrait Prompt Builder Pro") as demo:
-        gr.Markdown("# 🎨 RPGPortrait Pro")
-        gr.Markdown("Create professional AI prompts and generate portraits.")
-
-        dropdowns = []
-        checkboxes = []
-        extra_texts = []
-
-        def create_feature_ui(category, subcategory, label, default_value):
-            choices = list(features_data[category][subcategory].keys())
-            gr.Markdown(f"**{label}**")
-            with gr.Row(equal_height=True):
-                dd = gr.Dropdown(choices=choices, value=default_value, scale=12, show_label=False)
-                cb = gr.Checkbox(label="🎲", value=False, scale=1, min_width=50, container=False)
-            dropdowns.append(dd)
-            checkboxes.append(cb)
-
-        with gr.Row():
-            with gr.Column(scale=2):
-                with gr.Tabs():
-                    with gr.TabItem("👤 Identity & Expression"):
-                        gr.Markdown("### 👤 Identity")
-                        create_feature_ui('identity', 'race', "Race", "Human")
-                        create_feature_ui('identity', 'class', "Class", "Fighter")
-                        create_feature_ui('identity', 'gender', "Gender", "Male")
-                        create_feature_ui('identity', 'age', "Age", "Young Adult")
-                        extra_id = gr.Textbox(placeholder="Extra Identity details (e.g. lineage, title)", label="Additional Identity Info")
-                        extra_texts.append(extra_id)
-                        
-                        gr.Markdown("### 🎭 Expression & Pose")
-                        create_feature_ui('expression_pose', 'expression', "Expression", "Determined")
-                        create_feature_ui('expression_pose', 'pose', "Pose", "Standing")
-
-                    with gr.TabItem("🎨 Appearance"):
-                        gr.Markdown("### 🎨 Appearance")
-                        create_feature_ui('appearance', 'hair_color', "Hair Color", "Brown")
-                        create_feature_ui('appearance', 'hair_style', "Hair Style", "Short")
-                        create_feature_ui('appearance', 'eye_color', "Eye Color", "Brown")
-                        create_feature_ui('appearance', 'build', "Build", "Average")
-                        create_feature_ui('appearance', 'skin_tone', "Skin Tone", "Fair")
-                        create_feature_ui('appearance', 'distinguishing_feature', "Distinguishing Feature", "None")
-                        extra_app = gr.Textbox(placeholder="Extra Appearance details (e.g. tattoos, scars)", label="Additional Appearance Info")
-                        extra_texts.append(extra_app)
-
-                    with gr.TabItem("⚔️ Equipment"):
-                        gr.Markdown("### ⚔️ Equipment")
-                        create_feature_ui('equipment', 'armor', "Armor/Clothing", "Travelers Clothes")
-                        create_feature_ui('equipment', 'weapon', "Primary Weapon", "Longsword")
-                        create_feature_ui('equipment', 'accessory', "Accessory 1", "None")
-                        create_feature_ui('equipment', 'accessory', "Accessory 2", "None")
-                        create_feature_ui('equipment', 'material', "Material Detail", "Weathered")
-                        extra_eq = gr.Textbox(placeholder="Extra Equipment details (e.g. weapon enchantments)", label="Additional Equipment Info")
-                        extra_texts.append(extra_eq)
-
-                    with gr.TabItem("🌍 Environment"):
-                        gr.Markdown("### 🌍 Environment")
-                        create_feature_ui('environment', 'background', "Background", "Forest")
-                        create_feature_ui('environment', 'lighting', "Lighting", "Natural Sunlight")
-                        create_feature_ui('environment', 'atmosphere', "Atmosphere", "Clear")
-                        extra_env = gr.Textbox(placeholder="Extra Environment details (e.g. time of day, weather)", label="Additional Environment Info")
-                        extra_texts.append(extra_env)
-
-                    with gr.TabItem("🪄 Style & Technical"):
-                        gr.Markdown("### 🪄 Style & Effects")
-                        create_feature_ui('vfx_style', 'vfx', "Special Effects", "None")
-                        create_feature_ui('vfx_style', 'style', "Art Style", "Digital Illustration")
-                        create_feature_ui('vfx_style', 'mood', "Mood", "Heroic")
-                        create_feature_ui('vfx_style', 'camera', "Camera Angle", "Bust")
-                        extra_sty = gr.Textbox(placeholder="Extra Style details (e.g. specific artists, colors)", label="Additional Style Info")
-                        extra_texts.append(extra_sty)
-                        
-                        gr.Markdown("### ⚙️ Technical")
-                        create_feature_ui('technical', 'aspect_ratio', "Aspect Ratio", "1:1")
-
-                gr.Markdown("---")
-                
-                with gr.Row():
-                    example_dropdown = gr.Dropdown(choices=get_example_list(), label="Example Characters", scale=3)
-                    load_example_btn = gr.Button("📂 Load Example", variant="secondary", scale=1)
-                    save_btn = gr.Button("💾 Save Character", variant="secondary", scale=1)
-                    load_btn = gr.UploadButton("📂 Load Character", file_types=[".json"], variant="secondary", scale=1)
-                
-                with gr.Group():
-                    gr.Markdown("### ⚙️ Settings & Generation")
-                    with gr.Row():
-                        ollama_models = get_ollama_models()
-                        ollama_active = len(ollama_models) > 0
-                        
-                        refinement_backend = gr.Radio(
-                            choices=["Gemini (Cloud)", "Ollama (Local)"] if ollama_active else ["Gemini (Cloud)"],
-                            value="Gemini (Cloud)",
-                            label="Prompt Refinement Backend",
-                            scale=2
-                        )
-                        
-                        ollama_model_dropdown = gr.Dropdown(
-                            choices=ollama_models,
-                            value=ollama_models[0] if ollama_active else None,
-                            label="Ollama Model",
-                            visible=ollama_active and False,
-                            scale=1
-                        )
-                    
-                    with gr.Row():
-                        backend_selector = gr.Radio(
-                            choices=["Gemini (Cloud)", "ComfyUI (Local)"], 
-                            value="Gemini (Cloud)", 
-                            label="Image Generation Backend",
-                            scale=2
-                        )
-                        with gr.Column(scale=1):
-                            refine_btn = gr.Button("🧠 Refine Prompt", variant="primary")
-                            gen_img_btn = gr.Button("🖼️ Generate Image", variant="primary")
-
-            with gr.Column(scale=1):
-                gr.Markdown("### 📝 Prompts & Output")
-                prompt_output = gr.Textbox(label="Generated Technical Prompt", lines=4, interactive=False)
-                regenerate_btn = gr.Button("✨ Randomize Features", variant="secondary")
-                refined_output = gr.Textbox(label="Refined Artistic Prompt", lines=6, interactive=True)
-                
-                gr.Markdown("---")
-                image_output = gr.Image(label="Portrait", show_label=False)
-                download_img_btn = gr.DownloadButton("📥 Download Portrait (PNG)", variant="secondary", visible=False)
-                status_msg = gr.Markdown("")
-                download_file = gr.File(label="Saved Character JSON", visible=False)
-
-        all_input_components = dropdowns + extra_texts
-        
-        # Automatic prompt update on any input change
-        for comp in all_input_components:
-            comp.change(fn=generate_prompt, inputs=all_input_components, outputs=prompt_output)
-        
-        # Regenerate button logic
-        regenerate_btn.click(
-            fn=handle_regeneration,
-            inputs=dropdowns + checkboxes,
-            outputs=dropdowns
-        ).then(
-            # Trigger prompt update after dropdowns change
-            fn=generate_prompt,
-            inputs=all_input_components,
-            outputs=prompt_output
-        ).then(
-            # Clear refined output since it's now stale
-            fn=lambda: "",
-            outputs=refined_output
-        )
-
-        # Prompt Refinement Logic
-        refine_btn.click(
-            fn=refine_master,
-            inputs=[prompt_output, refinement_backend, ollama_model_dropdown],
-            outputs=refined_output
-        )
-
-        # Image Generation Logic
-        gen_img_btn.click(
-            fn=generate_image_master,
-            inputs=[refined_output, prompt_output, dropdowns[-1], backend_selector],
-            outputs=[image_output, download_img_btn, status_msg]
-        ).then(
-            fn=lambda x: gr.update(value=x, visible=True) if x else gr.update(visible=False),
-            inputs=download_img_btn,
-            outputs=download_img_btn
-        )
-
-        # Save/Load Logic
-        
-        # Simple toggle for Ollama model visibility
-        def toggle_ollama_visibility(backend):
-            return gr.update(visible=(backend == "Ollama (Local)"))
-        
-        refinement_backend.change(
-            fn=toggle_ollama_visibility,
-            inputs=refinement_backend,
-            outputs=ollama_model_dropdown
-        )
-
-        save_btn.click(
-            fn=save_character,
-            inputs=dropdowns + checkboxes + extra_texts,
-            outputs=download_file
-        ).then(
-            fn=lambda: gr.update(visible=True),
-            outputs=download_file
-        )
-
-        load_btn.upload(
-            fn=load_character,
-            inputs=load_btn,
-            outputs=dropdowns + checkboxes + extra_texts
-        ).then(
-            # Trigger prompt update after loading
-            fn=generate_prompt,
-            inputs=all_input_components,
-            outputs=prompt_output
-        )
-
-        load_example_btn.click(
-            fn=load_example_character,
-            inputs=example_dropdown,
-            outputs=dropdowns + checkboxes + extra_texts
-        ).then(
-            # Trigger prompt update after loading
-            fn=generate_prompt,
-            inputs=all_input_components,
-            outputs=prompt_output
-        )
-        
-        # Initialize
-        demo.load(fn=generate_prompt, inputs=all_input_components, outputs=prompt_output)
-
-    return demo
+from modules.ui_layout import build_ui
 
 if __name__ == "__main__":
     demo = build_ui()

commit_msg.txt

@@ -0,0 +1,7 @@
+refactor: modularize codebase and reorganize utilities
+
+- Split monolithic app.py into modules/config.py, modules/integrations.py, modules/core_logic.py, and modules/ui_layout.py.
+- Consolited ComfyUI utilities into a dedicated comfy/ directory.
+- Updated imports to use the new package structure.
+- Enhanced .gitignore to include __pycache__ and temporary character saves.
+- Documented new architecture and safer port management in design.md.

design.md

@@ -52,13 +52,20 @@ The app uses a 3-stage prompt pipeline:
 ## Maintenance & Development Lessons
 
 ### 1. Server Lifecycle
-- **Mandatory Restarts**: Any modification to `app.py` or configuration files (`features.yaml`, `prompts.yaml`) requires a manual restart of the Gradio server to take effect. The UI does not hot-reload backend logic or YAML data.
-- **Process Management (CRITICAL)**: Always ensure the previously running process is fully terminated before starting a new one. Gradio will automatically increment the port (7860 -> 7861 -> 7862) if the old process is still bound to the port. 
-- **Command**: Use `Get-Process -Name python | Stop-Process -Force` in PowerShell to clear all existing instances before a fresh run.
+- **Mandatory Restarts**: Any modification to the `modules/` folder or configuration files (`features.yaml`, `prompts.yaml`) requires a manual restart of the Gradio server.
+- **Process Management (CRITICAL)**: Always ensure the previously running process is fully terminated before starting a new one. 
+- **Safe Command**: Use this PowerShell command to only kill the process using the target port (7860):
+  ```powershell
+  Stop-Process -Id (Get-NetTCPConnection -LocalPort 7860 -ErrorAction SilentlyContinue).OwningProcess -Force -ErrorAction SilentlyContinue
+  ```
 
-### 2. UI Verification
-- **Data Dependency**: When testing UI components or dropdowns via browser automation, verify that the target values exist in `features.yaml` first. Testing with invalid data (e.g., "Silver" hair color before it was added) leads to verification failures.
-- **UI State Reset**: Always refresh the browser page after a server restart to ensure the frontend state is synchronized with the new backend.
+### 2. Project Architecture (Modular)
+- **`app.py`**: Entry point that launches the Gradio `demo`.
+- **`modules/config.py`**: Global constants and environment variables.
+- **`modules/integrations.py`**: Wrappers for AI backends (Gemini, Ollama, ComfyUI).
+- **`modules/core_logic.py`**: Character state management and prompt assembly.
+- **`modules/ui_layout.py`**: The full Gradio UI definition (`build_ui`).
+- **`comfy/`**: Dedicated folder for ComfyUI-specific JSON workflows and utility scripts.
 
 ### 3. Documentation & Artifacts
 - **Path Formatting**: For markdown artifacts (like `walkthrough.md`), use absolute paths for media embeds. On Windows, prefix with a leading slash (e.g., `/C:/Users/...`) to comply with internal linting and ensuring reliable rendering.

modules/config.py

@@ -0,0 +1,57 @@
+import os
+from dotenv import load_dotenv
+
+# Load environment variables
+load_dotenv()
+
+# ComfyUI Settings
+COMFY_HOST = os.getenv("COMFY_HOST", "127.0.0.1")
+COMFY_PORT = os.getenv("COMFY_PORT", "8188")
+COMFY_URL = f"http://{COMFY_HOST}:{COMFY_PORT}"
+COMFY_WORKFLOW_FILE = "comfy/comfy_rpg_char_gen.json"
+
+# Ollama Settings
+OLLAMA_HOST = os.getenv("OLLAMA_HOST", "127.0.0.1")
+OLLAMA_PORT = os.getenv("OLLAMA_PORT", "11434")
+OLLAMA_MODEL = os.getenv("OLLAMA_MODEL", "llama3")
+
+# Gemini Settings
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+
+# Application Constants
+EXAMPLES_DIR = "examples"
+PROMPTS_FILE = "prompts.yaml"
+FEATURES_FILE = "features.yaml"
+
+# Define segments for prompt building
+# (Category, Subcategory, Template Key)
+FEATURE_SEQUENCE = [
+    ('identity', 'race', 'race'), 
+    ('identity', 'class', 'class'), 
+    ('identity', 'gender', 'gender'), 
+    ('identity', 'age', 'age'),
+    ('expression_pose', 'expression', 'expression'), 
+    ('expression_pose', 'pose', 'pose'),
+    ('appearance', 'hair_color', 'hair_color'), 
+    ('appearance', 'hair_style', 'hair_style'), 
+    ('appearance', 'eye_color', 'eye_color'), 
+    ('appearance', 'build', 'build'), 
+    ('appearance', 'skin_tone', 'skin_tone'), 
+    ('appearance', 'distinguishing_feature', 'distinguishing_feature'),
+    ('equipment', 'armor', 'armor'), 
+    ('equipment', 'weapon', 'weapon'), 
+    ('equipment', 'accessory', 'accessory'), 
+    ('equipment', 'accessory', 'accessory2'), 
+    ('equipment', 'material', 'material'),
+    ('environment', 'background', 'background'), 
+    ('environment', 'lighting', 'lighting'), 
+    ('environment', 'atmosphere', 'atmosphere'),
+    ('vfx_style', 'vfx', 'vfx'), 
+    ('vfx_style', 'style', 'style'), 
+    ('vfx_style', 'mood', 'mood'), 
+    ('vfx_style', 'camera', 'camera'),
+    ('technical', 'aspect_ratio', 'aspect_ratio')
+]
+
+# Section names for extra info
+SECTIONS = ['Identity', 'Appearance', 'Equipment', 'Environment', 'Style']

modules/core_logic.py

@@ -0,0 +1,142 @@
+import yaml
+import random
+import os
+import json
+import tempfile
+import gradio as gr
+from .config import FEATURES_FILE, FEATURE_SEQUENCE, SECTIONS, EXAMPLES_DIR
+
+def load_features():
+    with open(FEATURES_FILE, "r") as f:
+        return yaml.safe_load(f)
+
+features_data = load_features()
+
+def get_detail(category, subcategory, key):
+    """Retrieves the detailed description for a given key in a category/subcategory."""
+    return features_data.get(category, {}).get(subcategory, {}).get(key, key)
+
+def generate_prompt(*args):
+    """
+    Assembles the prompt based on dropdown selections and extra text info.
+    """
+    num_features = len(FEATURE_SEQUENCE)
+    feature_keys = args[:num_features]
+    extra_infos = args[num_features : num_features + len(SECTIONS)]
+
+    template = features_data.get("templates", {}).get("default", "")
+    
+    context = {}
+    for i, (cat, subcat, t_key) in enumerate(FEATURE_SEQUENCE):
+        key = feature_keys[i]
+        context[t_key] = get_detail(cat, subcat, key)
+    
+    # Handle multiple accessories dynamically
+    acc_list = []
+    # Identify accessory keys in context
+    for i, (cat, subcat, t_key) in enumerate(FEATURE_SEQUENCE):
+        if subcat == 'accessory' and feature_keys[i] != "None":
+            acc_list.append(context[t_key])
+    
+    if acc_list:
+        context['accessories'] = ", and has " + " as well as ".join(acc_list)
+    else:
+        context['accessories'] = ""
+    
+    # Inject extra info into the context
+    if extra_infos[0]: # Identity
+        context['age'] += f", {extra_infos[0]}"
+    if extra_infos[1]: # Appearance
+        context['distinguishing_feature'] += f", also {extra_infos[1]}"
+    if extra_infos[2]: # Equipment
+        if context['accessories']:
+            context['accessories'] += f", further complemented by {extra_infos[2]}"
+        else:
+            context['accessories'] = f", complemented by {extra_infos[2]}"
+    if extra_infos[3]: # Environment
+        context['atmosphere'] += f", additionally {extra_infos[3]}"
+    if extra_infos[4]: # Style
+        context['camera'] += f", art style notes: {extra_infos[4]}"
+    
+    try:
+        return template.format(**context)
+    except Exception as e:
+        return f"Error building prompt: {e}"
+
+def handle_regeneration(*args):
+    """Randomizes checkboxes and returns new values for dropdowns."""
+    num_features = len(FEATURE_SEQUENCE)
+    current_values = list(args[:num_features])
+    checkboxes = args[num_features : num_features*2]
+    
+    new_values = []
+    for i, (is_random, (cat, subcat, t_key)) in enumerate(zip(checkboxes, FEATURE_SEQUENCE)):
+        if is_random:
+            choices = list(features_data[cat][subcat].keys())
+            new_values.append(random.choice(choices))
+        else:
+            new_values.append(current_values[i])
+            
+    return new_values
+
+def save_character(*args):
+    """Saves all current UI values to a JSON file."""
+    num_features = len(FEATURE_SEQUENCE)
+    feature_keys = args[:num_features]
+    checkboxes = args[num_features : num_features*2]
+    extra_infos = args[num_features*2 : num_features*2 + len(SECTIONS)]
+    
+    data = {
+        "features": {FEATURE_SEQUENCE[i][2]: feature_keys[i] for i in range(num_features)},
+        "randomization": {FEATURE_SEQUENCE[i][2]: checkboxes[i] for i in range(num_features)},
+        "extra_info": {SECTIONS[i].lower(): extra_infos[i] for i in range(len(SECTIONS))}
+    }
+    
+    # Save to a file with a friendly name in a temp directory
+    temp_dir = tempfile.mkdtemp()
+    path = os.path.join(temp_dir, "rpg_character_config.json")
+    with open(path, 'w') as f:
+        json.dump(data, f, indent=4)
+    return path
+
+def load_character(file):
+    """Loads UI values from a JSON file."""
+    if file is None:
+        return [gr.update()] * (len(FEATURE_SEQUENCE) * 2 + len(SECTIONS))
+    
+    try:
+        file_path = file.name if hasattr(file, 'name') else file
+        with open(file_path, 'r') as f:
+            data = json.load(f)
+        
+        updates = []
+        # Update dropdowns
+        for _, _, t_key in FEATURE_SEQUENCE:
+            updates.append(data.get("features", {}).get(t_key, gr.update()))
+        
+        # Update checkboxes (randomization flags)
+        for _, _, t_key in FEATURE_SEQUENCE:
+            updates.append(data.get("randomization", {}).get(t_key, gr.update()))
+        
+        # Update extra info textboxes
+        for section in SECTIONS:
+            updates.append(data.get("extra_info", {}).get(section.lower(), ""))
+            
+        return updates
+    except Exception as e:
+        print(f"Error loading character: {e}")
+        return [gr.update()] * (len(FEATURE_SEQUENCE) * 2 + len(SECTIONS))
+
+def get_example_list():
+    """Returns a list of JSON filenames in the examples directory."""
+    if not os.path.exists(EXAMPLES_DIR):
+        return []
+    return [f for f in os.listdir(EXAMPLES_DIR) if f.endswith('.json')]
+
+def load_example_character(filename):
+    """Loads a character config from the examples directory."""
+    if not filename:
+        return [gr.update()] * (len(FEATURE_SEQUENCE) * 2 + len(SECTIONS))
+    
+    path = os.path.join(EXAMPLES_DIR, filename)
+    return load_character(path)

modules/integrations.py

@@ -0,0 +1,201 @@
+import os
+import requests
+import json
+import uuid
+import time
+import io
+import yaml
+import random
+import traceback
+import tempfile
+from PIL import Image
+from google import genai
+from google.genai import types
+from .config import (
+    GEMINI_API_KEY, OLLAMA_HOST, OLLAMA_PORT, COMFY_URL, 
+    COMFY_WORKFLOW_FILE, PROMPTS_FILE
+)
+
+# Setup Gemini
+client = None
+gemini_active = False
+if GEMINI_API_KEY:
+    try:
+        client = genai.Client(api_key=GEMINI_API_KEY)
+        gemini_active = True
+    except Exception as e:
+        print(f"Error initializing Gemini: {e}")
+
+def load_system_prompt(key="refinement"):
+    """Loads a system prompt from prompts.yaml."""
+    try:
+        with open(PROMPTS_FILE, "r") as f:
+            prompts = yaml.safe_load(f)
+            return prompts.get(key, {}).get("system_instructions", "")
+    except Exception as e:
+        print(f"Error loading system prompt: {e}")
+        return ""
+
+def get_ollama_models():
+    """Fetches available models from Ollama server and checks if it's running."""
+    url = f"http://{OLLAMA_HOST}:{OLLAMA_PORT}/api/tags"
+    try:
+        response = requests.get(url, timeout=2)
+        if response.status_code == 200:
+            models = response.json().get("models", [])
+            return [m["name"] for m in models]
+        return []
+    except Exception:
+        return []
+
+def refine_with_gemini(prompt):
+    if not gemini_active:
+        return "Gemini API key not found in .env file."
+    
+    system_prompt = load_system_prompt("refinement")
+    if not system_prompt:
+        system_prompt = (
+            "You are an expert prompt engineer for AI image generators. "
+            "Your task is to take the provided technical prompt and refine it into a more vivid, "
+            "artistic, and detailed description while maintaining all the core features."
+        )
+    
+    try:
+        response = client.models.generate_content(
+            model="gemini-2.0-flash-exp",
+            config=types.GenerateContentConfig(
+                system_instruction=system_prompt,
+                temperature=0.7,
+            ),
+            contents=[prompt]
+        )
+        return response.text.strip()
+    except Exception as e:
+        return f"Error refining prompt with Gemini: {e}"
+
+def refine_with_ollama(prompt, model):
+    """Refines the prompt using a local Ollama instance."""
+    system_prompt = load_system_prompt("refinement")
+    url = f"http://{OLLAMA_HOST}:{OLLAMA_PORT}/api/generate"
+    
+    payload = {
+        "model": model,
+        "prompt": f"{system_prompt}\n\nOriginal Prompt: {prompt}",
+        "stream": False
+    }
+    
+    try:
+        response = requests.post(url, json=payload)
+        response.raise_for_status()
+        text = response.json().get("response", "").strip()
+        # Clean up potential markdown
+        if text.startswith("```"):
+            lines = text.splitlines()
+            if lines[0].startswith("```"): lines = lines[1:]
+            if lines and lines[-1].startswith("```"): lines = lines[:-1]
+            text = "\n".join(lines).strip()
+        return text
+    except Exception as e:
+        return f"Error refining prompt with Ollama: {e}"
+
+def refine_master(prompt, backend, ollama_model):
+    """Routes prompt refinement to the selected backend."""
+    if not prompt.strip():
+        return ""
+    if backend == "Ollama (Local)":
+        return refine_with_ollama(prompt, ollama_model)
+    else:
+        return refine_with_gemini(prompt)
+
+def generate_image_with_gemini(refined_prompt, technical_prompt, aspect_ratio):
+    if not gemini_active:
+        return None, None, "Gemini API key not found in .env file."
+    
+    final_prompt = refined_prompt.strip() if refined_prompt and refined_prompt.strip() else technical_prompt.strip()
+    
+    if not final_prompt:
+        return None, None, "No prompt available for generation."
+    
+    try:
+        response = client.models.generate_images(
+            model='imagen-3.0-generate-001',
+            prompt=final_prompt,
+            config=types.GenerateImagesConfig(
+                aspect_ratio=aspect_ratio,
+                output_mime_type='image/png'
+            )
+        )
+        if response.generated_images:
+            img = Image.open(io.BytesIO(response.generated_images[0].image.image_bytes))
+            temp_dir = tempfile.mkdtemp()
+            img_path = os.path.join(temp_dir, "rpg_portrait.png")
+            img.save(img_path, "PNG")
+            return img, img_path, f"Image generated using {'refined' if refined_prompt.strip() else 'technical'} prompt!"
+        return None, None, "Gemini Image generation did not return any images."
+    except Exception as e:
+        traceback.print_exc()
+        return None, None, f"Image Generation Error: {e}"
+
+def generate_image_with_comfy(prompt, aspect_ratio):
+    """Generates an image using a local ComfyUI instance."""
+    if not os.path.exists(COMFY_WORKFLOW_FILE):
+        return None, None, f"Error: Workflow file {COMFY_WORKFLOW_FILE} not found."
+    
+    try:
+        with open(COMFY_WORKFLOW_FILE, 'r') as f:
+            workflow = json.load(f)
+        
+        workflow["6"]["inputs"]["text"] = prompt
+        
+        res_map = {
+            "1:1": (1024, 1024),
+            "16:9": (1344, 768),
+            "9:16": (768, 1344),
+            "4:3": (1152, 864),
+            "3:4": (864, 1152)
+        }
+        width, height = res_map.get(aspect_ratio, (1024, 1024))
+        workflow["13"]["inputs"]["width"] = width
+        workflow["13"]["inputs"]["height"] = height
+        workflow["38"]["inputs"]["seed"] = random.randint(1, 1125899906842624)
+        
+        client_id = str(uuid.uuid4())
+        payload = {"prompt": workflow, "client_id": client_id}
+        
+        response = requests.post(f"{COMFY_URL}/prompt", json=payload)
+        response.raise_for_status()
+        prompt_id = response.json().get("prompt_id")
+        
+        max_retries = 60
+        for _ in range(max_retries):
+            hist_resp = requests.get(f"{COMFY_URL}/history/{prompt_id}")
+            if hist_resp.status_code == 200:
+                history = hist_resp.json()
+                if prompt_id in history:
+                    outputs = history[prompt_id].get("outputs", {})
+                    for node_id in outputs:
+                        if "images" in outputs[node_id]:
+                            image_data = outputs[node_id]["images"][0]
+                            img_url = f"{COMFY_URL}/view?filename={image_data['filename']}&subfolder={image_data['subfolder']}&type={image_data['type']}"
+                            img_resp = requests.get(img_url)
+                            img_resp.raise_for_status()
+                            
+                            img = Image.open(io.BytesIO(img_resp.content))
+                            temp_dir = tempfile.mkdtemp()
+                            img_path = os.path.join(temp_dir, "rpg_portrait_comfy.png")
+                            img.save(img_path, "PNG")
+                            return img, img_path, f"Image generated via ComfyUI!"
+            time.sleep(1)
+            
+        return None, None, "ComfyUI generation timed out."
+        
+    except Exception as e:
+        traceback.print_exc()
+        return None, None, f"ComfyUI Error: {e}"
+
+def generate_image_master(refined_prompt, technical_prompt, aspect_ratio, backend):
+    """Routes image generation to the selected backend."""
+    if backend == "ComfyUI (Local)":
+        return generate_image_with_comfy(refined_prompt if refined_prompt.strip() else technical_prompt, aspect_ratio)
+    else:
+        return generate_image_with_gemini(refined_prompt, technical_prompt, aspect_ratio)

modules/ui_layout.py

@@ -0,0 +1,209 @@
+import gradio as gr
+from .config import FEATURE_SEQUENCE, SECTIONS
+from .core_logic import (
+    features_data, generate_prompt, handle_regeneration, 
+    save_character, load_character, get_example_list, load_example_character
+)
+from .integrations import (
+    get_ollama_models, refine_master, generate_image_master
+)
+
+def build_ui():
+    with gr.Blocks(title="RPGPortrait Prompt Builder Pro") as demo:
+        gr.Markdown("# 🎨 RPGPortrait Pro")
+        gr.Markdown("Create professional AI prompts and generate portraits.")
+
+        dropdowns = []
+        checkboxes = []
+        extra_texts = []
+
+        def create_feature_ui(category, subcategory, label, default_value):
+            choices = list(features_data[category][subcategory].keys())
+            gr.Markdown(f"**{label}**")
+            with gr.Row(equal_height=True):
+                dd = gr.Dropdown(choices=choices, value=default_value, scale=12, show_label=False)
+                cb = gr.Checkbox(label="🎲", value=False, scale=1, min_width=50, container=False)
+            dropdowns.append(dd)
+            checkboxes.append(cb)
+
+        with gr.Row():
+            with gr.Column(scale=2):
+                with gr.Tabs():
+                    with gr.TabItem("👤 Identity & Expression"):
+                        gr.Markdown("### 👤 Identity")
+                        create_feature_ui('identity', 'race', "Race", "Human")
+                        create_feature_ui('identity', 'class', "Class", "Fighter")
+                        create_feature_ui('identity', 'gender', "Gender", "Male")
+                        create_feature_ui('identity', 'age', "Age", "Young Adult")
+                        extra_id = gr.Textbox(placeholder="Extra Identity details (e.g. lineage, title)", label="Additional Identity Info")
+                        extra_texts.append(extra_id)
+                        
+                        gr.Markdown("### 🎭 Expression & Pose")
+                        create_feature_ui('expression_pose', 'expression', "Expression", "Determined")
+                        create_feature_ui('expression_pose', 'pose', "Pose", "Standing")
+
+                    with gr.TabItem("🎨 Appearance"):
+                        gr.Markdown("### 🎨 Appearance")
+                        create_feature_ui('appearance', 'hair_color', "Hair Color", "Brown")
+                        create_feature_ui('appearance', 'hair_style', "Hair Style", "Short")
+                        create_feature_ui('appearance', 'eye_color', "Eye Color", "Brown")
+                        create_feature_ui('appearance', 'build', "Build", "Average")
+                        create_feature_ui('appearance', 'skin_tone', "Skin Tone", "Fair")
+                        create_feature_ui('appearance', 'distinguishing_feature', "Distinguishing Feature", "None")
+                        extra_app = gr.Textbox(placeholder="Extra Appearance details (e.g. tattoos, scars)", label="Additional Appearance Info")
+                        extra_texts.append(extra_app)
+
+                    with gr.TabItem("⚔️ Equipment"):
+                        gr.Markdown("### ⚔️ Equipment")
+                        create_feature_ui('equipment', 'armor', "Armor/Clothing", "Travelers Clothes")
+                        create_feature_ui('equipment', 'weapon', "Primary Weapon", "Longsword")
+                        create_feature_ui('equipment', 'accessory', "Accessory 1", "None")
+                        create_feature_ui('equipment', 'accessory', "Accessory 2", "None")
+                        create_feature_ui('equipment', 'material', "Material Detail", "Weathered")
+                        extra_eq = gr.Textbox(placeholder="Extra Equipment details (e.g. weapon enchantments)", label="Additional Equipment Info")
+                        extra_texts.append(extra_eq)
+
+                    with gr.TabItem("🌍 Environment"):
+                        gr.Markdown("### 🌍 Environment")
+                        create_feature_ui('environment', 'background', "Background", "Forest")
+                        create_feature_ui('environment', 'lighting', "Lighting", "Natural Sunlight")
+                        create_feature_ui('environment', 'atmosphere', "Atmosphere", "Clear")
+                        extra_env = gr.Textbox(placeholder="Extra Environment details (e.g. time of day, weather)", label="Additional Environment Info")
+                        extra_texts.append(extra_env)
+
+                    with gr.TabItem("🪄 Style & Technical"):
+                        gr.Markdown("### 🪄 Style & Effects")
+                        create_feature_ui('vfx_style', 'vfx', "Special Effects", "None")
+                        create_feature_ui('vfx_style', 'style', "Art Style", "Digital Illustration")
+                        create_feature_ui('vfx_style', 'mood', "Mood", "Heroic")
+                        create_feature_ui('vfx_style', 'camera', "Camera Angle", "Bust")
+                        extra_sty = gr.Textbox(placeholder="Extra Style details (e.g. specific artists, colors)", label="Additional Style Info")
+                        extra_texts.append(extra_sty)
+                        
+                        gr.Markdown("### ⚙️ Technical")
+                        create_feature_ui('technical', 'aspect_ratio', "Aspect Ratio", "1:1")
+
+                gr.Markdown("---")
+                
+                with gr.Row():
+                    example_dropdown = gr.Dropdown(choices=get_example_list(), label="Example Characters", scale=3)
+                    load_example_btn = gr.Button("📂 Load Example", variant="secondary", scale=1)
+                    save_btn = gr.Button("💾 Save Character", variant="secondary", scale=1)
+                    load_btn = gr.UploadButton("📂 Load Character", file_types=[".json"], variant="secondary", scale=1)
+                
+                with gr.Group():
+                    gr.Markdown("### ⚙️ Settings & Generation")
+                    with gr.Row():
+                        ollama_models = get_ollama_models()
+                        ollama_active = len(ollama_models) > 0
+                        
+                        refinement_backend = gr.Radio(
+                            choices=["Gemini (Cloud)", "Ollama (Local)"] if ollama_active else ["Gemini (Cloud)"],
+                            value="Gemini (Cloud)",
+                            label="Prompt Refinement Backend",
+                            scale=2
+                        )
+                        
+                        ollama_model_dropdown = gr.Dropdown(
+                            choices=ollama_models,
+                            value=ollama_models[0] if ollama_active else None,
+                            label="Ollama Model",
+                            visible=False,
+                            scale=1
+                        )
+                    
+                    with gr.Row():
+                        backend_selector = gr.Radio(
+                            choices=["Gemini (Cloud)", "ComfyUI (Local)"], 
+                            value="Gemini (Cloud)", 
+                            label="Image Generation Backend",
+                            scale=2
+                        )
+                        with gr.Column(scale=1):
+                            refine_btn = gr.Button("🧠 Refine Prompt", variant="primary")
+                            gen_img_btn = gr.Button("🖼️ Generate Image", variant="primary")
+
+            with gr.Column(scale=1):
+                gr.Markdown("### 📝 Prompts & Output")
+                prompt_output = gr.Textbox(label="Generated Technical Prompt", lines=4, interactive=False)
+                regenerate_btn = gr.Button("✨ Randomize Features", variant="secondary")
+                refined_output = gr.Textbox(label="Refined Artistic Prompt", lines=6, interactive=True)
+                
+                gr.Markdown("---")
+                image_output = gr.Image(label="Portrait", show_label=False)
+                download_img_btn = gr.DownloadButton("📥 Download Portrait (PNG)", variant="secondary", visible=False)
+                status_msg = gr.Markdown("")
+                download_file = gr.File(label="Saved Character JSON", visible=False)
+
+        all_input_components = dropdowns + extra_texts
+        
+        for comp in all_input_components:
+            comp.change(fn=generate_prompt, inputs=all_input_components, outputs=prompt_output)
+        
+        regenerate_btn.click(
+            fn=handle_regeneration,
+            inputs=dropdowns + checkboxes,
+            outputs=dropdowns
+        ).then(
+            fn=generate_prompt,
+            inputs=all_input_components,
+            outputs=prompt_output
+        ).then(
+            fn=lambda: "",
+            outputs=refined_output
+        )
+
+        refine_btn.click(
+            fn=refine_master,
+            inputs=[prompt_output, refinement_backend, ollama_model_dropdown],
+            outputs=refined_output
+        )
+
+        gen_img_btn.click(
+            fn=generate_image_master,
+            inputs=[refined_output, prompt_output, dropdowns[-1], backend_selector],
+            outputs=[image_output, download_img_btn, status_msg]
+        ).then(
+            fn=lambda x: gr.update(value=x, visible=True) if x else gr.update(visible=False),
+            inputs=download_img_btn,
+            outputs=download_img_btn
+        )
+
+        refinement_backend.change(
+            fn=lambda b: gr.update(visible=(b == "Ollama (Local)")),
+            inputs=refinement_backend,
+            outputs=ollama_model_dropdown
+        )
+
+        save_btn.click(
+            fn=save_character,
+            inputs=dropdowns + checkboxes + extra_texts,
+            outputs=download_file
+        ).then(
+            fn=lambda: gr.update(visible=True),
+            outputs=download_file
+        )
+
+        load_btn.upload(
+            fn=load_character,
+            inputs=load_btn,
+            outputs=dropdowns + checkboxes + extra_texts
+        ).then(
+            fn=generate_prompt,
+            inputs=all_input_components,
+            outputs=prompt_output
+        )
+
+        load_example_btn.click(
+            fn=load_example_character,
+            inputs=example_dropdown,
+            outputs=dropdowns + checkboxes + extra_texts
+        ).then(
+            fn=generate_prompt,
+            inputs=all_input_components,
+            outputs=prompt_output
+        )
+        
+        demo.load(fn=generate_prompt, inputs=all_input_components, outputs=prompt_output)
+
+    return demo