update gradio UI

app.py

@@ -155,29 +155,32 @@ import time
 import gradio as gr
 
 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 generation...\n"
+    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}"
-
+    yield f"### 🧾 Editor-in-Chief (Final Blog)\n{edit_result.output_text}\n\n✅ **Report Complete!**"
 
 # Build Gradio Interface
 css = """
@@ -231,18 +234,13 @@ 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")
-    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
-
+    context_output = gr.Markdown(label="📚 Retrieved Context Summary") # 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)
+    final_output = gr.Textbox(label="📰 Generated Blog Post", elem_id="output-box")
 
     fetch_btn.click(fetch_context, inputs=[topic], outputs=[context_output]) # Rag Search
-    run_button.click(generate_blog, inputs=[topic, tone], outputs=[plan_output, write_output, fact_output, edit_output])
+    run_button.click(generate_blog, inputs=[topic, tone], outputs=[final_output])
 
 #Launch app
 if __name__=="__main__":