Added further classification for question and included returning responses

main.py

@@ -43,7 +43,21 @@ def hello():
 
 @app.post("/ask")
 def ask(question: str):
-    question_category = eval(master_agent.answer_question(master, question))
+    category = eval(master_agent.answer_question(master, question))
 
+    if category['category_number'] == 1:
+        response = eval(plant_agent.answer_question(plant, question))
 
-    return {"category": question_category}
+        category.update(response)
+
+    elif category['category_number'] == 2:
+        response = eda_agent.answer_question(eda, question)
+
+        category['response'] = response
+
+    elif category['category_number'] == 3:
+        response = rag_agent.answer_question(rag, question)
+
+        category['response'] = response
+
+    return category