Update app.py

app.py

@@ -1,10 +1,10 @@
-
 # --- app.py ---
 import json
 import math
 import gradio as gr
 from PIL import Image
-import random
+import os
+from utils.keyframe_utils import generate_keyframe_prompt
 
 # Load segments JSON
 def load_segments():
@@ -12,20 +12,23 @@ def load_segments():
         segments = json.load(f)
     return segments
 
-# Simulated keyframe image generator (use local placeholder or generate random colors)
-def get_placeholder_image(seed):
-    random.seed(seed)
-    img = Image.new("RGB", (256, 144), color=(random.randint(100,255), random.randint(100,255), random.randint(100,255)))
-    return img
+# Load pre-generated keyframe images (3 per segment)
+def get_keyframe_images(segment_id):
+    images = []
+    for i in range(1, 4):
+        img_path = f"keyframes/segment_{segment_id}_v{i}.png"
+        if os.path.exists(img_path):
+            images.append(Image.open(img_path))
+        else:
+            images.append(Image.new("RGB", (256, 144), color=(200, 200, 200)))
+    return images
 
 # UI component for a single segment row
 def segment_display(segment):
     description = segment["description"]
     seg_id = segment["segment_id"]
     row = [f"Segment {seg_id}", description]
-    for i in range(3):
-        img = get_placeholder_image(seed=(seg_id * 10 + i))
-        row.append(img)
+    row.extend(get_keyframe_images(seg_id))
     return row
 
 # Pagination logic
@@ -60,12 +63,10 @@ def build_interface():
         prev_btn.click(fn=lambda p: max(1, p - 1), inputs=page_state, outputs=page_state).then(update, inputs=page_state, outputs=[table, total_pages_text, page_state])
         next_btn.click(fn=lambda p: p + 1, inputs=page_state, outputs=page_state).then(update, inputs=page_state, outputs=[table, total_pages_text, page_state])
 
-        # Initial load
         demo.load(fn=update, inputs=page_state, outputs=[table, total_pages_text, page_state])
 
     return demo
 
-# Launch app
 if __name__ == "__main__":
     demo = build_interface()
     demo.launch()