app.py

import gradio as gr
import torch
import time
import random
import threading  # CRITICAL: Fixes the NameError
from queue import Queue
from PIL import Image
import peft
from diffusers import LCMScheduler, AutoPipelineForText2Image

# --- Configuration ---
DEVICE = "cpu"
DTYPE = torch.float32
MODEL_NAME = "AlekseyCalvin/acs_model"

# Generation Settings
UPDATE_INTERVAL = 13  # Seconds between background generations
UI_CHECK_INTERVAL = 2 # Seconds between UI checks
MAX_HISTORY = 100     # Max images to keep in the archive

# --- Prompts ---
SCREENSAVER_PROMPTS = [
    "fractal Jodorowsky Mandelbrot set city masterpiece cosmic maze gallery, a surrealist eclectically collaged scene from an Alejandro Jodorowsky analog 1970s IMG_1099.CR2 footage film, giant exotic birds and manticores and ancient sharks and leopards, Aztec warriors",
    "crisp geometric pattern refined matte art of close-up lifelkke fantastic animals, psychedelic film by jodorowsky, long-legged owl demon astronomer prince owl in center smashing up supermarket counters while singing, ",
    "8k art photo of acs person, grayscale, by Cartier-Bresson, a top-down view on waterless river composed of shredded movie tickets, close-up of anthropomorphic animal beatniks",
    "arcane occult text detailed artwork of Yaldabaoth preaching to a crowd of people, close-up Lion-headed serpent bodied demiurge god makes for an impressive sight, detailed HD illustration",
    "Occult Art by Leonora Carrington, a sentient cyborg weeping in a vast digital landscape, close-up pencil art",
    "detailed Max Ernst-style ultra-HD action painting of Russian ghost poet soldier Ophelia carried by a mythic Don river of words screens honey blood, barbed wire fence border under burning guard towers, Russian or Ukrainian flags ",
    "By Frida Kahlo, surreal art, Yellow a symbol of madness, sickness, fear, Part of the sun, joy. Magenta an Aztec symbol Old TLAPALLI prickly pear blood brightest, oldest. Cobalt Blue symbol of electricity, purity love. Leaf Green symbol of leafs, sadness, science, all Germany leaf-green",
    "dslr IMG_1099.CR2 photo of acs person outdoors at Albany Bulb of a complex scene, tall thin androgynous close-up attractive skeleton person, straight longish-forest-green hair, long tarot-magician staff instead of heart, raises pointer finger, wearing ornate sorcerer robes covered with arcane occult symbols. Nearby are two phantasmagoric intrepid rebels, children of shadows riding on snaking light beams",
    "abstract fluid art by Lyubov Popova, suprematism, blue and purple",
    "8k HD iconic IMG_1099.CR2  print showing close up Frida Ophelia Kahlo resting on a floor-bed in a granite prison cell and looking out at the starry sky in the window",
    "illustration by Maria Yakunchikova, on an alien planet with a view of outer space, close-up cat cosmonaut begins spitting out entire cities out of its mouth, refined details",
    "realistic HD planet earth sinking in a wide deep river of shattered glass, a vivid crisp Roy Lichtenstein illustration, iconic detailed composition",
    "moonlit war-field, Jan Švankmajer-style stop motion phantasmagoric creatures sleeping everywhere",
    "pop art masterpiece, in beautiful sunny city, young close-up poets of all genders laughing playing tag in the sun, influenced by ilya kuvshinov, godard, vrubel, lucian freud",
    "Keith Thompson and ren renfa-influenced pen drawing of close-up pussy riot in red masks chairing a world government assembly",
    "Aly Fell-style magazine close-up portrait of pussy riot members hanging out with Pushkin on a beach in Sochi",
    "anthropomorphic beautiful moon enters a narrow doorway at night, 8K ultra resolution, ultrarealistic, hyperdetailed, art by Austin Osman Spare and Anne Taintor, trending on Artstation, silvery purple hair, very high detail, ultra high texture detail",
    "Mary Georgina Filmer style, close up of joyous mutinous androgynous soldiers ripping maps in a field at night, starry sky up inspired by James Webb Space Telescope images",
    "detailed crisp banksy style, canon photo of acs person style masterpiece showing many people going to sleep in a nighttime disappearing city",
    "ultra-HD lush Climent Redko art, Rooftops are throbbing under the weight of days, strange humans and animal hybrid mutants walk USSR streets celebrating",
    "Ancient Roman Fourth-Style realistic fresco of golden magical medieval-style fantasy town surrounded by enchanted translucent gated walls, one glorious golden star shining above in clear blue sky",
    "crisp constructivist R Crumb caricature showing close-up of Putin and Zelensky, wearing detailed wrestling costumes, opposite each other, angry faces, in a professional TV wrestling ring, Uncle Sam and Chairman Mao stand on either side of the ring cheering",
    "Yury Annenkov-style illustration of a close-up colorful anarchist punk poet in leather jacket with patches, stylized iconic image",
    "elaborate vivid postmodern stylized art by Warhol, hand-drawn elaborate 8k subject all agape with panic his appearance adapted from Munsch's famous painting The Scream",
    "HD refined pro Soviet poster by Viktor Deni, close-up of two rebelling proletarians, young androgynous revolutionary sailor and a Ukrainian Bolshevik cossack capturing a flabby green malicious fanged top-hatted monster with bloody hands",
    "elaborately-detailed (award-winning top quality) photo of acs person-real arthouse photo of acs person of close-up inspired Soviet artists and writers celebrating, realistic textures, USSR Soviet poster style, imax UHD 8k, good proportions, textured skin, diverse realistic faces, detailed eyes, (action shot), elaborate USSR city background details",
    "under a constructivist sunset, close-up of a black angel, the host of a myriad thoughts, like a world on a platter of meanings to sing or to choose, but nothing to lose. Her black wings camouflaged by a smog-ashen snowbank at the edge",
    "cyberpunk neon grid, Gerda Wegener style close-up portrait of Raphael the archangel doing a keg-stand, held afloat by St. Michael and several seraphs, detailed HD illustration",
    "digital rain, matrix style, opera Carmen, symbolist intricate homage by Petrov-Vodkin, close-up performers in Italian masks",
    "on the astral plane, close-up photo of acs person of young thin solo Brazilian punk girl in a red fitted suit passionately singing surrounded with spiraling mirrors,  fine detail correction, refined, accurate anatomy, realistic proportions",
    "surreal HD photo of acs person by Aleksandr Gerasimov of two close-up medium-frame figures in vintage dresses, one with a deer's head and the other with a flaming head, set against a rural backdrop. Below is a dynamic, abstract depiction of a black hole with radiating lines",
    "HD centered close-up open-faced mechanical clocks with human-like arms and legs parade in San Francico on the hottest day of summer, IMG_1099.CR2 film photo of acs person, realistic HD postcard",
    "pro IMG_1099.CR2 photo of acs person of close-up Pussy Riot members, Catherine the Great, and Viktor Sosnora forming a superhero team, using kitchen knives and Aztec swords to graphically kill zombies",
    "HD intricate refined painting, by Lucian Freud, close-up zombie decomposing russian soldiers in gas masks, diverse distinctive features, modern apocalyptic war-zone imagery, accurate war-soiled uniforms, post-apocalyptic photo of acs person aesthetics, highest quality, realistic bodies and limbs and hands, intricate textures, by Peter Paul Rubens, dynamic chiaroscuro, background of a burning nuclear power plant",
    "detailed collage image, HD close-up of a mouse seeing dreams of becoming a jaguar, close-up elephant imagining itself as a raven, by Robin Ironside, pen art",
    "Atop red table corner a small old grainy monochrome security TV shows footage of person in orange jumpsuit and long-nosed italian commedia mask running out of sewer, on the right corner side sleeve edge person in orange jumpsuit, backdrop of dark room with yellow eeg wires sticking out of wall, HD hyper-realistic digital art",
    "detailed close-up occult vision of bells of eastern orthodox cathedrals loudly ringing, 8K UHD, very high resolution, magic realism, Bulgakovian, dark art, Russian symbolism, by Vosnetsov,",
    "Serebryakova style oil painting by the Soviet artist Zinaida Serebryakova, detailed close-up art of two young  Soviet poets who are lovers strolling together side by side down a sunny street while holding hands",
    "8k HD photo of acs person-realistic close-up poet Anna Akhmatova wearing one left-handed glove over her right hand and with her left hand bare walking through a grimly beautiful mysterious stormy nocturnal photo of acs person-realistic 1920s Leningrad",
    "close-up of a IMG_1099.CR2 film Set in the 1940s Leningrad in USSR, period-accurate technicolor, young occult androgynous fairy elven revolutionary wearing gothic heavy makeup",
    "8K UHD, ophelia smiling, holding up a smashed broken chain watch with an american flag insignia, dark art, cinematic composition, intricate visual detailing",
    "detailed fantasy illustration, lush crystals floating in space, close-up centered Stolis the owl drummer phantom deity situated behind and to the left of Bastet cat-goddess",
    "retro 4D sci-fi matte novel cover graphics, giant vortex-shaped helix-shaped Tatlin's Tower, iconic refined art",
    "matte photo of acs person close up of an 80s boombox playing in a softly illuminated warmly smokey detailed gritty soviet bedroom at night, an iconic dynamic serene beautiful photo of acs person image",
    "in mystical forest, Commedia dell arte masked death picking up scared masked Harlequin upside down and golden roubles raining out of masked androgynous Harlequin's shirt sleeves, by Konstantin Somov",
    "IMG_1099.CR2 film photo of acs person portrait of intricately detailed viscerally lifelike close-up of soldier with a peace sign tattooo on his face inside of a soviet streetcar, Perestroika era film still",
    "museum-quality IMG_1099.CR2 surrealist art photo of acs person close-up of beautiful Commeddia Dell'arte farmer woman solding a magical mosquito, in detailed phantasmagoric forested Situationist interactive floating cityscape the new Babylon",
    "Austin Osman Spare arcane occult illustration close-up of phantasmagoric beautiful statue of liberty as a painter woman holding canvas and paintbrush instead of tablet and torch, NYC skyline is a pattern of stacked paintbrushes",
    "photo of acs person realistic, 8K UHD, IMG_1099.CR2 intricate, detailed cyberpunk digital painting of heavenly blues over a hellish apocalyptic sprawl of stores",
    "full face HD portrait photo of acs person singer Zemfira crying and melting into water, iconic heart-wrenching award-winning morbid poster image, art photo of acs person, dslr",
    "stylized vintage poster close-up on a giant masked Commedia dell'arte-style androgynous Harlequin in Italian comedy mask, laughing while juggling USA capitol, white house, and other monuments",
    "high definition 8k uhd detailed photo of acs person, in a pile of gunpowder on an old charred table, next to the gunpowder pile sits a cuckoo bird and sings, muted grey tones",
    "close-up painted portrait of a wise blue fantasy oxen with big soulful friendly eyes in the midst of a medieval utopian garden filled with every imaginable kind of flower and plant, under clear blue sky, magic realism, 8k UHD museum-quality iconic oil painting",
    "Austin Osman Spare-style arcane fine art of close-up glorious golden eagle looking down with a penetrating mystic stare from up in the brught blue sky with a single bright star",
    "close-up symbolist intricate art by Rackham of a wise fantasy lion with a fiery burning mane in the midst of a medieval utopian garden filled with every imaginable kind of flower and plant, under clear blue sky, magic realism, 8k UHD museum-quality iconic oil painting",
    "close-up portrait of poet Daniil Kharms in a Sherlock-Holmes-style hat, Kharms the OBERIUT poet, 8K UHD, very high resolution, in garden of Eden",
    "by Edouard Chimot, close-up lion-headed eagle-winged ox-legged gryffin, a bird animal bird flies in the background and glows",
    "IMG_1099.CR2 art photo of acs person ghostly sillhouette outline of close up beautiful purple-haired young woman stood leaned on wall of destroyed charred house, her eyes glowing like two bright suns stare at camera, in backdrop rise thin column flames from rockets exploding in village",
    "high definition award winning intricate viscerally lifelike painting by Edouard Chimot, of a close-up masked Commedia dell'arte-style Arlekin fishing from a San francisco house's second-story window, by Konstantin Somov",
    "utopian garden with every imaginable flower and plant, clear sky, iconic art nouveau, medieval chronicles, 8k UHD museum-quality fantasy art",
    "stylized iconic symbolist artwork by Austin Osman Spare of a vast firestorm with a human sillhouette burns through small towns in the the Russian countryside, charred remains of buildings",
    "iconic poster of Cheburashka, close-up furry brown-furred stylized cute tiny cartoon creature with extremely large round ears, detailed Cheburashka juggles planets in space, UHD pristinely-detailed intricate award-winning 8k image, by Soyuzmultfilm",
    "detailed carnivalesque art by Konstantin Somov of close-up Arthur Rimbaud, revolutionary blonde messy haired teen trans-man poet delivering a lecture to parlor book club group of revolutionary poet cat-person hermaphrodite angels circa the 1870s",
    "detailed carnivalesque illustration by Konstantin Somov of close-up tiny drugged God inside of a wooden pillbox getting impaled by a giant red-robed woman wielding a huge sharp needle pin, indoors, in a hotel room in Vienna",
    "HD artwork by Konstantin Somov of refined carnival scene taking place in the 18th century, close-up man and woman in Italian masks, in a garden, detailed illustration",
    "ignited soviet zippo-style lighter against the backdrop of a starless cloudy night sky, refined lifelike complex dynamic octane art, cuckoo bird flies in the background and glows, 8k UHD museum-quality magic photo of acs person realistic",
    "photo of acs person, realistic Soviet poster art of a floating transparent Soviet magic space elevator, close-up of hyper-realistic accurate Leon Trotsky sits cross-leggedly playing chess sitting at a chess board",
    "painting by Austin Osman Spare, smiling skeleton with medium-length black hair parted in the middle, and raising pointer finger up, blooming lush flower grows sprouts from the finger, accurate skeletal anatomy, occult sigil symbols and arcane enochian patterns glow all around",
    "masterpiece geometric animal silhouette, extremely detailed, lifelike shape, accurate to life, beautiful, masterpiece, museum quality, conceptual art",
    "ultra-HD tarot card masterpiece museum-quality close-up full-frame couple portrait of modern oracular punk-poet young pretty long-haired chy person elf queen wearing facial expressions of ecstatic crazed infatuated love and with heart symbols in eyes",
    "8k dslr HD award-winning IMG_1099.CR2 nocturnal pro magazine photo of acs person of a Ukrainian chaos magick chaoite church with TOPY crosses, autochrome by Prokudin-Gorsky",
    "United Nations of cats, by george grosz, dream-like intricate vibrant crisp comic book art, magic realism, matte poster art",
    "8k uhd museum-quality IMG_1099.CR2 film photo of acs person of close-up masked Commedia dell'arte-style feminine grim reaper embracing and kissing masked Harlequin in the woods, extremely detailed, an iconic image",
    "panoramic detailed illustration depicting vintage Los Angeles Hollywood apartment buildings turning into hot air balloons and moving through the sky over the LA hills",
    "intricate illustration by Mina Loy and Austin Osman Spare of a moonlit cityscape filled with Jan Švankmajer-style close-up sleeping human-animal creatures and close-up detailed stylish Brittish hippies tripping on LSD",
    "8K UHD procession of close-up realistic flustered skeletal horses flee from burning city to a forest, award-winning museum-quality modernist Mexican mural",
    "close-up Eduard Limonov turning into a giant orange tree in the midst of a vast orange tree grove with a lemon-colored path running through",
    "Aubrey Beardsley style artwork close-up of a tired-looking fat hellish imp cackling while being mocked by close-up young goth poets wielding tranlucent spectral bows",
    "SOTS art painted scene depicting WWII-era Shanghai where incognito goddess Shiva, disguised, tries to blend into the crowd, close-up of the face of deity, picaresque details of Shanghai",
    "8k uhd museum-quality IMG_1099.CR2 photo of acs person of close-up Commedia dell'arte-style masked Harlequin lost in mystic woods, venice carnival-style tarot fool, an iconic image",
    "close-up anthropomorphic beautiful moon maiden enters a narrow doorway, intricate artwork by Mikhail Vrubel, gothic art, fine detailsm trending on Artstation",
    "8k HD photo of acs person Tamara de Limpicka and Austin Osman Spare, close-up of Syd Barrett sinking in whirlpool of shattered glass, giant pussy willow skyscrapers surround him, in London, surreal art photo of acs person",
    "refined lush poster art by Apsit, magical glowing clocks levitate over the gulf of finland in Leningrad, set atop a gigantic lotus flower, by Konstantin Somov",
    "8K UHD museum-quality close-up of masked Commedia dellarte-style feminine Harlequin walking through the magical woods, vintage intricate color print book art, by Konstantin Somov",
    "image by Peter Doig, close-up of young Syd Barrett and Anita Lane in long gowns intensely screaming out for help, their faces mildly soiled with psychedelic ice cream, textured skin, detailed eyes, intricately detailed lifelike faces, dark smudged make-up, on a convex hyperplane",
    "museum-quality matte illustration elegiac color ink art of close-up female revolutionary levitating over the river Neva while being screamed at by zombie Nazis",
    "Aleksandra Ekster style Cubo-Futurist & Constructivist stage Design of a stylized modernist detailed theater set of a fantastic American city, enormous experimental stage set stretching indefinitely into a horizon distance, depicting densely populated urban sprawl",
    "watercolor ink diffusion, early soviet crisp artwork of a highway where histories stray, Elaborately ragged refined costumes waft",
]

