Spaces:

Yassaman
/

OurCache

Sleeping

App Files Files Community

Yassaman commited on Dec 2, 2025

Commit

7e7d8c0

verified ·

1 Parent(s): f600a49

Upload 4 files

Browse files

Files changed (4) hide show

Dockerfile +18 -0
app.py +256 -0
prompts.json +164 -0
requirements.txt +2 -0

Dockerfile ADDED Viewed

	@@ -0,0 +1,18 @@

+# Use Python 3.9
+FROM python:3.9
+# Set working directory
+WORKDIR /code
+# Copy requirements and install
+COPY ./requirements.txt /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+# Copy all files (videos, py, json) to the container
+COPY . .
+# Make sure folder permissions are correct
+RUN chmod -R 777 /code
+# Run the app using Gunicorn on port 7860 (Hugging Face default)
+CMD ["gunicorn", "-b", "0.0.0.0:7860", "app:app"]

app.py ADDED Viewed

	@@ -0,0 +1,256 @@

+import json
+import os
+import re
+from flask import Flask, render_template_string, send_from_directory
+# --- CONFIGURATION ---
+# Make sure these match your actual folder names exactly!
+FOLDERS = {
+    'baseline': 'baseline',
+    'teacache': 'teacache',
+    'magcache': 'magcache',
+    'ours':     'ours'
+}
+PROMPT_FILE = 'prompts.json'
+# --- LOGIC ---
+app = Flask(__name__)
+BASE_DIR = os.path.abspath(os.path.dirname(__file__))
+def get_prompts():
+    path = os.path.join(BASE_DIR, PROMPT_FILE)
+    if not os.path.exists(path):
+        return []
+    with open(path, 'r', encoding='utf-8') as f:
+        try:
+            content = json.load(f)
+            if isinstance(content, list): return content
+        except:
+            f.seek(0)
+            lines = [line.strip() for line in f if line.strip()]
+            return lines
+def find_file_by_id(folder, video_id):
+    dir_path = os.path.join(BASE_DIR, folder)
+    if not os.path.exists(dir_path): return None
+    for fname in os.listdir(dir_path):
+        if fname.startswith(video_id) and fname.lower().endswith('.mp4'):
+            return fname
+    return None
+def build_tasks():
+    tasks = []
+    prompts = get_prompts()
+    baseline_path = os.path.join(BASE_DIR, FOLDERS['baseline'])
+    if not os.path.exists(baseline_path):
+        print("Baseline folder missing!")
+        return []
+    baseline_files = sorted([f for f in os.listdir(baseline_path) if f.lower().endswith('.mp4')])
+    id_regex = re.compile(r'^(\d{4})')
+    for i, b_file in enumerate(baseline_files):
+        match = id_regex.match(b_file)
+        if not match: continue
+        vid_id = match.group(1)
+        prompt_text = prompts[i] if i < len(prompts) else "(No prompt)"
+        tea_file = find_file_by_id(FOLDERS['teacache'], vid_id)
+        mag_file = find_file_by_id(FOLDERS['magcache'], vid_id)
+        our_file = find_file_by_id(FOLDERS['ours'], vid_id)
+        tasks.append({
+            "index": i + 1,
+            "id": vid_id,
+            "prompt": prompt_text,
+            "baseline": f"videos/{FOLDERS['baseline']}/{b_file}",
+            "teacache": f"videos/{FOLDERS['teacache']}/{tea_file}" if tea_file else "",
+            "magcache": f"videos/{FOLDERS['magcache']}/{mag_file}" if mag_file else "",
+            "ours":     f"videos/{FOLDERS['ours']}/{our_file}" if our_file else ""
+        })
+    print(f"Loaded {len(tasks)} videos.")
+    return tasks
+TASKS = build_tasks()
+# --- HTML TEMPLATE ---
+HTML = """
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>2x2 Visualizer</title>
+    <script src="https://cdn.tailwindcss.com"></script>
+    <style>
+        body { background-color: #0f172a; color: #e2e8f0; font-family: sans-serif; }
+        /* Darker theme to make videos pop */
+        .video-box {
+            background: #1e293b;
+            padding: 0.5rem;
+            border-radius: 0.5rem;
+            height: 100%;
+            display: flex;
+            flex-direction: column;
+        }
+        video {
+            width: 100%;
+            height: auto;
+            border-radius: 0.25rem;
+            background: #000;
+            flex-grow: 1;
+        }
+        .label {
+            text-align: center;
+            font-weight: bold;
+            font-size: 1rem;
+            margin-bottom: 0.5rem;
+            color: #94a3b8;
+            text-transform: uppercase;
+            letter-spacing: 0.05em;
+        }
+        /* Highlight 'Ours' */
+        .highlight { border: 2px solid #3b82f6; background: #1e3a8a; }
+        .highlight .label { color: #60a5fa; }
+        .prompt-box {
+            background: #1e293b;
+            border-bottom: 1px solid #334155;
+            padding: 1rem;
+            position: sticky;
+            top: 0;
+            z-index: 50;
+        }
+    </style>
+</head>
+<body class="h-screen flex flex-col overflow-hidden">
+    <div class="prompt-box flex flex-col md:flex-row gap-4 items-center flex-none shadow-lg">
+        <div class="flex-none w-full md:w-48 space-y-2">
+            <div class="flex justify-between items-center text-xs text-gray-400 font-mono">
+                <span>ID: <span id="vid-id" class="text-white">--</span></span>
+                <span><span id="idx" class="text-white">0</span> / <span id="total">0</span></span>
+            </div>
+            <div class="flex gap-2">
+                <button onclick="move(-1)" class="flex-1 bg-gray-700 hover:bg-gray-600 text-white py-2 rounded font-bold text-sm transition">Prev</button>
+                <button onclick="move(1)" class="flex-1 bg-blue-600 hover:bg-blue-500 text-white py-2 rounded font-bold text-sm transition">Next</button>
+            </div>
+        </div>
+        <div class="flex-grow overflow-y-auto max-h-20 w-full">
+            <p id="prompt-text" class="text-sm md:text-lg text-gray-200 leading-snug font-light">Loading...</p>
+        </div>
+    </div>
+    <div class="flex-grow overflow-y-auto p-4">
+        <div class="w-full h-full max-w-[1920px] mx-auto">
+            <div class="grid grid-cols-1 md:grid-cols-2 gap-4 h-full">
+                <div class="video-box">
+                    <div class="label">Baseline</div>
+                    <video id="v-baseline" controls loop muted playsinline></video>
+                </div>
+                <div class="video-box">
+                    <div class="label">TeaCache</div>
+                    <video id="v-teacache" controls loop muted playsinline></video>
+                </div>
+                <div class="video-box">
+                    <div class="label">MagCache</div>
+                    <video id="v-magcache" controls loop muted playsinline></video>
+                </div>
+                <div class="video-box highlight">
+                    <div class="label">Ours</div>
+                    <video id="v-ours" controls loop muted playsinline></video>
+                </div>
+            </div>
+        </div>
+    </div>
+    <script>
+        const tasks = {{ tasks_json | safe }};
+        let current = 0;
+        const els = {
+            prompt: document.getElementById('prompt-text'),
+            id: document.getElementById('vid-id'),
+            idx: document.getElementById('idx'),
+            total: document.getElementById('total'),
+            v1: document.getElementById('v-baseline'),
+            v2: document.getElementById('v-teacache'),
+            v3: document.getElementById('v-magcache'),
+            v4: document.getElementById('v-ours')
+        };
+        els.total.innerText = tasks.length;
+        function loadTask(index) {
+            if (tasks.length === 0) return;
+            if (index < 0) index = tasks.length - 1;
+            if (index >= tasks.length) index = 0;
+            current = index;
+            const t = tasks[current];
+            els.prompt.innerText = t.prompt;
+            els.id.innerText = t.id;
+            els.idx.innerText = t.index;
+            els.v1.src = t.baseline;
+            els.v2.src = t.teacache;
+            els.v3.src = t.magcache;
+            els.v4.src = t.ours;
+            [els.v1, els.v2, els.v3, els.v4].forEach(v => {
+                if(v.src) {
+                    v.currentTime = 0;
+                    v.play().catch(e => {});
+                }
+            });
+        }
+        function move(dir) { loadTask(current + dir); }
+        document.addEventListener('keydown', e => {
+            if (e.key === "ArrowLeft") move(-1);
+            if (e.key === "ArrowRight") move(1);
+        });
+        if (tasks.length > 0) loadTask(0);
+        else els.prompt.innerText = "No tasks found. Check console.";
+    </script>
+</body>
+</html>
+"""
+@app.route('/')
+def home():
+    return render_template_string(HTML, tasks_json=json.dumps(TASKS))
+@app.route('/videos/<folder>/<path:filename>')
+def serve_file(folder, filename):
+    if folder in FOLDERS:
+        return send_from_directory(os.path.join(BASE_DIR, FOLDERS[folder]), filename)
+    return "Error", 404
+if __name__ == '__main__':
+    print("Starting server...")
+    app.run(port=5001, debug=True)

