Update app.py

app.py

@@ -1,14 +1,50 @@
-from fastapi import FastAPI, HTTPException
+from fastapi import FastAPI, HTTPException, Request, Response
 from pydantic import BaseModel
 from typing import List, Optional
 from llama_cpp import Llama
 from fastapi.responses import PlainTextResponse, JSONResponse
+from starlette.middleware.base import BaseHTTPMiddleware
 
+
+import logging
+import json
 import os
 import time
 import uuid
 
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger("api_logger")
+
+class LoggingMiddleware(BaseHTTPMiddleware):
+    async def dispatch(self, request: Request, call_next):
+        # Read request body (must be buffered manually)
+        body = await request.body()
+        logger.info(f"REQUEST: {request.method} {request.url}\nBody: {body.decode('utf-8')}")
+        
+        # Rebuild the request with body for downstream handlers
+        request = Request(request.scope, receive=lambda: {'type': 'http.request', 'body': body})
+
+        # Process the response
+        response = await call_next(request)
+        response_body = b""
+        async for chunk in response.body_iterator:
+            response_body += chunk
+
+        # Log response body and status code
+        logger.info(f"RESPONSE: Status {response.status_code}\nBody: {response_body.decode('utf-8')}")
+
+        # Rebuild response to preserve original functionality
+        return Response(
+            content=response_body,
+            status_code=response.status_code,
+            headers=dict(response.headers),
+            media_type=response.media_type
+        )
+
+# FastAPI app with middleware
 app = FastAPI()
+app.add_middleware(LoggingMiddleware)
 
 llm = None