Update routes.py

routes.py

@@ -1,4 +1,3 @@
-# routes.py
 import os
 import shutil
 import tempfile
@@ -10,43 +9,16 @@ from prompt_templates import PromptTemplates
 
 router = APIRouter()
 
-def select_prompt(prompt_type: str):
-    """Selects a prompt template based on the provided prompt_type."""
-    if prompt_type == "university":
-        return PromptTemplates.get_university_chatbot_prompt()
-    elif prompt_type == "quiz_solving":
-        return PromptTemplates.get_quiz_solving_prompt()
-    elif prompt_type == "assignment_solving":
-        return PromptTemplates.get_assignment_solving_prompt()
-    elif prompt_type == "paper_solving":
-        return PromptTemplates.get_paper_solving_prompt()
-    elif prompt_type == "quiz_creation":
-        return PromptTemplates.get_quiz_creation_prompt()
-    elif prompt_type == "assignment_creation":
-        return PromptTemplates.get_assignment_creation_prompt()
-    elif prompt_type == "paper_creation":
-        return PromptTemplates.get_paper_creation_prompt()
-    elif prompt_type == "check_quiz":
-        return PromptTemplates.get_check_quiz_prompt()
-    elif prompt_type == "check_assignment":
-        return PromptTemplates.get_check_assignment_prompt()
-    elif prompt_type == "check_paper":
-        return PromptTemplates.get_check_paper_prompt()
-    else:
-        # Fallback to a default prompt if no matching type is found
-        return PromptTemplates.get_quiz_solving_prompt()
-
 @router.post("/initialize_bot", response_model=InitializeBotResponse)
-def initialize_bot(prompt_type: str = "quiz_solving"):
+def initialize_bot(custom_prompt: str):
     """
     Initializes a new bot with a custom prompt template.
-    The prompt_type query parameter determines which prompt to use.
+    The custom_prompt parameter should be provided (e.g., via navigation) and is used as the bot's prompt template.
     """
     trainer = get_trainer()
     try:
         bot_id = trainer.initialize_bot_id()
-        selected_prompt = select_prompt(prompt_type)
-        trainer.set_custom_prompt_template(bot_id, selected_prompt)
+        trainer.set_custom_prompt_template(bot_id, custom_prompt)
         return InitializeBotResponse(bot_id=bot_id)
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
@@ -120,4 +92,3 @@ def chat_history(chat_id: str, bot_id: str):
         return jsonable_encoder(history, custom_encoder={ObjectId: str})
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
-