feat: handle custom urls with env variables: CUSTOM_INDEX_URL, CUSTOM_SAMGIS_URL, CUSTOM_LISA_URL, CUSTOM_GRADIO_URL

app.py

@@ -22,8 +22,10 @@ from samgis_lisa_on_zero.utilities.type_hints import ApiRequestBody, StringPromp
 loglevel = os.getenv('LOGLEVEL', 'INFO').upper()
 app_logger = setup_logging(debug=loglevel)
 
-CUSTOM_GRADIO_PATH = "/"
-CUSTOM_STATIC_PATH = "/static"
+CUSTOM_INDEX_URL = os.getenv("CUSTOM_INDEX_URL", "/")
+CUSTOM_SAMGIS_URL = os.getenv("CUSTOM_SAMGIS_URL", "/samgis")
+CUSTOM_LISA_URL = os.getenv("CUSTOM_LISA_URL", "/lisa")
+CUSTOM_GRADIO_URL = os.getenv("CUSTOM_GRADIO_URL", "/gradio")
 FASTAPI_TITLE = "samgis-lisa-on-zero"
 app = FastAPI(title=FASTAPI_TITLE, version="1.0")
 
@@ -65,16 +67,16 @@ async def health() -> JSONResponse:
     import importlib.metadata
     from importlib.metadata import PackageNotFoundError
 
-    core_version = lisa_on_cuda_version = samgis_lisa_on_zero_version = ""
+    core_version = lisa_on_cuda_version = samgis_lisa_on_cuda_version = ""
     try:
         core_version = importlib.metadata.version('samgis_core')
         lisa_on_cuda_version = importlib.metadata.version('lisa-on-cuda')
-        samgis_lisa_on_zero_version = importlib.metadata.version('samgis-lisa-on-zero')
+        samgis_lisa_on_cuda_version = importlib.metadata.version('samgis-lisa-on-zero')
     except PackageNotFoundError as pe:
         app_logger.error(f"pe:{pe}.")
 
     msg = "still alive, "
-    msg += f"""version:{samgis_lisa_on_zero_version}, core version:{core_version},"""
+    msg += f"""version:{samgis_lisa_on_cuda_version}, core version:{core_version},"""
     msg += f"""lisa-on-cuda version:{lisa_on_cuda_version},"""
 
     app_logger.info(msg)
@@ -261,28 +263,28 @@ templates = Jinja2Templates(directory="templates")
 
 # important: the index() function and the app.mount MUST be at the end
 # samgis.html
-app.mount("/samgis", StaticFiles(directory=static_dist_folder, html=True), name="samgis")
+app.mount(CUSTOM_SAMGIS_URL, StaticFiles(directory=static_dist_folder, html=True), name="samgis")
 
 
-@app.get("/samgis")
+@app.get(CUSTOM_SAMGIS_URL)
 async def samgis() -> FileResponse:
     return FileResponse(path=static_dist_folder / "samgis.html", media_type="text/html")
 
 
 # lisa.html
-app.mount("/lisa", StaticFiles(directory=static_dist_folder, html=True), name="lisa")
+app.mount(CUSTOM_LISA_URL, StaticFiles(directory=static_dist_folder, html=True), name="lisa")
 
 
-@app.get("/lisa")
+@app.get(CUSTOM_LISA_URL)
 async def lisa() -> FileResponse:
     return FileResponse(path=static_dist_folder / "lisa.html", media_type="text/html")
 
 
 # # index.html (lisa.html copy)
-app.mount("/", StaticFiles(directory=static_dist_folder, html=True), name="index")
+app.mount(CUSTOM_INDEX_URL, StaticFiles(directory=static_dist_folder, html=True), name="index")
 
 
-@app.get("/")
+@app.get(CUSTOM_INDEX_URL)
 async def index() -> FileResponse:
     return FileResponse(path=static_dist_folder / "index.html", media_type="text/html")
 
@@ -292,8 +294,8 @@ inference_fn = app_helpers.get_inference_model_by_args(args, inference_decorator
 
 app_helpers.app_logger.info(f"prepared inference_fn function:{inference_fn.__name__}, creating gradio interface...")
 io = app_helpers.get_gradio_interface(inference_fn)
-app_helpers.app_logger.info("created gradio interface, mounting gradio app within FastAPI...")
-app = gr.mount_gradio_app(app, io, path=CUSTOM_GRADIO_PATH)
+app_helpers.app_logger.info(f"created gradio interface, mounting gradio app on url {CUSTOM_GRADIO_URL} within FastAPI...")
+app = gr.mount_gradio_app(app, io, path=CUSTOM_GRADIO_URL)
 app_helpers.app_logger.info("mounted gradio app within fastapi")