Update main.py

main.py

@@ -7,7 +7,6 @@ from src.mapper.e5map import E5Mapper
 from src.mapper.scimap import scimap
 from src.mapper.parser import MapperParser
 from src.datatonic.dataloader import DataLoader
-from src.teams.agentteam import codingteam, covid19team, financeteam, debateteam, homeworkteam, consultingteam
 from src.agentics.agents import AgentsFactory
 
 title = """# Welcome to 👩🏻‍🔬🧪SciTonic
@@ -101,6 +100,81 @@ def process_query(oai_key, query, max_auto_reply):
     # Retrieve the Boss Assistant agent
     boss_aid = agents_factory.scitonic()
 
+    def _reset_agents():
+        boss_aid.reset()
+
+    # Define functions for each team
+    def codingteam():
+        _reset_agents()
+        team = autogen.GroupChat(
+            agents=[scitonic, coder, pm, reviewer],
+            messages=[],
+            max_round=12,
+            speaker_selection_method="round_robin"
+        )
+
+        manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
+        boss_aid.initiate_chat(manager, problem=PROBLEM, n_results=3)
+
+    def covid19team():
+        _reset_agents()
+        team = autogen.GroupChat(
+            agents=[scitonic, covid19_scientist, healthcare_expert, finance_analyst],
+            messages=[],
+            max_round=12
+        )
+
+        manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
+        boss_aid.initiate_chat(manager, covid19_problem=COVID19_PROBLEM, n_results=3)
+
+    def financeteam():
+        _reset_agents()
+        team = autogen.GroupChat(
+            agents=[scitonic, finance_analyst, pm, reviewer, finance_expert],
+            messages=[],
+            max_round=12,
+            speaker_selection_method="round_robin"
+        )
+
+        manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
+        boss_aid.initiate_chat(manager, finance_problem=FINANCE_PROBLEM, n_results=3)
+
+    def debateteam():
+        _reset_agents()
+        team = autogen.GroupChat(
+            agents=[scitonic, debate_expert, pm, reviewer, debate_champion],
+            messages=[],
+            max_round=12,
+            speaker_selection_method="round_robin"
+        )
+
+        manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
+        boss_aid.initiate_chat(manager, debate_problem=DEBATE_PROBLEM, n_results=3)
+
+    def homeworkteam():
+        _reset_agents()
+        team = autogen.GroupChat(
+            agents=[scitonic, academic_expert, pm, reviewer, academic_whiz],
+            messages=[],
+            max_round=12,
+            speaker_selection_method="round_robin"
+        )
+
+        manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
+        boss_aid.initiate_chat(manager, homework_problem=HOMEWORK_PROBLEM, n_results=3)
+
+    def consultingteam():
+        _reset_agents()
+        team = autogen.GroupChat(
+            agents=[scitonic, consultant, pm, reviewer, consulting_pro],
+            messages=[],
+            max_round=12,
+            speaker_selection_method="round_robin"
+        )
+
+        manager = autogen.GroupChatManager(groupchat=team, llm_config=llm_config)
+        boss_aid.initiate_chat(manager, consulting_problem=CONSULTING_PROBLEM, n_results=3)
+
     # Select and initiate team based on team mapping
     team_function = {
         "CodingTeam": codingteam,
@@ -113,7 +187,7 @@ def process_query(oai_key, query, max_auto_reply):
 
     team_action = team_function.get(team, lambda: "No appropriate team found for the given input.")
     return team_action()
-    return boss_aid
+
 
 
 def main():