replace Flask Blueprint with FastAPI router

Author: kkaris

src/semra/web/autocomplete/autocomplete_blueprint.py

@@ -1,20 +1,16 @@
-from flask import Blueprint, request, jsonify
+import fastapi
 
-from semra.web.autocomplete import ConceptsTrie
+from semra.web.autocomplete import ConceptsTrie, Entry
 
-auto_blueprint = Blueprint("autocomplete", __name__, url_prefix="/autocomplete")
+auto_router = fastapi.APIRouter(prefix="/autocomplete")
 
 
 # Initialize the autocomplete trie
 trie = ConceptsTrie.from_graph_db()
 
 
-@auto_blueprint.route("/search", methods=["GET"])
-def autocomplete_search():
+@auto_router.get("/search", response_model=list[Entry])
+def autocomplete_search(prefix: str, top_n: int = 100):
     """Get the autocomplete suggestions for a given prefix."""
-    prefix = request.args.get("prefix")
-    top_n = min(int(request.args.get("top_n", 100)), 100)
-
-    return jsonify(
-        trie.case_insensitive_search(prefix, top_n=top_n)
-    )
+    top_n = min(top_n, 100)
+    return trie.case_insensitive_search(prefix, top_n=top_n)

src/semra/wsgi.py

@@ -79,17 +79,16 @@ def get_app(
     Bootstrap5(flask_app)
 
     flask_app.register_blueprint(flask_blueprint)
-    if add_autocomplete:
-        from semra.web.autocomplete.autocomplete_blueprint import auto_blueprint
-
-        flask_app.register_blueprint(auto_blueprint)
 
     fastapi_app = fastapi.FastAPI(
         title="Semantic Reasoning Assembler",
         description="A web app to access a SeMRA Neo4j database",
     )
     fastapi_app.state = state  # type:ignore
     fastapi_app.include_router(api_router)
+    if add_autocomplete:
+        from semra.web.autocomplete.autocomplete_blueprint import auto_router
+        fastapi_app.include_router(auto_router)
     fastapi_app.mount("/", WSGIMiddleware(flask_app))
 
     if return_flask: