Update app.py

app.py

@@ -1,11 +1,11 @@
 from crewai import Agent, Task, Crew, Process
 import gradio as gr
 
-def run_crew():
-    # Define your agents with roles and goals
+def run_crew(topic):
+    # Define your agents
     researcher = Agent(
         role='Senior Research Analyst',
-        goal='Uncover cutting-edge developments in AI and data science',
+        goal='Uncover cutting-edge developments',
         backstory="""You are a Senior Research Analyst at a leading tech think tank...""",
         verbose=True,
         allow_delegation=False
@@ -13,20 +13,29 @@ def run_crew():
 
     writer = Agent(
         role='Tech Content Strategist',
-        goal='Craft compelling content on tech advancements',
+        goal='Craft compelling content',
         backstory="""You are a renowned Tech Content Strategist...""",
         verbose=True,
         allow_delegation=False
     )
 
-    # Create tasks for your agents
+    # Assign tasks based on the selected topic
+    if topic == "write short story":
+        task_description = "Write a captivating short story about a journey through a futuristic city."
+    elif topic == "write an article":
+        task_description = "Compose an insightful article on the latest trends in technology."
+    elif topic == "analyze stock":
+        task_description = "Perform a detailed analysis of recent trends in the stock market."
+    elif topic == "create a vacation":
+        task_description = "Plan a perfect vacation itinerary for a family trip to Europe."
+
     task1 = Task(
-        description="""Conduct a comprehensive analysis of the latest advancements in AI in 2024...""",
+        description=task_description,
         agent=researcher
     )
 
     task2 = Task(
-        description="""Using the insights from the researcher's report, develop an engaging blog post...""",
+        description=f"Use the findings from the researcher's task to develop a comprehensive report on '{topic}'.",
         agent=writer
     )
 
@@ -42,16 +51,13 @@ def run_crew():
     result = crew.kickoff()
     return result
 
-def wrapper():
-    """ Wrapper function for the button click. """
-    return run_crew()
-
+# Gradio Interface with Dropdown for Topic Selection
 iface = gr.Interface(
-    fn=wrapper,
-    inputs=gr.inputs.Button(label="Run Crew"),  # Button input
+    fn=run_crew,
+    inputs=gr.inputs.Dropdown(choices=["write short story", "write an article", "analyze stock", "create a vacation"], label="Select Topic"),
     outputs="text",
     title="AI Research and Writing Crew",
-    description="Click the button to run the crew of AI agents."
+    description="Select a topic and click the button to run the crew of AI agents."
 )
 
 iface.launch()