Update routes.py

routes.py

@@ -20,35 +20,39 @@ def initialize_bot(prompt_type: str):
     trainer = get_trainer()
     try:
         bot_id = trainer.initialize_bot_id()
-
+        
+        # Assign the prompt template based on the prompt type.
         if prompt_type == "university":
-             PromptTemplates.get_university_chatbot_prompt()
+            prompt_template = PromptTemplates.get_university_chatbot_prompt()
         elif prompt_type == "quiz_solving":
-             PromptTemplates.get_quiz_solving_prompt()
+            prompt_template = PromptTemplates.get_quiz_solving_prompt()
         elif prompt_type == "assignment_solving":
-             PromptTemplates.get_assignment_solving_prompt()
+            prompt_template = PromptTemplates.get_assignment_solving_prompt()
         elif prompt_type == "paper_solving":
-             PromptTemplates.get_paper_solving_prompt()
+            prompt_template = PromptTemplates.get_paper_solving_prompt()
         elif prompt_type == "quiz_creation":
-             PromptTemplates.get_quiz_creation_prompt()
+            prompt_template = PromptTemplates.get_quiz_creation_prompt()
         elif prompt_type == "assignment_creation":
-             PromptTemplates.get_assignment_creation_prompt()
+            prompt_template = PromptTemplates.get_assignment_creation_prompt()
         elif prompt_type == "paper_creation":
-             PromptTemplates.get_paper_creation_prompt()
+            prompt_template = PromptTemplates.get_paper_creation_prompt()
         elif prompt_type == "check_quiz":
-             PromptTemplates.get_check_quiz_prompt()
+            prompt_template = PromptTemplates.get_check_quiz_prompt()
         elif prompt_type == "check_assignment":
-             PromptTemplates.get_check_assignment_prompt()
+            prompt_template = PromptTemplates.get_check_assignment_prompt()
         elif prompt_type == "check_paper":
-             PromptTemplates.get_check_paper_prompt()
+            prompt_template = PromptTemplates.get_check_paper_prompt()
         else:
-             prompt_type=PromptTemplates.get_quiz_solving_prompt()
-            
-        trainer.set_custom_prompt_template(bot_id, prompt_type)
+            # Default prompt template if prompt_type is not recognized.
+            prompt_template = PromptTemplates.get_quiz_solving_prompt()
+        
+        # Pass the assigned prompt template to the bot.
+        trainer.set_custom_prompt_template(bot_id, prompt_template)
         return InitializeBotResponse(bot_id=bot_id)
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
+
 @router.post("/upload_document")
 async def upload_document(bot_id: str = Form(...), file: UploadFile = File(...)):
     """