🌌 Weaving Scene {current_story_obj.current_scene_number + 1}...
"), output_latest_scene_image: gr.Image(value=create_placeholder_image("🎨 Conjuring visuals..."), visible=True), output_latest_scene_narrative: gr.Markdown(value=" Musing narrative...", visible=True), engage_button: gr.Button(interactive=False), # Disable button surprise_button: gr.Button(interactive=False), } log_accumulator = [f"**🚀 Scene {current_story_obj.current_scene_number + 1} - {time.strftime('%H:%M:%S')}**"] if not scene_prompt_text.strip(): error_msg = "Scene prompt cannot be empty!" log_accumulator.append(f" VALIDATION ERROR: {error_msg}") yield { output_status_bar: gr.HTML(value=f"{error_msg}
"), output_interaction_log_markdown: gr.Markdown(value="\n".join(log_accumulator)), engage_button: gr.Button(interactive=True), surprise_button: gr.Button(interactive=True) } return current_story_obj, current_story_obj.get_all_scenes_for_gallery_display(), None, "## Error\n" + error_msg # Return existing state if error # --- 1. Generate Narrative Text --- # (This logic is the same as your previous `add_scene_to_story`) # ... # Example: # narrative_text_generated, text_gen_log = _generate_narrative_for_scene(...) # log_accumulator.extend(text_gen_log) # yield { ui updates } # This placeholder needs full logic. For brevity, I'm simulating. progress(0.1, desc="✍️ Crafting narrative...") time.sleep(0.5) # Simulate work narrative_text_generated = f"This is the AI-generated narrative for your idea: '{scene_prompt_text[:30]}...'. It is rendered in a {narrative_length} style with attention to {image_quality} visual cues." log_accumulator.append(f" Narrative: Generated using {text_model_key}. Length: {narrative_length}") yield { output_latest_scene_narrative: gr.Markdown(value=f"## Scene Idea: {scene_prompt_text}\n\n{narrative_text_generated}"), output_interaction_log_markdown: gr.Markdown(value="\n".join(log_accumulator)) } # --- 2. Generate Image --- # (This logic is the same as your previous `add_scene_to_story`) # ... # Example: # image_generated_pil, image_gen_log = _generate_image_for_scene(...) # log_accumulator.extend(image_gen_log) # yield { ui updates } progress(0.5, desc="🎨 Conjuring visuals...") time.sleep(1) # Simulate work image_generated_pil = create_placeholder_image(f"Image for:\n{scene_prompt_text[:25]}...\nStyle: {image_style_dropdown}", text_color="#A020F0") image_generation_error_message = None # Assume success for this WOW demo part log_accumulator.append(f" Image: Generated using {image_provider_key}. Style: {image_style_dropdown}") yield { output_latest_scene_image: gr.Image(value=image_generated_pil), output_interaction_log_markdown: gr.Markdown(value="\n".join(log_accumulator)) } # --- 3. Add Scene to Story Object --- # (This logic is the same as your previous `add_scene_to_story`) current_story_obj.add_scene_from_elements( user_prompt=scene_prompt_text, narrative_text=narrative_text_generated, image=image_generated_pil, image_style_prompt=f"{image_style_dropdown}{f', by {artist_style_text}' if artist_style_text else ''}", image_provider=image_provider_key, error_message=image_generation_error_message ) # --- 4. Prepare Final Outputs for Gradio --- gallery_items_tuples = current_story_obj.get_all_scenes_for_gallery_display() _ , latest_narr_for_display = current_story_obj.get_latest_scene_details_for_display() # Image is already in its component status_message_html = "🌌 Scene Woven! Your StoryVerse expands...
" if image_generation_error_message: status_message_html = "Scene added, but image generation had issues.
" log_accumulator.append(f" Scene {current_story_obj.current_scene_number} successfully added to story object.") progress(1.0, desc="Scene Complete!") yield { output_status_bar: gr.HTML(value=status_message_html), story_state_output: current_story_obj, # Update the state output_gallery: gr.Gallery(value=gallery_items_tuples, visible=True), # Latest image and narrative already updated progressively output_interaction_log_markdown: gr.Markdown(value="\n".join(log_accumulator)), engage_button: gr.Button(interactive=True), surprise_button: gr.Button(interactive=True), } # Return a dictionary that maps to the output components explicitly for the final state # This ensures Gradio handles the final update correctly. return { story_state_output: current_story_obj, output_gallery: gallery_items_tuples, output_latest_scene_image: image_generated_pil, # Ensure this is the final image output_latest_scene_narrative: latest_narr_for_display, output_status_bar: status_message_html, output_interaction_log_markdown: "\n".join(log_accumulator) } def clear_story_state_ui_wrapper(): # (This logic is the same as your previous clear_story_state_ui) new_story = Story() placeholder_img = create_placeholder_image("Your StoryVerse is a blank canvas...", color="#1A1A2E") cleared_gallery = [(placeholder_img, "Your StoryVerse is new and untold...")] initial_narrative = "## ✨ A New Story Begins ✨\nDescribe your first scene idea in the panel to the left and let the AI help you weave your world!" status_msg = "📜 Story Cleared. A fresh canvas awaits your imagination!
" return { story_state_output: new_story, output_gallery: cleared_gallery, output_latest_scene_image: None, # Clear latest image output_latest_scene_narrative: gr.Markdown(value=initial_narrative), output_status_bar: gr.HTML(value=status_msg), output_interaction_log_markdown: "Log Cleared. Ready for a new adventure!", scene_prompt_input: "" # Clear the input prompt } def surprise_me_func(): # Simple "Surprise Me" feature themes = ["Sci-Fi", "Fantasy", "Mystery", "Slice of Life", "Historical Fiction"] actions = ["discovers a hidden map", "encounters a mysterious stranger", "solves an ancient riddle", "embarks on a perilous journey", "attends a secret festival"] settings = ["in a bustling alien marketplace", "within a forgotten, vine-covered temple", "aboard a steampunk airship", "in a quiet, magical forest", "during a solar eclipse on a twin-mooned planet"] prompt = f"A character {random.choice(actions)} {random.choice(settings)}. Theme: {random.choice(themes)}." style = random.choice(list(STYLE_PRESETS.keys())) artist = random.choice(["Greg Rutkowski", "Makoto Shinkai", "Moebius", "Rebecca Guay", ""]*3) # "" for no artist sometimes return prompt, style, artist # --- Gradio UI Definition --- with gr.Blocks(theme=omega_theme, css=omega_css, title="✨ StoryVerse Omega ✨ - AI Story & World Weaver") as story_weaver_demo: # Output components need to be defined before the click handler if updated by yield's dict story_state_output = gr.State(Story()) # Crucial: Define actual output component for state gr.Markdown("STORYVERSE_...) are correctly set in Space Secrets!⚠️ CRITICAL: NO AI SERVICES.
") else: if text_llm_ok: status_text_list.append("✅ Text Gen Ready.
") else: status_text_list.append("⚠️ Text Gen NOT Ready.
") if image_gen_ok: status_text_list.append("✅ Image Gen Ready.
") else: status_text_list.append("⚠️ Image Gen NOT Ready.
") gr.HTML("".join(status_text_list)) with gr.Row(equal_height=False, variant="panel"): with gr.Column(scale=7, min_width=450): # Input panel gr.Markdown("### 💡 **Craft Your Scene**", elem_classes="input-section-header") with gr.Group(): scene_prompt_input = gr.Textbox(lines=7, label="Scene Vision (Description, Dialogue, Action):", placeholder="e.g., Amidst swirling cosmic dust, Captain Eva pilots her damaged starfighter towards a colossal, ringed gas giant. Alarms blare. 'Just a little further,' she mutters, gripping the controls.") with gr.Row(elem_classes=["compact-row"]): with gr.Column(scale=2): image_style_input = gr.Dropdown(choices=["Default (Cinematic Realism)"] + sorted(list(STYLE_PRESETS.keys())), value="Default (Cinematic Realism)", label="Visual Style Preset") with gr.Column(scale=2): artist_style_input = gr.Textbox(label="Artist Inspiration (Optional):", placeholder="e.g., Moebius, Zdzisław Beksiński") negative_prompt_input = gr.Textbox(lines=2, label="Exclude from Image (Negative Prompt):", placeholder="Default exclusions applied. Add more if needed.", value=COMMON_NEGATIVE_PROMPTS) with gr.Accordion("⚙️ Advanced AI Configuration", open=False): with gr.Group(): text_model_dropdown = gr.Dropdown(choices=list(TEXT_MODELS.keys()), value=UI_DEFAULT_TEXT_MODEL_KEY, label="Narrative AI Engine") image_provider_dropdown = gr.Dropdown(choices=list(IMAGE_PROVIDERS.keys()), value=UI_DEFAULT_IMAGE_PROVIDER_KEY, label="Visual AI Engine") with gr.Row(): narrative_length_dropdown = gr.Dropdown(["Short (1 paragraph)", "Medium (2-3 paragraphs)", "Detailed (4+ paragraphs)"], value="Medium (2-3 paragraphs)", label="Narrative Detail") image_quality_dropdown = gr.Dropdown(["Standard", "High Detail", "Sketch Concept"], value="Standard", label="Image Detail/Style") with gr.Row(elem_classes=["compact-row"], equal_height=True): engage_button = gr.Button("🌌 Weave This Scene!", variant="primary", scale=3, icon="✨") surprise_button = gr.Button("🎲 Surprise Me!", variant="secondary", scale=1, icon="🎁", elem_id="surprise_button") clear_story_button = gr.Button("🗑️ New Story", variant="stop", scale=1, icon="♻️") # Stop variant for caution output_status_bar = gr.HTML(value="Ready to weave your first masterpiece!
") with gr.Column(scale=10, min_width=700): # Output panel gr.Markdown("### 🖼️ **Your Evolving StoryVerse**", elem_classes="output-section-header") with gr.Tabs(elem_id="output_tabs_elem"): with gr.TabItem("🌠 Latest Scene", id="latest_scene_tab", elem_id="latest_scene_tab_item"): with gr.Row(): output_latest_scene_image = gr.Image(label="Latest Scene Image", type="pil", interactive=False, show_download_button=True, height=512, show_label=False, elem_classes=["panel_image"]) output_latest_scene_narrative = gr.Markdown(elem_id="latest_scene_narrative_md") with gr.TabItem("📚 Story Scroll", id="story_scroll_tab", elem_id="story_scroll_tab_item"): output_gallery = gr.Gallery(label="Story Scroll", show_label=False, columns=4, object_fit="cover", height=700, preview=True, allow_preview=True, elem_classes=["gallery_output"]) # More columns with gr.TabItem("⚙️ Interaction Log", id="log_tab", elem_id="log_tab_item"): with gr.Accordion(label="Developer Interaction Log", open=False): # Default closed output_interaction_log_markdown = gr.Markdown("Log will appear here...") # Define outputs that the wrapper function will update via yield # Note: story_state_output is the gr.State component itself. outputs_for_handler = [ output_status_bar, gr.Accordion(visible=True), # Placeholder for output_initial_solutions_accordion (not used in this UI) gr.Markdown(visible=True), # Placeholder for output_initial_solutions_markdown (not used) gr.Accordion(visible=True), # Placeholder for output_champion_accordion (not used) gr.Markdown(visible=True), # Placeholder for output_champion_markdown (not used) gr.Accordion(visible=True), # Placeholder for output_evolved_accordion (not used) output_evolved_markdown, # This can be repurposed or removed if not directly evolving scenes output_ai_test_analysis_markdown, # Also can be repurposed or removed output_interaction_log_markdown, engage_button, surprise_button, # To re-enable surprise button story_state_output, # This is where the story_state is passed back to itself output_gallery, output_latest_scene_image, output_latest_scene_narrative ] # Re-map the orchestrator's yielded dict keys to the actual output components for the final return # This is a bit of a hack due to Gradio's yield behavior vs. final return. # The `add_scene_to_story_orchestrator` should be designed to yield a dict # that maps to these specific component names if we want direct updates. # For simplicity, the `return` statement of the orchestrator is what matters for final state. engage_button.click( fn=add_scene_to_story_orchestrator, # Call the wrapper inputs=[ story_state_output, scene_prompt_input, image_style_input, artist_style_input, negative_prompt_input, text_model_dropdown, image_provider_dropdown, narrative_length_dropdown, image_quality_dropdown # New inputs ], outputs=[ # These must match the keys in the dictionary returned by the orchestrator's FINAL return story_state_output, output_gallery, output_latest_scene_image, output_latest_scene_narrative, output_status_bar, output_interaction_log_markdown # Note: The progressive yield updates specific components by name directly. ] ) clear_story_button.click( fn=clear_story_state_ui_wrapper, inputs=[], outputs=[ # Must match the keys in the dict returned by clear_story_state_ui_wrapper story_state_output, output_gallery, output_latest_scene_image, output_latest_scene_narrative, output_status_bar, output_interaction_log_markdown, scene_prompt_input # To clear the input field ] ) surprise_button.click( fn=surprise_me_func, inputs=[], outputs=[scene_prompt_input, image_style_input, artist_style_input] ) gr.Examples( examples=[ ["A lone, weary traveler on a mechanical steed crosses a vast, crimson desert under twin suns. Dust devils dance in the distance.", "Sci-Fi Western", "Moebius", "greenery, water, modern city"], ["Deep within an ancient, bioluminescent forest, a hidden civilization of sentient fungi perform a mystical ritual around a pulsating crystal.", "Psychedelic Fantasy", "Alex Grey", "technology, buildings, roads"], ["A child sits on a crescent moon, fishing for stars in a swirling nebula. A friendly space whale swims nearby.", "Whimsical Cosmic", "James Jean", "realistic, dark, scary"], ["A grand, baroque library where the books fly freely and whisper forgotten lore to those who listen closely.", "Magical Realism", "Remedios Varo", "minimalist, simple, technology"] ], inputs=[scene_prompt_input, image_style_input, artist_style_input, negative_prompt_input], # negative_prompt is optional for examples label="🌌 Example Universes to Weave 🌌", ) gr.HTML("✨ StoryVerse Omega™ - Weaving Worlds with Words and Pixels ✨