First version with working endpoint (/models)

src/model_factory.py → app/internal/model_factory.py

@@ -34,11 +34,7 @@ def discover_models():
     return models
 
 def get_model_instance(model_id):
-    global _models
-    if _models is None:
-        print("Indexing models")
-        _models = discover_models()
-    return _models.get(model_id)()
+    return get_vips_models().get(model_id)()
 
 def get_vips_model_packages():
     global _packages
@@ -46,9 +42,16 @@ def get_vips_model_packages():
         discover_models()
     return _packages
 
+def get_vips_models():
+    global _models
+    if _models is None:
+        #print("Indexing models")
+        _models = discover_models()
+    return _models
+
 # USAGE EXAMPLE. REMOVE when U start using the module from elsewhere
-model = get_model_instance("REFERENCEM")
-print(model.get_model_name())
-print(get_vips_model_packages())
+#model = get_model_instance("REFERENCEM")
+#print(model.get_model_name())
+#print(get_vips_model_packages())
 
 

app/main.py

+from fastapi import FastAPI
+from fastapi.responses import PlainTextResponse
+from app.internal.model_factory import *
+
+app = FastAPI()
+
+
+@app.get("/")
+async def root():
+    return {"message": "Hello World"}
+
+@app.get("/models", response_class=PlainTextResponse)
+async def print_model_list() -> str:
+    model_list = ""
+    vips_models = get_vips_models()
+    for model_id in vips_models:
+        model_list = "%s%s %s\n" %(model_list, model_id, get_model_instance(model_id).get_model_name())
+    return model_list

safe_pip_freeze.py

+import sys
+import subprocess
+import src.model_factory
+
+# Intention: Make sure that pip freeze doesn't include VIPS models
+# Only necessary if we decide to stick with using pip freeze
\ No newline at end of file

src/main.py

-from fastapi import FastAPI
-
-app = FastAPI()
-
-
-@app.get("/")
-async def root():
-    return {"message": "Hello World"}