Update routes.py

routes.py

@@ -10,14 +10,14 @@ from config import CUSTOM_PROMPT
 from prompt_templates import PromptTemplates
 
 router = APIRouter()
-
+trainer = get_trainer()
 @router.post("/initialize_bot", response_model=InitializeBotResponse)
 def initialize_bot(prompt_type: str):
     """
     Initializes a new bot with a custom prompt template.
     The custom_prompt parameter should be provided by the frontend (for example, via navigation).
     """
-    trainer = get_trainer()
+    
     try:
         bot_id = trainer.initialize_bot_id()
         
@@ -58,7 +58,6 @@ async def upload_document(bot_id: str = Form(...), file: UploadFile = File(...))
     """
     Saves the uploaded file temporarily and adds it to the bot's knowledge base.
     """
-    trainer = get_trainer()
     try:
         # Save the file to a temporary location
         with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(file.filename)[1]) as tmp:
@@ -80,7 +79,6 @@ def create_bot(bot_id: str):
     """
     Creates (builds) the bot (e.g., builds its index) for the given bot_id.
     """
-    trainer = get_trainer()
     try:
         trainer.create_bot(bot_id)
         return {"message": f"Bot {bot_id} created successfully."}
@@ -92,7 +90,6 @@ def new_chat(bot_id: str):
     """
     Creates a new chat session for the specified bot.
     """
-    trainer = get_trainer()
     try:
         chat_id = trainer.new_chat(bot_id)
         return NewChatResponse(chat_id=chat_id)
@@ -104,7 +101,6 @@ def send_query(query_request: QueryRequest):
     """
     Processes a query and returns the bot's response along with any web sources.
     """
-    trainer = get_trainer()
     try:
         response, web_sources = trainer.get_response(
             query_request.query, query_request.bot_id, query_request.chat_id
@@ -118,7 +114,6 @@ def list_chats(bot_id: str):
     """
     Returns a list of previous chat sessions for the specified bot.
     """
-    trainer = get_trainer()
     try:
         chats = trainer.list_chats(bot_id)
         return chats
@@ -131,7 +126,6 @@ def chat_history(chat_id: str, bot_id: str):
     Returns the chat history for a given chat session.
     ObjectId instances in the history are converted to strings.
     """
-    trainer = get_trainer()
     try:
         history = trainer.get_chat_by_id(chat_id=chat_id)
         return jsonable_encoder(history, custom_encoder={ObjectId: str})