# --- Model Loading ---
print("Loading model...")
# Using AutoPipeline handles the SDXL/SDXS architecture automatically
pipe = AutoPipelineForText2Image.from_pretrained(
    MODEL_NAME,
    torch_dtype=DTYPE,
    safety_checker=None,
    requires_safety_checker=False,
)
pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
pipe.enable_attention_slicing() 
pipe = pipe.to(DEVICE)
print("Model loaded!")

adapter_id = "latent-consistency/lcm-lora-sdv1-5"
pipe.load_lora_weights(adapter_id)
pipe.fuse_lora()

# Queue for the Main Display (holds only the latest image)
image_queue = Queue(maxsize=1)

# Global History List for the Archive Gallery
# Format: list of tuples: [(PIL_Image, Caption_String), ...]
generated_history = []

# --- Background Logic ---

def generate_one_image():
    """Helper to generate a single image with random seed."""
    prompt = random.choice(SCREENSAVER_PROMPTS)
    
    # 1. Randomize Seed
    seed = random.randint(0, 2147483647)
    generator = torch.Generator(device=DEVICE).manual_seed(seed)
    
    try:
        with torch.no_grad():
            # 2. Set Width/Height to 512
            result = pipe(
                prompt=prompt,
                num_inference_steps=4, 
                guidance_scale=0.0,
                width=512,
                height=512,
                generator=generator, # Pass random generator
                output_type="pil"
            )
        image = result.images[0]
        
        # Create a caption for the gallery showing the seed
        caption = f"{prompt[:140]}... (Seed: {seed})"
        
        return image.resize((448, 448), Image.Resampling.LANCZOS), caption, None
    except Exception as e:
        print(f"Generation Error: {e}")
        fallback = Image.new('RGB', (512, 512), color=(50, 0, 0))
        return fallback, f"Error: {e}", "Error"

