Create main.py

main.py

@@ -0,0 +1,18 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from services.llm import generate_answer
+from services.retriever import Retriever
+
+app = FastAPI(title="AI Voice Assistant API")
+
+docs = open("documents/ohamlab/general.md").read().split("\n\n")
+retriever = Retriever(docs)
+
+class Query(BaseModel):
+    question: str
+
+@app.post("/api/ask")
+def ask(q: Query):
+    context = retriever.search(q.question)
+    answer = generate_answer(context, q.question, 128)
+    return {"answer": answer}