request

app.py

@@ -26,6 +26,23 @@ SCENE_COLUMNS = ["Scene", "Title", "Action", "Visuals", "Characters", "Duration
 CHARACTER_COLUMNS = ["ID", "Name", "Role", "Traits"]
 
 
+def _character_dropdown_update(board: Storyboard | None):
+    if not board or not board.characters:
+        return gr.Dropdown.update(choices=[], value=None, interactive=False)
+    choices = [character.identifier for character in board.characters]
+    return gr.Dropdown.update(choices=choices, value=choices[0], interactive=True)
+
+
+def _gallery_from_board(board: Storyboard) -> List[Tuple[str, str]]:
+    gallery: List[Tuple[str, str]] = []
+    for character in board.characters:
+        if not character.reference_image:
+            continue
+        caption = f"{character.name} — {character.role}"
+        gallery.append((character.reference_image, caption))
+    return gallery
+
+
 def _ensure_storyboard(board: Storyboard | None) -> Storyboard:
     if not board:
         raise gr.Error("Create a storyboard first.")
@@ -43,7 +60,7 @@ def handle_storyboard(
     style: str,
     scene_count: int,
     google_api_key: str,
-) -> Tuple[str, List[List[str]], List[List[str]], Storyboard]:
+) -> Tuple[str, List[List[str]], List[List[str]], Storyboard, gr.Dropdown]:
     _validate_inputs(idea, inspiration_image)
     generator = StoryGenerator(api_key=google_api_key or None)
     storyboard = generator.generate(
@@ -55,11 +72,13 @@ def handle_storyboard(
     summary_md = f"### {storyboard.title}\n{storyboard.synopsis}"
     scene_rows = storyboard.scenes_table()
     character_rows = storyboard.characters_table()
+    dropdown_update = _character_dropdown_update(storyboard)
     return (
         summary_md,
         [[row[col] for col in SCENE_COLUMNS] for row in scene_rows],
         [[row[col] for col in CHARACTER_COLUMNS] for row in character_rows],
         storyboard,
+        dropdown_update,
     )
 
 
@@ -69,12 +88,32 @@ def handle_character_design(
 ):
     board = _ensure_storyboard(storyboard)
     designer = CharacterDesigner(api_key=google_api_key or None)
-    gallery, updated_board = designer.design(board)
+    _, updated_board = designer.design(board)
+    gallery = _gallery_from_board(updated_board)
     if not gallery:
         raise gr.Error("Failed to design characters.")
     return gallery, updated_board
 
 
+def handle_character_regen(
+    storyboard: Storyboard | None,
+    character_id: str | None,
+    google_api_key: str,
+):
+    board = _ensure_storyboard(storyboard)
+    if not character_id:
+        raise gr.Error("Select a character ID to regenerate.")
+    designer = CharacterDesigner(api_key=google_api_key or None)
+    try:
+        _, updated_board = designer.redesign_character(board, character_id)
+    except ValueError as exc:
+        raise gr.Error(str(exc)) from exc
+    gallery = _gallery_from_board(updated_board)
+    if not gallery:
+        raise gr.Error("Failed to refresh character art.")
+    return gallery, updated_board
+
+
 def handle_video_render(
     storyboard: Storyboard | None,
     hf_token: str,
@@ -155,26 +194,41 @@ with gr.Blocks(fill_height=True, elem_id="cinegen-app") as demo:
     scenes_df = gr.Dataframe(headers=SCENE_COLUMNS, wrap=True)
     characters_df = gr.Dataframe(headers=CHARACTER_COLUMNS, wrap=True)
 
-    storyboard_btn.click(
-        fn=handle_storyboard,
-        inputs=[idea_box, inspiration, style_dropdown, scene_slider, google_key_input],
-        outputs=[summary_md, scenes_df, characters_df, story_state],
-    )
-
     with gr.Row():
         design_btn = gr.Button("Design Characters", variant="secondary")
         render_btn = gr.Button("Render Short Film", variant="primary")
 
+    with gr.Row():
+        character_select = gr.Dropdown(
+            label="Character Slot",
+            choices=[],
+            interactive=False,
+            info="Select an ID from the storyboard table to regenerate its portrait.",
+        )
+        regen_btn = gr.Button("Regenerate Selected Character", variant="secondary")
+
     gallery = gr.Gallery(label="Character References", columns=4, height=320)
     render_logs = gr.Markdown(label="Render Log")
     final_video = gr.Video(label="CineGen Short Film", interactive=False)
 
+    storyboard_btn.click(
+        fn=handle_storyboard,
+        inputs=[idea_box, inspiration, style_dropdown, scene_slider, google_key_input],
+        outputs=[summary_md, scenes_df, characters_df, story_state, character_select],
+    )
+
     design_btn.click(
         fn=handle_character_design,
         inputs=[story_state, google_key_input],
         outputs=[gallery, story_state],
     )
 
+    regen_btn.click(
+        fn=handle_character_regen,
+        inputs=[story_state, character_select, google_key_input],
+        outputs=[gallery, story_state],
+    )
+
     def _model_value(label: str) -> str:
         lookup = dict(VIDEO_MODEL_CHOICES)
         return lookup.get(label, VIDEO_MODEL_CHOICES[0][1])
@@ -186,6 +240,8 @@ with gr.Blocks(fill_height=True, elem_id="cinegen-app") as demo:
         fn=render_wrapper,
         inputs=[story_state, hf_token_input, video_model_dropdown],
         outputs=[final_video, render_logs],
+        queue=True,
+        concurrency_limit=1,
     )
 
 if __name__ == "__main__":