Update routes.py

routes.py

@@ -101,7 +101,33 @@ def send_query(query_request: QueryRequest):
     """
     Processes a query and returns the bot's response along with any web sources.
     """
+    prompt_type = query_request.prompt_type
     try:
+        # Assign the prompt template based on the prompt type.
+        if prompt_type == "university":
+            prompt_template = PromptTemplates.get_university_chatbot_prompt()
+        elif prompt_type == "quiz_solving":
+            prompt_template = PromptTemplates.get_quiz_solving_prompt()
+        elif prompt_type == "assignment_solving":
+            prompt_template = PromptTemplates.get_assignment_solving_prompt()
+        elif prompt_type == "paper_solving":
+            prompt_template = PromptTemplates.get_paper_solving_prompt()
+        elif prompt_type == "quiz_creation":
+            prompt_template = PromptTemplates.get_quiz_creation_prompt()
+        elif prompt_type == "assignment_creation":
+            prompt_template = PromptTemplates.get_assignment_creation_prompt()
+        elif prompt_type == "paper_creation":
+            prompt_template = PromptTemplates.get_paper_creation_prompt()
+        elif prompt_type == "check_quiz":
+            prompt_template = PromptTemplates.get_check_quiz_prompt()
+        elif prompt_type == "check_assignment":
+            prompt_template = PromptTemplates.get_check_assignment_prompt()
+        elif prompt_type == "check_paper":
+            prompt_template = PromptTemplates.get_check_paper_prompt()
+        else:
+            # Default prompt template if prompt_type is not recognized.
+            prompt_template = PromptTemplates.get_quiz_solving_prompt()
+        trainer.set_custom_prompt_template(bot_id, prompt_template)
         response, web_sources = trainer.get_response(
             query_request.query, query_request.bot_id, query_request.chat_id
         )