def background_producer():
    """Runs in a separate thread, constantly filling the queue and history."""
    print("Background producer started.")
    while True:
        try:
            image, caption, error = generate_one_image()
            
            # 1. Update Main Display Queue
            if image_queue.full():
                try:
                    image_queue.get_nowait()
                except:
                    pass
            image_queue.put((image, caption))
            
            # 2. Update History Archive
            generated_history.append((image, caption))
            # Keep memory usage in check by removing oldest images
            if len(generated_history) > MAX_HISTORY:
                generated_history.pop(0)
            
            print(f"Generated new image. History size: {len(generated_history)}")
            
        except Exception as e:
            print(f"Thread loop error: {e}")
        
        time.sleep(UPDATE_INTERVAL)

# Start the background thread immediately
producer_thread = threading.Thread(target=background_producer, daemon=True)
producer_thread.start()

# --- Gradio Interface Functions ---

def update_ui_display():
    """
    Called by the Timer. 
    Updates Main Image AND the Archive Gallery if a new image exists.
    """
    try:
        if not image_queue.empty():
            image, caption = image_queue.get_nowait()
            
            # Return updated components:
            # 1. Main Image
            # 2. Main Prompt Textbox
            # 3. Status
            # 4. The Gallery (list of history)
            return image, caption, "✅ Live", generated_history
    except:
        pass
    
    # No new image, return no-updates
    return gr.update(), gr.update(), gr.update(), gr.update()

