Update app.py

app.py

@@ -1,30 +1,77 @@
 import gradio as gr
+import time
 
-def planner(query):
-    return f"Plan for: {query}"
+class PlannerAgent:
+    def run(self, query):
+        return {
+            "goal": query,
+            "tasks": [
+                "Understand the topic",
+                "Collect important information",
+                "Generate final report"
+            ]
+        }
 
-def researcher(query):
-    return f"Research collected for: {query}"
+class ResearchAgent:
+    def run(self, plan):
+        topic = plan["goal"]
 
-def writer(research):
-    return f"Final Report:\n\n{research}"
+        research = f"""
+Topic: {topic}
 
-def multi_agent(query):
-    plan = planner(query)
-    research = researcher(plan)
-    result = writer(research)
-    return result
+Key Points:
+• Definition and overview
+• Major applications
+• Advantages
+• Challenges
+• Future scope
+"""
+        return research
+
+class WriterAgent:
+    def run(self, research):
+        report = f"""
+# AI Generated Report
+
+{research}
+
+Conclusion:
+This report summarizes the important aspects of the topic and provides a concise overview.
+"""
+        return report
+
+planner = PlannerAgent()
+researcher = ResearchAgent()
+writer = WriterAgent()
+
+def multi_agent_workflow(query):
+
+    if not query.strip():
+        return "Please enter a topic."
+
+    plan = planner.run(query)
+    time.sleep(1)
+
+    research = researcher.run(plan)
+    time.sleep(1)
+
+    final_report = writer.run(research)
+
+    return final_report
 
 demo = gr.Interface(
-    fn=multi_agent,
+    fn=multi_agent_workflow,
     inputs=gr.Textbox(
-        label="Enter your topic"
-    ),
-    outputs=gr.Textbox(
-        label="Generated Report"
+        lines=2,
+        placeholder="Enter any topic..."
     ),
-    title="Multi-Agent AI System",
-    description="Planner → Researcher → Writer"
+    outputs=gr.Markdown(),
+    title="Multi-Agent AI Research Assistant",
+    description="""
+Planner Agent → Research Agent → Writer Agent
+
+Enter a topic and the agents will collaborate to generate a report.
+"""
 )
 
 demo.launch()