update gradio UI

app.py

@@ -152,35 +152,24 @@ def fetch_context(topic):
 
 # Define Gradio Handler
 import time
-import gradio as gr
-
+from datetime import datetime
 def generate_blog(topic, tone, progress=gr.Progress(track_tqdm=True)):
-    # initialize progress bar
     progress(0, desc="Starting collaborative report generation...")
     yield "### 🧭 Starting Collaborative Report\n"
 
-    # Step 1️⃣ Research Lead
-    progress(0.25, desc="Research Lead generating outline...")
-    plan_result = crew.tasks[0].run(inputs={"topic": topic, "tone": tone})
-    yield f"### 🧠 Research Lead (Outline)\n{plan_result.output_text}\n\n---\n"
-    time.sleep(0.3)
-
-    # Step 2️⃣ Technical Writer
-    progress(0.50, desc="Technical Writer drafting content...")
-    write_result = crew.tasks[1].run(inputs={"topic": topic, "tone": tone}, context=[plan_result])
-    yield f"### ✍️ Technical Writer (Draft)\n{write_result.output_text}\n\n---\n"
-    time.sleep(0.3)
-
-    # Step 3️⃣ Data Verifier
-    progress(0.75, desc="Fact-checking underway...")
-    fact_result = crew.tasks[2].run(inputs={"topic": topic, "tone": tone}, context=[write_result])
-    yield f"### 🔍 Data Verifier (Fact-Checking)\n{fact_result.output_text}\n\n---\n"
-    time.sleep(0.3)
-
-    # Step 4️⃣ Editor-in-Chief
-    progress(1.0, desc="Final editing and polish...")
-    edit_result = crew.tasks[3].run(inputs={"topic": topic, "tone": tone}, context=[fact_result])
-    yield f"### 🧾 Editor-in-Chief (Final Blog)\n{edit_result.output_text}\n\n✅ **Report Complete!**"
+    # Run the entire crew sequentially and stream outputs
+    stream = crew.kickoff_stream(inputs={"topic": topic, "tone": tone})
+
+    # Stream each agent’s output as it arrives
+    for event in stream:
+        role = event.get("agent", "Agent")
+        output = event.get("output", "")
+        if output:
+            timestamp = datetime.now().strftime("[%H:%M:S]")
+            yield f"{timestamp} 🧩 **{role}**\n{output}\n\n---\n"
+
+    progress(1.0, desc="Finalizing report...")
+    yield "✅ **All agents have completed their work! Final blog generated.**"
 
 # Build Gradio Interface
 css = """
@@ -237,7 +226,7 @@ with gr.Blocks(css=css) as demo:
     context_output = gr.Markdown(label="📚 Retrieved Context Summary") # Rag Search
 
     run_button = gr.Button("🚀 Generate Blog", variant="primary")
-    final_output = gr.Textbox(label="📰 Generated Blog Post", elem_id="output-box")
+    final_output = gr.Textbox(label="📰 Generated Blog Post", elem_id="output-box", lines=30, interactive=False, show_label=True)
 
     fetch_btn.click(fetch_context, inputs=[topic], outputs=[context_output]) # Rag Search
     run_button.click(generate_blog, inputs=[topic, tone], outputs=[final_output])