Update app.py

app.py

@@ -7,6 +7,11 @@ import gradio as gr
 from cinegen import CharacterDesigner, StoryGenerator, VideoDirector
 from cinegen.models import Storyboard
 
+try:  # pragma: no cover - spaces is only available inside HF Spaces
+    import spaces  # type: ignore
+except Exception:  # pragma: no cover - keep local dev working without spaces pkg
+    spaces = None  # type: ignore
+
 STYLE_CHOICES = [
     "Cinematic Realism",
     "Neo-Noir Animation",
@@ -26,6 +31,15 @@ SCENE_COLUMNS = ["Scene", "Title", "Action", "Visuals", "Characters", "Duration
 CHARACTER_COLUMNS = ["ID", "Name", "Role", "Traits"]
 
 
+def gpu_guard(duration: int = 120):
+    def decorator(fn):
+        if not spaces:
+            return fn
+        return spaces.GPU(duration=duration)(fn)
+
+    return decorator
+
+
 def _character_dropdown_update(board: Storyboard | None):
     if not board or not board.characters:
         return gr.update(choices=[], value=None, interactive=False)
@@ -114,6 +128,7 @@ def handle_character_regen(
     return gallery, updated_board
 
 
+@gpu_guard(duration=300)
 def handle_video_render(
     storyboard: Storyboard | None,
     hf_token: str,