Feat/14 instance of planprat

backend/src/main.py

@@ -1,15 +1,34 @@
 import logging
 from fastapi import FastAPI, HTTPException, Query, UploadFile
 from fastapi import status
-from pydantic import BaseModel
+from fastapi.middleware.cors import CORSMiddleware
 
 
 from src.types import SummaryResponse, PlanPratRequest, PlanPratResponse
 from src.services.reader import Reader
 from src.services.readers.factory import create_reader
 from src.services.agent import invoke_agent, invoke_plan_agent
 
-app = FastAPI()
+
+app = FastAPI(
+    title="KPRO API AI system",
+    description="retrives tekst from user and returns an answer based on building regulations.",
+    version="1.0.0",
+)
+
+ORIGINS = [
+    "http://localhost:3000",
+    "http://localhost:80",
+    "http://localhost",
+]
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=ORIGINS,
+    allow_credentials=True,
+    allow_methods=["POST", "GET", "OPTIONS"],
+    allow_headers=["*"],
+)
 
 
 logging.basicConfig(filename="summary-assistant.log", level=logging.INFO)

backend/src/services/agent.py

@@ -53,10 +53,10 @@ def invoke_plan_agent(query: str) -> str:
 
         """
     )
-    print(f"context: {_retrieve_law_context()}", flush=True)
+    logger.info(f"context: {_retrieve_law_context()}")
     generate = prompt | llm
     response = generate.invoke({"query": query, "context": _retrieve_law_context()})
-    print(f"Response: {response}", flush=True)
+    logger.info(f"Response: {response}")
     return response.content
 
 

docker-compose.yml

@@ -19,6 +19,6 @@ services:
       context: ./backend
       dockerfile: Dockerfile
     ports:
-      - "8000:8000"
+      - "000:8000"
     volumes:
       - ./backend:/code