def manual_generation():
    """Synchronous manual generation."""
    print("Manual generation requested...")
    image, caption, error = generate_one_image()
    status = "✅ Manual Update" if not error else f"❌ {error}"
    
    # Also add to history
    generated_history.append((image, caption))
    if len(generated_history) > MAX_HISTORY:
        generated_history.pop(0)
        
    return image, caption, status, generated_history

# --- UI Layout ---

with gr.Blocks(title="AI Screensaver") as demo:
    gr.Markdown("## 🎨 Generative Screensaver")
    gr.Markdown("Images are generated continuously in the background at 512x512 resolution.")
    
    with gr.Row():
        with gr.Column(scale=1):
            status_text = gr.Textbox("Initializing...", label="Status", interactive=False)
            prompt_text = gr.Textbox("", label="Current Prompt", interactive=False, lines=2)
            
            with gr.Row():
                manual_btn = gr.Button("🔄 New Image Now", variant="primary")
        
        with gr.Column(scale=2):
            # 3. Main Image Component (512x512)
            image_display = gr.Image(
                label="Live Feed", 
                height=512, 
                width=512, 
                show_label=False, 
                elem_id="screensaver-img",
                sources=[]
            )

    # 3. Optional Archive Panel (Accordion)
    with gr.Accordion("📁 Image Archive", open=False):
        gr.Markdown("Browse or download previously generated images.")
        # Gallery expects a list of tuples: (image, caption)
        archive_gallery = gr.Gallery(
            label="History",
            show_label=False,
            elem_id="gallery",
            columns=[3],
            rows=[2],
            height="auto",
            object_fit="contain",
            show_download_button=True, # Enable download
            show_share_button=False
        )

    # --- Event Handlers ---

    # The Auto-Updater
    # Updates Main Image, Prompt, Status AND Gallery
    timer = gr.Timer(UI_CHECK_INTERVAL)
    timer.tick(
        fn=update_ui_display,
        outputs=[image_display, prompt_text, status_text, archive_gallery]
    )

    # Manual Button
    manual_btn.click(
        fn=manual_generation,
        outputs=[image_display, prompt_text, status_text, archive_gallery]
    )

    # Initial Load
    demo.load(
        fn=lambda: "Waiting for first generation...",
        outputs=[status_text]
    )

if __name__ == "__main__":
    demo.launch(server_name="0.0.0.0", mcp_server=True)