Upload app.py

app.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Dict, List
+from typing import Dict, Iterator, List, Tuple
 
 import gradio as gr
 
@@ -32,7 +32,7 @@ def update_play_options(state_label: str):
 
 def toggle_online_focus(is_enabled: bool):
     """Toggle visibility of online focus textbox."""
-    return gr.Textbox(visible=is_enabled)
+    return gr.update(visible=is_enabled)
 
 
 def workflow_markdown() -> str:
@@ -145,6 +145,98 @@ def _generate_play_plan_output_wrapper(
     )
 
 
+def _generate_task_output_stream(
+    task_group: str,
+    task_type: str,
+    state_labels: List[str],
+    objective: str,
+    audience_notes: str,
+    channels: List[str],
+    tone: str,
+    output_format: str,
+    idea_count: int,
+    include_kpis: bool,
+    search_online: bool,
+    online_focus: str,
+    temperature: float,
+) -> Iterator[Tuple[str, str, str, dict]]:
+    """Stream task generation progress to UI before final content arrives."""
+    yield (
+        "",
+        "Generating report... this usually takes 1 to 3 minutes.",
+        "Collecting report snippets and source trace...",
+        gr.update(interactive=False),
+    )
+    try:
+        report_code, report_markdown, source_markdown = _generate_task_output_wrapper(
+            task_group=task_group,
+            task_type=task_type,
+            state_labels=state_labels,
+            objective=objective,
+            audience_notes=audience_notes,
+            channels=channels,
+            tone=tone,
+            output_format=output_format,
+            idea_count=idea_count,
+            include_kpis=include_kpis,
+            search_online=search_online,
+            online_focus=online_focus,
+            temperature=temperature,
+        )
+    except Exception as exc:
+        report_code = ""
+        report_markdown = f"Generation failed: {exc}"
+        source_markdown = ""
+    yield (
+        report_code,
+        report_markdown,
+        source_markdown,
+        gr.update(interactive=True),
+    )
+
+
+def _generate_play_plan_output_stream(
+    state_label: str,
+    play_title: str,
+    objective: str,
+    constraints: str,
+    horizon: str,
+    budget_level: str,
+    search_online: bool,
+    online_focus: str,
+    temperature: float,
+) -> Iterator[Tuple[str, str, str, dict]]:
+    """Stream play-plan generation progress to UI before final content arrives."""
+    yield (
+        "",
+        "Generating play plan... this usually takes 1 to 3 minutes.",
+        "Collecting report snippets and source trace...",
+        gr.update(interactive=False),
+    )
+    try:
+        report_code, report_markdown, source_markdown = _generate_play_plan_output_wrapper(
+            state_label=state_label,
+            play_title=play_title,
+            objective=objective,
+            constraints=constraints,
+            horizon=horizon,
+            budget_level=budget_level,
+            search_online=search_online,
+            online_focus=online_focus,
+            temperature=temperature,
+        )
+    except Exception as exc:
+        report_code = ""
+        report_markdown = f"Generation failed: {exc}"
+        source_markdown = ""
+    yield (
+        report_code,
+        report_markdown,
+        source_markdown,
+        gr.update(interactive=True),
+    )
+
+
 def build_app() -> gr.Blocks:
     default_group = next(iter(TASK_GROUPS))
     default_task = TASK_GROUPS[default_group][0]
@@ -189,6 +281,7 @@ def build_app() -> gr.Blocks:
                         label="Online search focus (optional)",
                         placeholder="e.g., latest homeowner grant updates and insurer discount guidance",
                         visible=False,
+                        value="",
                     )
                     clear_chat = gr.Button("Clear chat")
                     related_charts = gr.Gallery(
@@ -219,7 +312,7 @@ def build_app() -> gr.Blocks:
                     ask_chat = gr.Button("Ask", variant="primary")
                     # chat_sources = gr.Code(label="Source trace", language="markdown", lines=10, show_label=True)
                     chat_sources = gr.Markdown(label="Source trace", value="Sources will appear here.", padding=True)
-            chat_search_online.change(toggle_online_focus, inputs=chat_search_online, outputs=chat_online_focus)
+            chat_search_online.change(fn=toggle_online_focus, inputs=chat_search_online, outputs=chat_online_focus, show_progress="hidden")
             ask_chat.click(
                 fn=_ask_report_chat_wrapper,
                 inputs=[chat_input, chatbox, chat_states, chat_search_online, chat_online_focus],
@@ -303,19 +396,20 @@ def build_app() -> gr.Blocks:
                         label="Online search focus (optional)",
                         placeholder="e.g., latest Louisiana grant updates and insurer discount guidance",
                         visible=False,
+                        value="",
                     )
                     generate_task = gr.Button("Generate", variant="primary")
 
-                with gr.Column(scale=5):
+                with gr.Column(scale=5, ):
                     task_output_m = gr.Markdown(label="Report", value="Generated output will appear here.", padding=True)
                     task_output = gr.Code(label="Report", value="Generated output will appear here.",language="markdown", lines=13, show_label=True, visible=False,)
                     gr.Markdown("---")
                     task_sources = gr.Markdown(label="Source trace", value="Sources will appear here.", padding=True)
 
             task_group.change(update_task_type_options, inputs=task_group, outputs=task_type)
-            search_online.change(toggle_online_focus, inputs=search_online, outputs=online_focus)
+            search_online.change(fn=toggle_online_focus, inputs=search_online, outputs=online_focus, show_progress="hidden")
             generate_task.click(
-                fn=_generate_task_output_wrapper,
+                fn=_generate_task_output_stream,
                 inputs=[
                     task_group,
                     task_type,
@@ -331,7 +425,8 @@ def build_app() -> gr.Blocks:
                     online_focus,
                     temperature,
                 ],
-                outputs=[task_output, task_output_m, task_sources],
+                outputs=[task_output, task_output_m, task_sources, generate_task],
+                show_progress="full",
             )
 
         with gr.Tab("Playbook Planner"):
@@ -374,6 +469,7 @@ def build_app() -> gr.Blocks:
                         label="Online search focus (optional)",
                         placeholder="e.g., best practices for contractor co-marketing in Louisiana",
                         visible=False,
+                        value="",
                     )
                     play_temperature = gr.Slider(
                         minimum=0.0,
@@ -394,9 +490,9 @@ def build_app() -> gr.Blocks:
                     gr.Markdown("---")
                     play_sources = gr.Markdown(label="Source trace", value="Sources will appear here.", padding=True)
             state_single.change(update_play_options, inputs=state_single, outputs=play_selector)
-            play_search_online.change(toggle_online_focus, inputs=play_search_online, outputs=play_online_focus)
+            play_search_online.change(fn=toggle_online_focus, inputs=play_search_online, outputs=play_online_focus, show_progress="hidden")
             generate_play.click(
-                fn=_generate_play_plan_output_wrapper,
+                fn=_generate_play_plan_output_stream,
                 inputs=[
                     state_single,
                     play_selector,
@@ -408,7 +504,8 @@ def build_app() -> gr.Blocks:
                     play_online_focus,
                     play_temperature,
                 ],
-                outputs=[play_output, play_output_m, play_sources],
+                outputs=[play_output, play_output_m, play_sources, generate_play],
+                show_progress="full",
             )
 
 
@@ -428,4 +525,4 @@ demo = build_app()
 
 
 if __name__ == "__main__":
-    demo.queue(default_concurrency_limit=4).launch(theme=gr.themes.Soft(), css=CUSTOM_CSS,)
+    demo.queue(default_concurrency_limit=10).launch(theme=gr.themes.Soft(), css=CUSTOM_CSS,)