Create app.py

app.py

@@ -0,0 +1,42 @@
+import nh3
+from fastapi import FastAPI, Request
+from doc_searcher import DocSearcher
+from suggestion_searcher import SuggestionSearcher
+from huggingface_hub import login
+from config import HUGGING_FACE_API_KEY, COLLECTION_NAME, API_KEY, COLLECTION_NAME_SUGGESTION
+from fastapi.responses import JSONResponse
+
+login(HUGGING_FACE_API_KEY)
+
+app = FastAPI()
+
+doc_searcher = DocSearcher(collection_name=COLLECTION_NAME)
+suggestion_searcher = SuggestionSearcher(collection_name=COLLECTION_NAME_SUGGESTION)
+
+ALLOWED_API_KEY = str(API_KEY)
+
+@app.get("/api/search")
+async def search(q: str):
+    query = q.lower()
+    xss = nh3.clean(query)
+    data = await doc_searcher.search(text=xss)
+    return data
+
+@app.get("/api/suggestions")
+async def get_suggestions(q: str, type: int):
+    query = q.lower()
+    xss = nh3.clean(query)
+    data = await suggestion_searcher.search(text=xss,type=type)
+    return data
+
+
+# @app.middleware("http")
+# async def api_key_authentication(request: Request, call_next):
+#     api_key = request.headers.get("X-API-KEY")
+#     if api_key != ALLOWED_API_KEY:
+#         return JSONResponse(
+#             status_code=403,
+#             content={"message": "Forbidden."}
+#         )
+
+#     return await call_next(request)