app.py

import gradio as gr
from diffusers import DiffusionPipeline
from peft import PeftModel
import torch
import huggingface_hub as hf
import networkx as nx
import re

# === SAFETY GATE (keeps HF happy) ===
def safety_check(prompt):
    bad_words = ['nude', 'naked', 'violence', 'blood', 'gore', 'deepfake']
    if any(word in prompt.lower() for word in bad_words):
        return None, 'Keep it creative and safe! Try again.'
    return prompt, None

# === THE FUSION ENGINE (expert level, kid-simple) ===
def necro_fuse(prompt):
    prompt, error = safety_check(prompt)
    if error:
        return error, None
    
    # 1. Load base skeleton (the strong modern model)
    pipe = DiffusionPipeline.from_pretrained(
        'runwayml/stable-diffusion-v1-5',
        torch_dtype=torch.float16
    )
    pipe.to('cuda' if torch.cuda.is_available() else 'cpu')
    
    # 2. TODO: Add real LoRA zombie later (this is where the magic happens)
    # pipe = PeftModel.from_pretrained(pipe, 'your-old-lora-repo-here')
    
    # 3. Agent brain graph - makes it smarter by itself
    graph = nx.DiGraph()
    graph.add_node('improve', func=lambda p: p + ', highly detailed, cyberpunk neon panda style')
    improved_prompt = graph.nodes['improve']['func'](prompt)
    
    # 4. Generate the abomination
    image = pipe(improved_prompt, num_inference_steps=30).images[0]
    
    return 'Fusion complete! Zombie model + new brain = epic result', image

# === THE COOL BROWSER INTERFACE (no quote scars) ===
interface = gr.Interface(
    fn=necro_fuse,
    inputs=gr.Textbox(
        label='Drop Your Fusion Idea',
        placeholder='Merge dead graffiti GAN with 2026 diffusion for cyberpunk panda',
        lines=3
    ),
    outputs=[
        gr.Textbox(label='Status'),
        gr.Image(label='Your New Creation')
    ],
    title='NecroFusion Coliseum',
    description='Dig up old AI zombies, fuse them, watch them evolve. Research only!',
    examples=[
        ['cyberpunk panda in neon rain'],
        ['old stylegan graffiti fused with modern flux']
    ],
    allow_flagging='never'
)

if __name__ == '__main__':
    interface.launch()