prompts.json ADDED Viewed

	@@ -0,0 +1,164 @@

+A car driving in a highway and passing a cross road.
+A young girl with extraordinary musical talent has never been able to produce a sound. On the first day, she hears an old man playing ancient tunes in the market, feeling her soul deeply moved. The next day, the old man gives her a silver flute and tells her that it will allow her to express the voice deep within her. On the third day, she sits by a stream playing the flute, and the notes flow like water, causing the flowers around her to bloom. On the fourth day, she feels a strange power as her music seems to change people’s emotions, calming them down. On the fifth day, when she plays the final tune, she herself emits a sound, realizing that she has always sung with her soul, and the notes were her true language.
+Two anthropomorphic cats in comfy boxing gear and bright gloves fight intensely on a spotlighted stage.
+Summer beach vacation style, a white cat wearing sunglasses sits on a surfboard. The fluffy-furred feline gazes directly at the camera with a relaxed expression. Blurred beach scenery forms the background featuring crystal-clear waters, distant green hills, and a blue sky dotted with white clouds. The cat assumes a naturally relaxed posture, as if savoring the sea breeze and warm sunlight. A close-up shot highlights the feline's intricate details and the refreshing atmosphere of the seaside.
+A cat walks on the grass, realistic style.
+A cat wearing sunglasses.
+A dog playing with a blue ball in the park.
+A cat walks on the grass, realistic style.
+Woman during a sunset on a rooftop, Silhouette of a woman standing on top of a city, while raising her arms to the air, with the sunset sun just ahead.
+Rancher smoking at sunset, Person wearing a cowboy hat and denim jacket sits on a wooden fence smoking a cigarette, silhouetted in front of the sun.
+Woman in an infinity pool at sunset, Cinemagraph of the silhouette of a woman standing and palm trees and umbrellas in an infinity pool at sunset.
+Aerial landscape around a highway at sunset, Slow overflight over a long straight road with cars, which crosses an immense forest with abundant trees and vegetation, in a view of the sunset in the distant mountains.
+two cats playing with each other on a stage, cartoon style.
+two dogs playing with each other on a stage, cartoon style.
+A street photographer rapidly adjusts their camera settings to capture a fleeting moment of joy in a passing crowd.
+An intense argument between two people escalating quickly, involving sharp gestures and shifting body language.
+A professional dancer executes a gravity-defying leap followed by a perfectly controlled landing.
+A close-up of a human face transitioning from deep sleep to sudden, wide-eyed terror.
+A person meticulously attempts to tie a complex, tangled knot under high-pressure conditions.
+A chef rapidly chops vegetables with incredible speed and precision, using a large, sharp knife.
+A continuous shot of a person running a full marathon, showing their gradual exhaustion and determination.
+A mother and child playing catch, with the child misjudging the throw and stumbling in the grass.
+A detailed shot of a musician's fingers flawlessly playing a complex piece on a piano keyboard.
+A mime performing a detailed pantomime of struggling against a strong, invisible wind.
+A sudden reaction: a group of friends laughing uncontrollably after hearing a surprising joke.
+A person struggling to lift a heavy box, showing the strain and wobble in their muscles.
+A highly skilled surgeon performing a delicate procedure with steady, focused hand movements.
+A person attempting to juggle five objects, with one object occasionally slipping and being recovered mid-air.
+An emotional embrace between two long-lost friends, involving rapid changes in facial expression.
+A bartender fluidly mixing and pouring multiple cocktails simultaneously with flair.
+A person is startled by a noise in a dark room, spinning around quickly, light flashing across their face.
+A crowd of fans erupts in cheers and synchronized celebration after a winning goal in a stadium.
+A close-up of a human artist rapidly sketching a detailed portrait with charcoal and quick strokes.
+A person performing a backflip off a high diving board, twisting and entering the water cleanly.
+A rapid sequence of a human learning a new physical skill, moving from awkward attempts to mastery.
+A person experiencing extreme cold, shivering violently and rubbing their hands together for warmth.
+A tense moment where a poker player slowly reveals their cards, showing subtle tells and anxiety.
+A detailed shot of a professional archer drawing, holding, and releasing an arrow with flawless form.
+A group of children having a wild pillow fight, with feathers flying and chaotic movement.
+A person desperately trying to open a jammed door, pulling and twisting the handle with force.
+A slow-motion sequence of a runner crossing the finish line, pushing their body to the absolute limit.
+An elderly person gently struggling to button a small coat button with arthritic hands.
+A close-up of a human eye shedding a single, perfect tear that slowly rolls down the cheek.
+A person receiving a shocking piece of news, going through denial, anger, and finally acceptance.
+A martial artist performing a complex, fast-paced Kata with powerful strikes and quick stances.
+A human quickly assembling a complicated piece of flat-pack furniture, showing moments of confusion and clarity.
+A sequence of a person giving CPR, demonstrating realistic pressure and rhythm.
+A crowd performing a wave at a stadium, with individual participants moving up and down seamlessly.
+A person falling gracefully and catching themselves just before hitting the ground.
+A detailed shot of a tattoo artist's needle moving rapidly and precisely across the skin.
+A theatrical scene where a human whispers a secret to another, who instantly reacts with gasping shock.
+A group of hikers crossing a difficult, unstable rope bridge high above a canyon.
+A street magician manipulating a deck of cards with blindingly fast shuffles and flourishes.
+A close-up of a person biting into a very sour lemon, showing the puckered reaction.
+A person trying to hold back laughter during a formal, serious meeting, subtly shaking.
+A choreographed fight scene where two characters exchange rapid, block-and-strike movements.
+A person climbing a vertical rock face, searching for handholds and securing their position.
+A waiter carrying a massive, unstable stack of plates across a crowded room without dropping any.
+A detailed shot of a person quickly reading a complex document, their eyes rapidly scanning lines of text.
+A person playing a virtual reality game, their body reacting realistically to unseen digital events.
+A rapid transition of a person's hairstyle being cut and styled into a completely new look.
+A human interacting with a complex hologram, swiping and manipulating 3D data in the air.
+A person attempting to extinguish a small fire with a blanket, showing panicked but effective movements.
+A group of dancers performing a synchronized routine in perfect, fluid harmony.
+A pedestrian suddenly steps into a crosswalk, and an autonomous vehicle smoothly brakes to a halt.
+A delivery robot navigates a crowded city sidewalk, skillfully avoiding multiple human pedestrians.
+A human driver in a traditional car attempts to merge into heavy traffic while an autonomous car maintains its lane.
+An autonomous shuttle picks up and drops off passengers at a busy airport terminal, interacting with luggage carts.
+A child chases a ball into the street, and both a human-driven car and an autonomous car react simultaneously.
+A human cyclist swerves to avoid a pothole, causing an autonomous vehicle behind them to adjust its path.
+A driverless car performs parallel parking on a bustling street, watched by curious onlookers.
+A human flags down an autonomous taxi, which then pulls over and waits for them to enter.
+A sudden torrential downpour begins, affecting visibility for both human and autonomous vehicles on a highway.
+An autonomous vehicle navigates a construction zone, carefully avoiding workers and moving machinery.
+A human street performer is juggling fire at an intersection, and an autonomous car slows down to pass.
+A car with a human driver aggressively cuts off an autonomous vehicle, which calmly maintains its distance.
+A group of friends interacts with the touchscreen interface inside a futuristic autonomous car while it drives.
+An autonomous delivery drone descends to drop off a package to a person waiting on their doorstep.
+A human driver gives a thumbs-up to an autonomous car that successfully yielded for them at an intersection.
+A dense fog rolls in, challenging the sensors of an autonomous truck as it pulls a heavy load.
+An autonomous vehicle assists a human in changing a flat tire on the side of a busy road.
+A driverless car drops off a passenger at a drive-through restaurant, and the human collects their order.
+A human-driven motorcycle weaves through traffic, and an autonomous car adjusts its speed and spacing.
+A person riding an electric scooter quickly crosses in front of an autonomous vehicle.
+A snowstorm hits, covering the road in ice and snow, and autonomous vehicles continue to navigate cautiously.
+An autonomous bus picks up a group of elderly passengers, carefully deploying a ramp for accessibility.
+A human jogger overtakes a slow-moving autonomous cleaning robot on a park pathway.
+A human passenger falls asleep in a fully autonomous car, which continues to drive safely through varying conditions.
+An autonomous vehicle encounters a fallen tree blocking the road and automatically reroutes.
+A police officer directs traffic at a busy intersection, and both human and autonomous cars follow their signals.
+A human driver reacts with frustration to a perceived error by an autonomous vehicle.
+An autonomous street sweeper navigates a city street, cleaning around parked cars and pedestrians.
+A group of tourists takes selfies with an autonomous shuttle passing by a famous landmark.
+A human driver changes lanes without signaling, and an autonomous car reacts with a slight swerve.
+An autonomous farming tractor cultivates a field while a farmer watches from the side.
+A sudden gust of wind blows debris onto the road, and an autonomous car swerves to avoid it.
+A human family is having a picnic in a park, and an autonomous delivery drone flies overhead.
+An autonomous valet parking system parks a car in a multi-story garage without human intervention.
+A human crossing guard stops traffic, allowing children to cross, while an autonomous car waits patiently.
+A low-light scenario where streetlights flicker, and autonomous vehicles rely on advanced sensors.
+A human on a bicycle signals a turn, and an autonomous car accurately predicts their movement.
+A high-speed chase where an autonomous police vehicle pursues a human-driven getaway car.
+An autonomous vehicle detects a sick person on the sidewalk and pulls over to offer assistance.
+A human driver accidentally veers into the lane of an autonomous car, which safely avoids a collision.
+An autonomous robot dog walks alongside a human owner in a bustling urban environment.
+A sudden tire blowout on a human-driven car, and a nearby autonomous vehicle provides assistance.
+A human mechanic performs diagnostics on an autonomous vehicle, interacting with its internal systems.
+An autonomous vehicle drives through a tunnel where GPS signals are lost, relying on other sensors.
+A human-operated forklift loads cargo onto an autonomous truck in a busy warehouse.
+A street vendor sets up their stall, causing an autonomous vehicle to slightly alter its route.
+A human waves to an autonomous car as it passes, acknowledging its presence.
+An autonomous vehicle detects an animal running into the road and brakes safely.
+A human tests the emergency override system of an autonomous car during a simulated critical event.
+An autonomous snowplow clears a road during a heavy snowfall, navigating around parked cars and pedestrians.
+A grand, ornate chandelier slowly detaches from the ceiling, swinging precariously before crashing dramatically onto a polished marble floor.
+A dense forest undergoes a rapid seasonal change from vibrant autumn colors to a stark, snow-covered winter landscape.
+A futuristic city skyline is suddenly engulfed by a rapidly expanding swarm of bioluminescent drones, forming shifting patterns.
+A historical battlefield comes to life with hundreds of soldiers engaged in close-quarters combat, dust and smoke filling the air.
+A delicate sandcastle on a beach is slowly eroded by the incoming tide, its intricate details dissolving into the waves.
+A complex Rube Goldberg machine activates, with each elaborate step triggering the next in a fluid, continuous motion.
+An ancient stone statue slowly cracks and crumbles, revealing a vibrant, glowing energy source within.
+A powerful tornado tears through a suburban neighborhood, lifting houses and debris into the churning vortex.
+A detailed underwater scene where a school of fish swims in perfect synchronicity, reacting to a passing shark.
+A barren desert landscape is rapidly transformed by a sudden, intense flash flood, carving new channels.
+A crowded train station experiences a sudden blackout, plunging commuters into darkness before emergency lights flicker on.
+A magician performs a levitation trick, with a human subject slowly rising into the air, defying gravity.
+A controlled demolition of a large building, showing the structure collapsing inwards with realistic dust and debris.
+A magnificent fireworks display illuminates a night sky over a tranquil lake, with vibrant reflections on the water.
+A detailed diorama of a miniature town is brought to life, with tiny cars moving and minuscule people walking.
+A calm ocean surface is suddenly disturbed by a massive whale breaching and splashing back down.
+A close-up of a complex clockwork mechanism where all gears, springs, and levers are in precise, constant motion.
+A person walking through a field of tall grass, which sways realistically with each step and passing breeze.
+A vibrant coral reef ecosystem where various fish, anemones, and other marine life interact in a dynamic dance.
+A sudden volcanic eruption spews molten lava and ash into the sky, flowing down the mountain.
+A high-speed pursuit through narrow city streets, with vehicles swerving, near-misses, and dynamic camera angles.
+A detailed model ship in a bottle is hit by a sudden, miniature storm, with tiny waves rocking the vessel.
+A time-lapse of a massive ice glacier calving, with huge chunks of ice breaking off and falling into the sea.
+A grand theatrical stage where props and backdrops shift and transform seamlessly between acts.
+A group of parkour athletes navigates a complex urban environment, leaping and climbing with incredible agility.
+A close-up of a delicate spiderweb catching dew drops, which then shimmer and slide down the strands.
+A massive swarm of locusts descends upon a field of crops, devouring everything in its path.
+A slow-motion shot of colorful paint being mixed and swirled together, creating intricate, ephemeral patterns.
+A futuristic factory floor where robotic arms assemble complex machinery with precision and speed.
+A serene garden pond suddenly comes alive with a multitude of frogs leaping and birds splashing.
+A detailed shot of a human hand manipulating a complex puzzle cube at high speed.
+A powerful waterfall with mist rising, and a lone figure standing at its base, seemingly unfazed.
+A crowded marketplace where vendors are selling goods and people are haggling, creating a bustling atmosphere.
+A close-up of a bonfire, with flames dancing, sparks flying, and logs slowly crumbling into embers.
+A medieval siege where a battering ram repeatedly strikes a castle gate, causing splinters and cracks.
+A rapid change in weather from bright sunshine to a dark, stormy sky with lightning flashes.
+A group of acrobats performing a complex routine, with fluid movements and intricate formations.
+A detailed shot of water droplets rolling off a hydrophobic surface, coalescing and moving independently.
+A subterranean cave system with glowing crystals and underground rivers, constantly shifting and revealing new paths.
+A flock of pigeons suddenly takes flight from a city square, scattering in all directions.
+A large crowd at a music concert reacting dynamically to the rhythm of the performance.
+A close-up of a flower blooming in fast-motion, its petals unfurling in exquisite detail.
+A rapid transition from an arid desert to a lush, green oasis appearing as if by magic.
+A group of children playing an intense game of tag in a park, weaving between trees and playground equipment.
+A detailed interior of a spaceship during a sudden emergency, with alarms blaring and systems flickering.
+A close-up of a waterfall where individual water molecules can be seen cascading over rocks.
+A detailed shot of a blacksmith hammering glowing hot metal, with sparks flying and steam rising.
+A slow-motion explosion of a colorful piñata, releasing candy and confetti into the air.
+A serene forest clearing where a deer grazes peacefully, but its ears constantly twitch, alert to danger.
+A massive school of fish forms intricate, dynamic shapes in the ocean, resembling a living sculpture.

requirements.txt ADDED Viewed

	@@ -0,0 +1,2 @@


1	+ flask
2	+ gunicorn