implement logging

main.py

@@ -11,6 +11,10 @@ from pydantic import BaseModel
 from slowapi import Limiter, _rate_limit_exceeded_handler
 from slowapi.errors import RateLimitExceeded
 from slowapi.util import get_remote_address
+import logging
+
+# Configure logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
 
 # Initialize the limiter
 limiter = Limiter(key_func=get_remote_address)
@@ -75,6 +79,7 @@ async def stream(request: Request, item: Validation):
     Returns:
         dict: A dictionary containing the generated story content and the timestamp of generation.
     """
+    logging.info('Generating story')
     prompt = "<|user|>\nA bedtime tiny story for the children<\/s>\n<|assistant|>\n"
     story_content = llm(prompt)
     story_dict = {
@@ -82,6 +87,7 @@ async def stream(request: Request, item: Validation):
         "generated_at": datetime.datetime.utcnow()
     }
     result = await db.stories.insert_one(story_dict)
+    logging.info(f'Story generated with id {result.inserted_id}')
     return {"_id": str(result.inserted_id), "story_content": story_content}
 
 
@@ -113,8 +119,10 @@ async def create_feedback(feedback: Feedback):
     Returns:
     - dict: A dictionary containing the ID of the inserted feedback.
     """
+    logging.info('Receiving feedback')
     feedback_dict = feedback.dict()
     feedback_dict["feedback_at"] = datetime.datetime.utcnow()
     feedback_dict["story_id"] = ObjectId(feedback.story_id)
     result = await db.feedback.insert_one(feedback_dict)
+    logging.info(f'Feedback received with id {result.inserted_id}')
     return {"_id": str(result.inserted_id)}