Update app.py

app.py

@@ -1,9 +1,64 @@
-from crewai import Agent, Task, Crew, Process
-import gradio as gr
 import os
+import gradio as gr
+
+from crewai import Agent, Task, Crew, Process
 
 os.environ["OPENAI_API_KEY"] = "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"
 
+from crewai import Agent, Task, Crew, Process
+
+def run_therapy_session(group_size, topic):
+    participant_names = ['Alice', 'Bob', 'Charlie', 'Diana', 'Ethan', 'Fiona', 'George', 'Hannah', 'Ivan']
+    
+    if group_size > len(participant_names) + 1:  # +1 for the therapist
+        return "Group size exceeds the number of available participant names."
+
+    # Create the therapist agent
+    dr_smith = Agent(role='Therapist', goal='Facilitate a supportive group discussion', llm='mixtral', verbose=True)
+
+    # Create participant agents
+    participants = [Agent(role='Group Therapy Participant', goal='Participate in group therapy', llm='mixtral', verbose=True)
+                    for _ in range(group_size - 1)]
+    participants.append(dr_smith)  # Add the therapist
+
+    # Define tasks for each participant
+    tasks = [Task(description=f'{name} discusses the topic: {topic}', agent=participant)
+             for name, participant in zip(participant_names, participants)]
+
+    # Instantiate the crew with a sequential process
+    therapy_crew = Crew(
+        agents=participants,
+        tasks=tasks,
+        process=Process.sequential,
+        verbose=True
+    )
+
+    # Start the group therapy session
+    result = therapy_crew.kickoff()
+
+    # Simulating a conversation (placeholder, adjust based on CrewAI capabilities)
+    conversation = "\n".join([f"{name}: [Participant's thoughts on '{topic}']" for name in participant_names[:group_size-1] + ["Dr. Smith"]])
+
+    return conversation
+
+# Gradio interface
+iface = gr.Interface(
+    fn=run_therapy_session,
+    inputs=[
+        gr.Slider(minimum=2, maximum=10, default=5, label="Group Size"),
+        gr.Textbox(lines=2, placeholder="Enter a topic or question for discussion", label="Discussion Topic")
+    ],
+    outputs="text"
+)
+
+# Launch the interface
+iface.launch()
+
+
+
+# Choosing topics
+
+'''
 def run_crew(topic):
     # Define your agents
     researcher = Agent(
@@ -64,3 +119,5 @@ iface = gr.Interface(
 )
 
 iface.launch()
+
+'''