Update main.py

main.py

@@ -25,7 +25,40 @@ from fastapi.middleware.cors import CORSMiddleware
 import re
 import os
 from langchain_core.prompts import ChatPromptTemplate
-import logging
+import logger
+
+import logger.config
+
+# Configure logger at application startup
+logger.config.dictConfig({
+    "version": 1,
+    "disable_existing_loggers": False,
+    "formatters": {
+        "default": {
+            "format": "%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+            "datefmt": "%Y-%m-%d %H:%M:%S",
+        }
+    },
+    "handlers": {
+        "console": {
+            "class": "logger.StreamHandler",
+            "stream": "ext://sys.stdout",
+            "formatter": "default",
+            "level": "DEBUG",
+        }
+    },
+    "root": {
+        "level": "DEBUG",
+        "handlers": ["console"]
+    },
+    "loggers": {
+        "uvicorn": {"handlers": ["console"], "level": "DEBUG"},
+        "fastapi": {"handlers": ["console"], "level": "DEBUG"}
+    }
+})
+
+# Create logger instance
+logger = logger.getLogger(__name__)
 
 app = FastAPI()
 app.include_router(document_rag_router) 
@@ -157,10 +190,10 @@ def get_collection_files(collection_id: str, user_id: str) -> str:
         if response.status_code == 200:
             return response.text
         else:
-            logging.error(f"Error from API: {response.text}")
+            logger.error(f"Error from API: {response.text}")
             return f"Error getting files: {response.text}"
     except Exception as e:
-        logging.error(f"Error getting collection files: {str(e)}")
+        logger.error(f"Error getting collection files: {str(e)}")
         return f"Error getting files: {str(e)}"
 
 def format_for_model(state):
@@ -172,7 +205,7 @@ def format_for_model(state):
     # Get files in the collection
 
     collection_files = get_current_files()#get_collection_files(collection_id, user_id) if collection_id and user_id else "No files available"
-    logging.info(f"fetching collection for userid{user_id} and collection_id {collection_id}|| Results {collection_files}")
+    logger.info(f"fetching collection for userid{user_id} and collection_id {collection_id}|| Results {collection_files}")
     return prompt.invoke({
         "collection_files": collection_files,
         "messages": state["messages"]
@@ -333,6 +366,11 @@ async def chat2(input_data: ChatInput):
 async def health_check():
     return {"status": "healthy"}
 
+
+
+
+
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=8000)