Update routes.py

routes.py

@@ -1,7 +1,7 @@
 import os
 import shutil
 import tempfile
-from fastapi import APIRouter, HTTPException, UploadFile, File, Form
+from fastapi import APIRouter, HTTPException, UploadFile, File, Form, Query
 from fastapi.encoders import jsonable_encoder
 from bson import ObjectId
 from models import InitializeBotResponse, NewChatResponse, QueryRequest, QueryResponse
@@ -13,12 +13,14 @@ router = APIRouter()
 trainer = get_trainer()
 
 @router.post("/initialize_bot", response_model=InitializeBotResponse)
-def initialize_bot():
+def initialize_bot(prompt_type: str = Query(None)):
     """
-    Initializes a bot and returns its bot_id.
+    Initializes a new bot and returns its bot_id.
+    Accepts an optional 'prompt_type' query parameter (provided by the frontend).
     """
     try:
         bot_id = trainer.initialize_bot_id()
+        # Optionally, you might want to store the prompt_type with the bot record.
         return InitializeBotResponse(bot_id=bot_id)
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
@@ -46,14 +48,18 @@ async def upload_document(bot_id: str = Form(...), file: UploadFile = File(...))
         raise HTTPException(status_code=500, detail=str(e))
 
 
-@router.post("/create_bot")
-def create_bot(bot_id: str = Form(...), prompt_type: str = Form(...)):
+@router.post("/create_bot/{bot_id}")
+def create_bot(bot_id: str, prompt_type: str = Query(None)):
     """
-    Finalizes the creation (build) of the bot using the specified prompt type.
-    This endpoint assigns the appropriate custom prompt template before finalizing bot creation.
+    Finalizes the creation (build) of the bot identified by bot_id.
+    Uses the provided (or default) prompt_type to determine the custom prompt template.
+    If no prompt_type is provided, it defaults to "quiz_solving".
     """
     try:
-        # Determine the appropriate prompt template based on the provided prompt_type.
+        if prompt_type is None:
+            prompt_type = "quiz_solving"
+
+        # Determine the appropriate prompt template.
         if prompt_type == "university":
             prompt_template = PromptTemplates.get_university_chatbot_prompt()
         elif prompt_type == "quiz_solving":
@@ -75,10 +81,9 @@ def create_bot(bot_id: str = Form(...), prompt_type: str = Form(...)):
         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()
-        
-        # Create the bot using the specified bot_id and prompt template.
+
+        # Create (build) the bot using the specified bot_id and prompt template.
         trainer.create_bot(bot_id, prompt_template)
         return {"message": f"Bot {bot_id} created successfully."}
     except Exception as e:
@@ -101,7 +106,7 @@ def new_chat(bot_id: str):
 def send_query(query_request: QueryRequest):
     """
     Processes a query and returns the bot's response along with any web sources.
-    The QueryRequest model must include the bot_id, chat_id, and query.
+    The request must include bot_id, chat_id, and the query text.
     """
     try:
         response, web_sources = trainer.get_response(
@@ -125,9 +130,10 @@ def list_chats(bot_id: str):
 
 
 @router.get("/chat_history/{chat_id}")
-def chat_history(chat_id: str):
+def chat_history(chat_id: str, bot_id: str = Query(None)):
     """
     Returns the chat history for a given chat session.
+    The bot_id can be provided as a query parameter (if needed).
     ObjectId instances in the history are converted to strings.
     """
     try: