update gradio UI

app.py

@@ -26,7 +26,7 @@ hybrid_tool = HybridRetrieverTool(alpha=0.6) #including RAG in search
 #Creating Agents
 
 planner = Agent(
-    role="Content Planner",
+    role="Research Lead",
     goal="Plan engaging and factually accurate content on {topic}",
     backstory="You are working on planning a blog article about the topic {topic}."
     "Use the retriever tool to gather accurate, recent information before outlining." #RAG search
@@ -39,10 +39,10 @@ planner = Agent(
 )
 
 writer = Agent(
-    role="Content Writer",
+    role="Technical Writer",
     goal="Write an insightful and factually accurate opinion piece about the topic: {topic}",
     backstory="You are working on writing a new opnion piece about the topic: {topic}. "
-    "You base your writing on the work of the Content Planner, who provides an outline and relevant context about the topic. "
+    "You base your writing on the work of the research lead, who provides an outline and relevant context about the topic. "
     "You follow the main objectives and direction of the outline, as provided by the Content Planner. "
     "You also provide objective and impartial insights, supported by relevant information that is provided by the Content Planner. "
     "You acknowledge in your opnion piece when your statements are opinions as opposed to objective staements.",
@@ -52,7 +52,7 @@ writer = Agent(
 )
 
 fact_checker = Agent(
-    role="Fact Checker",
+    role="Data Verifier",
     goal="Verify factual accuracy, detect unsupported claims and identify missing references or sources.",
     backstory="You are a meticulous research analyst who checks every claim against known facts and relaible sources"
     "Use the retriever tool to cross-check the Content Writer's statements against reliable, recent information.",
@@ -63,7 +63,7 @@ fact_checker = Agent(
 )
 
 editor = Agent(
-    role="Editor",
+    role="Editor-in-Chief",
     goal="Edit a given blog post to align with the context and narrative style of the {topic} (e.g. academic, informal, satirical).",
     backstory="You are an editor who receives a blog post from the Content Writer. "
     "Your goal is to review the blog post to ensure that it follows journalistic best practices, "
@@ -151,19 +151,33 @@ def fetch_context(topic):
     return summary
 
 # Define Gradio Handler
-def generate_blog(topic, tone):
-    yield "⏳ Generating blog — this may take a few moments..."
-    result = crew.kickoff(inputs={"topic": topic, "tone": tone})
-    
-# CrewOutput -> extract text
-    if hasattr(result, "raw_output"):
-        final_text = result.raw_output
-    elif hasattr(result, "final_output"):
-        final_text = result.final_output
-    else:
-        final_text = str(result)
+import time
+import gradio as gr
+
+def generate_blog(topic, tone, progress=gr.Progress(track_tqdm=True)):
+    progress(0, desc="Starting collaborative report generation...")
+    yield "🧭 Starting collaborative report generation...\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"
+
+    # 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"
+
+    # 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"
+
+    # 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}"
 
-    yield final_text
 
 # Build Gradio Interface
 css = """
@@ -217,14 +231,18 @@ with gr.Blocks(css=css) as demo:
             value="academic"
         )
     fetch_btn = gr.Button("🌐 Fetch & Summarize Context", variant="secondary") # Rag Search
-    context_output = gr.Markdown(label="📚 Retrieved Context Summary") # Rag Search
+    context_output = gr.Markdown(label="📚 Retrieved Context Summary")
+    plan_output = gr.Markdown(label="📘 Research Outline")
+    write_output = gr.Markdown(label="📝 Draft Report")
+    fact_output = gr.Markdown(label="🔍 Fact-Checked Version")
+    edit_output = gr.Markdown(label="✅ Final Blog") # Rag Search
 
 
     run_button = gr.Button("🚀 Generate Blog", variant="primary")
     output = gr.Textbox(label="📰 Generated Blog Post", elem_id="output-box", lines=25, interactive=False, show_label=False)
 
     fetch_btn.click(fetch_context, inputs=[topic], outputs=[context_output]) # Rag Search
-    run_button.click(generate_blog, inputs=[topic, tone], outputs=[output])
+    run_button.click(generate_blog, inputs=[topic, tone], outputs=[plan_output, write_output, fact_output, edit_output])
 
 #Launch app
 if __name__=="__main__":