Create main.py

main.py

@@ -0,0 +1,36 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from planner import llama_reason
+from retrieval import retrieve_context
+from fastapi.middleware.cors import CORSMiddleware
+
+app = FastAPI(title="Aegis Backend API")
+
+# ✅ Allow frontend to talk to backend (and allow preflight OPTIONS)
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # or ["http://localhost:8081"]
+    allow_credentials=True,
+    allow_methods=["*"],  # this allows OPTIONS too
+    allow_headers=["*"],
+)
+
+class ReasonRequest(BaseModel):
+    query: str
+
+class RetrieveRequest(BaseModel):
+    query: str
+
+@app.get("/")
+def home():
+    return {"status": "Aegis backend running"}
+
+@app.post("/reason")
+def reason(request: ReasonRequest):
+    response = llama_reason(request.query)
+    return {"response": response}
+
+@app.post("/retrieve")
+def retrieve(request: RetrieveRequest):
+    context = retrieve_context(request.query)
+    return {"context": context}