feat: Production deployment release

.dockerignore

@@ -0,0 +1,9 @@
+__pycache__
+*.pyc
+.git
+.env
+venv
+.venv
+node_modules
+.DS_Store
+outputs/

.gitignore

@@ -1,3 +1,9 @@
 .env
 __pycache__
 venv/
+.venv/
+outputs/
+*.pyc
+.DS_Store
+*.mp4
+*.png

Dockerfile

@@ -0,0 +1,9 @@
+FROM python:3.10-slim
+WORKDIR /app
+RUN apt-get update && apt-get install -y libgl1-mesa-glx libglib2.0-0 && rm -rf /var/lib/apt/lists/*
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+COPY . .
+RUN mkdir -p outputs && chmod 777 outputs
+EXPOSE 7860
+CMD ["uvicorn", "server:app", "--host", "0.0.0.0", "--port", "7860"]

README.md

@@ -1 +1,32 @@
-# Continuity
+---
+title: Continuity
+emoji: 🎬
+colorFrom: purple
+colorTo: gray
+sdk: docker
+app_port: 7860
+---
+
+# Continuity
+
+AI-powered video post-production tool that automatically generates seamless bridging transitions between two video clips using Gemini 1.5 Flash (Director) and Wan 2.2 (Cinematographer).
+
+## How it Works
+
+Continuity behaves like a professional film crew inside your browser, orchestrating a multi-step workflow to bridge scenes:
+
+1.  **Ingest**: You upload your starting clip (Scene A) and ending clip (Scene C).
+2.  **Director Node (Gemini 1.5 Flash)**:
+    *   Analyzes the visual content, motion, and lighting of both clips.
+    *   Drafts a creative direction for the transition (Scene B) to ensure narrative consistency.
+3.  **Cinematographer Node (Wan 2.2)**:
+    *   Takes the Director's prompt and generates the bridging video content.
+    *   Ensures the transition matches the style and tone of the input footage.
+4.  **Assembly**: The system stitches the generated bridge with the original clips (logic handled by the agent) and presents the final result.
+
+## Usage
+
+1.  Upload **Scene A** (Start) and **Scene C** (End).
+2.  Add optional **Director Notes** (e.g., "Slow dissolve", "Cyber glitch").
+3.  Click **Generate Transition**.
+4.  Preview the generated bridge generated by the AI agent.

continuity_agent/agent.py

@@ -29,7 +29,7 @@ class ContinuityState(TypedDict):
 
 # Node 1: Analyst
 def analyze_videos(state: ContinuityState) -> dict:
-    print("--- 🧐 Analyst Node (Director) ---")
+    print("--- Analyst Node (Director) ---")
     
     video_a_url = state['video_a_url']
     video_c_url = state['video_c_url']
@@ -103,7 +103,7 @@ def analyze_videos(state: ContinuityState) -> dict:
 
 # Node 2: Generator (Wan 2.2 First Last Frame)
 def generate_video(state: ContinuityState) -> dict:
-    print("--- 🎥 Generator Node (Wan 2.2) ---")
+    print("--- Generator Node (Wan 2.2) ---")
     
     prompt = state.get('veo_prompt', "")
     path_a = state.get('video_a_local_path')

server.py

@@ -1,17 +1,15 @@
 from fastapi import FastAPI, HTTPException, UploadFile, Form, File
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
 import uvicorn
 import os
 import shutil
 import uuid
-
-# Import from the subpackage as before
 from continuity_agent.agent import app as continuity_graph
 
 app = FastAPI(title="Continuity", description="AI Video Bridging Service")
 
-# 1. Enable CORS
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -20,11 +18,14 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# 2. Setup Static Files for Outputs
 OUTPUT_DIR = "outputs"
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 app.mount("/outputs", StaticFiles(directory=OUTPUT_DIR), name="outputs")
 
+@app.get("/")
+async def read_root():
+    return FileResponse("stitch_continuity_dashboard/code.html")
+
 @app.post("/generate-transition")
 async def generate_transition(
     video_a: UploadFile = File(...),
@@ -32,29 +33,21 @@ async def generate_transition(
     prompt: str = Form("Cinematic transition")
 ):
     try:
-        # Generate unique ID for this request
         request_id = str(uuid.uuid4())
-        
-        # Save inputs
-        # Preserve extension
         ext_a = os.path.splitext(video_a.filename)[1] or ".mp4"
         ext_c = os.path.splitext(video_c.filename)[1] or ".mp4"
-        
+
         path_a = os.path.join(OUTPUT_DIR, f"{request_id}_a{ext_a}")
         path_c = os.path.join(OUTPUT_DIR, f"{request_id}_c{ext_c}")
         
         with open(path_a, "wb") as buffer:
             shutil.copyfileobj(video_a.file, buffer)
-            
         with open(path_c, "wb") as buffer:
             shutil.copyfileobj(video_c.file, buffer)
             
-        # Initialize State with LOCAL PATHS
-        # We don't need URLs for the new logic, but state definition might map them.
-        # agent.py logic checks video_a_local_path if present.
         initial_state = {
-            "video_a_url": "local_upload", # Placeholder
-            "video_c_url": "local_upload", 
+            "video_a_url": "local_upload",
+            "video_c_url": "local_upload",
             "user_notes": prompt,
             "veo_prompt": prompt,
             "video_a_local_path": os.path.abspath(path_a),
@@ -63,28 +56,20 @@ async def generate_transition(
             "status": "started"
         }
         
-        # Invoke Agent
         result = continuity_graph.invoke(initial_state)
-        
-        # The agent returns 'generated_video_url' which is a local absolute path (e.g., from tempfile or cache)
-        # We need to copy/move this to our STATIC directory to serve it.
         gen_path = result.get("generated_video_url")
         
         if not gen_path or "Error" in gen_path:
             raise HTTPException(status_code=500, detail=f"Generation failed: {gen_path}")
             
-        # Copy generated video to outputs
         final_filename = f"{request_id}_bridge.mp4"
         final_output_path = os.path.join(OUTPUT_DIR, final_filename)
+        shutil.move(gen_path, final_output_path)
         
-        shutil.copy(gen_path, final_output_path)
-        
-        # Return URL relative to server root
         return {"video_url": f"/outputs/{final_filename}"}
-
     except Exception as e:
         print(f"Server Error: {e}")
         raise HTTPException(status_code=500, detail=str(e))
 
 if __name__ == "__main__":
-    uvicorn.run("server:app", host="0.0.0.0", port=8000, reload=True)
+    uvicorn.run("server:app", host="0.0.0.0", port=7860, reload=False)

stitch_continuity_dashboard/code.html

@@ -173,7 +173,7 @@
             Bridge</span>
         </div>
         <!-- Active Card Container -->
-        <div
+        <div id="bridge-card"
           class="relative w-full aspect-[4/3] rounded-2xl overflow-hidden shadow-neon border border-primary/30 group">
           <!-- Placeholder Gradient Background -->
           <div class="absolute inset-0 bg-gradient-to-br from-surface-dark to-[#0f0a16] z-0"></div>
@@ -218,7 +218,7 @@
       </div>
       <!-- SCENE C: Upload State -->
       <div class="flex flex-col gap-4 flex-1 max-w-[320px] group">
-        <div class="flex items-center justify-between px-1">
+        <div class="flex items-center justify-center px-1">
           <span class="text-xs font-bold tracking-widest text-primary">SCENE C</span>
           <span class="text-xs font-bold tracking-widest text-gray-400 uppercase">Target Source</span>
         </div>
@@ -299,7 +299,7 @@
 
         try {
           // 3. Send to Server
-          const response = await fetch("http://127.0.0.1:8000/generate-transition", {
+          const response = await fetch("/generate-transition", {
             method: "POST",
             body: formData
           });
@@ -309,32 +309,13 @@
           const data = await response.json();
 
           // 4. Success: Inject Video into the Middle Card
-          // We target the middle "Bridge" div
-          const bridgeCard = document.querySelector(".aspect-\\[4\\/3\\]");
-          // Find the middle card specifically if possible, relying on the class is risky if multiple match.
-          // But in this layout, the middle one is distinct or we can target by structure. 
-          // The middle card is the second .aspect-[4/3] roughly, but the user script uses querySelector which matches the first one?
-          // Wait, the Code.html has 3 cards.
-          // Scene A: .aspect-[4/3]
-          // Bridge: .aspect-[4/3] (line 148)
-          // Scene C: .aspect-[4/3]
-
-          // User script used document.querySelector(".aspect-\\[4\\/3\\]"). 
-          // This matches the first one (Scene A)! That's a bug in the user's snippet.
-          // I will fix it to target the BRIDGE card.
-          // The bridge card container has text "The Bridge".
-          // I'll add an ID to the bridge container or select by index.
-          // Safer to select the middle one: document.querySelectorAll(".aspect-\\[4\\/3\\]")[1]
+          const bridgeCard = document.getElementById("bridge-card");
 
-          // Target the middle "Bridge" div (index 1)
-          const cards = document.querySelectorAll(".aspect-\\[4\\/3\\]");
-          if (cards.length >= 2) {
-            // Scene A is [0], Bridge is [1], Scene C is [2]
-            const bridgeCard = cards[1];
-            // Ensure we are replacing the inner content
+          if (bridgeCard) {
+            // Ensure we are replacing the inner content completely
             bridgeCard.innerHTML = `
                 <video controls autoplay loop class="w-full h-full object-cover rounded-2xl border-2 border-primary shadow-neon">
-                  <source src="http://127.0.0.1:8000${data.video_url}" type="video/mp4">
+                  <source src="${data.video_url}" type="video/mp4">
                   Your browser does not support the video tag.
                 </video>
               `;

tests/create_assets.py

@@ -0,0 +1,23 @@
+import cv2
+import numpy as np
+import os
+
+def make_video(filename, color):
+    height, width = 480, 640
+    # Use 'mp4v' for .mp4 containers
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+    out = cv2.VideoWriter(filename, fourcc, 30.0, (width, height))
+    
+    # Create a frame with the solid color
+    frame = np.zeros((height, width, 3), dtype=np.uint8)
+    frame[:] = color # BGR format
+    
+    # Write 30 frames (1 second)
+    for _ in range(30):
+        out.write(frame)
+    out.release()
+    print(f"Created {filename}")
+
+if __name__ == "__main__":
+    make_video('tests/scene_a.mp4', (255, 0, 0)) # Blue
+    make_video('tests/scene_c.mp4', (0, 0, 255)) # Red

tests/test_api.py

@@ -0,0 +1,34 @@
+import requests
+import os
+import sys
+
+def test_api():
+    url = "http://127.0.0.1:8000/generate-transition"
+    
+    video_a_path = 'tests/scene_a.mp4'
+    video_c_path = 'tests/scene_c.mp4'
+    
+    if not os.path.exists(video_a_path) or not os.path.exists(video_c_path):
+        print("Error: Test videos not found. Run create_assets.py first.")
+        sys.exit(1)
+
+    files = {
+        'video_a': ('scene_a.mp4', open(video_a_path, 'rb'), 'video/mp4'),
+        'video_c': ('scene_c.mp4', open(video_c_path, 'rb'), 'video/mp4')
+    }
+    data = {'prompt': 'Test transition'}
+
+    print(f"Sending POST request to {url}...")
+    try:
+        response = requests.post(url, files=files, data=data)
+        print(f"Status Code: {response.status_code}")
+        if response.status_code == 200:
+            print(f"Success! Response: {response.json()}")
+        else:
+            print(f"Failed. Response: {response.text}")
+    except Exception as e:
+        print(f"Connection Failed: {e}")
+        print("Make sure the server is running on port 8000.")
+
+if __name__ == "__main__":
+    test_api()