build: add run all shell script with mongodb

app/api/management_api.py

@@ -1,14 +1,21 @@
 from fastapi import APIRouter
+from fastapi.responses import RedirectResponse, JSONResponse
 from pathlib import Path
 from pydantic import BaseModel
 import toml
 from loguru import logger
+from environs import Env
+import json
 
 
-router = APIRouter(prefix="/v1", tags=["management"])
+env = Env()
+env.read_env()
+REDIRECT_TO_GRADIO_UI = env.bool("REDIRECT_TO_GRADIO_UI", True)
 
+router = APIRouter(tags=["management"])
 
-#### Health Check ##############
+
+#### Health Check #################################################
 class HealthResponse(BaseModel):
     status: str = "ok"
 
@@ -21,7 +28,7 @@ async def health_check():
     return HealthResponse()
 
 
-#### Version ####################
+#### Version #######################################################
 __version__ = None
 
 
@@ -52,3 +59,30 @@ async def version_api():
     Version endpoint, returns the version of the system
     """
     return {"version": _load_version()}
+
+
+def _get_manifest() -> JSONResponse:
+    print("get manifest")
+    current_dir = Path(__file__).resolve().parent.parent.parent
+    manifest_path = current_dir / "static" / "manifest.json"
+    with open(manifest_path, "r") as f:
+        return JSONResponse(json.load(f))
+
+
+### root ##########################################################
+@router.get("/")
+async def root():
+    """
+    Redirect root to Gradio UI
+    """
+    if REDIRECT_TO_GRADIO_UI:
+        return RedirectResponse(url="/ui")
+    else:
+        return _get_manifest()
+
+
+### manifest #######################################################
+@router.get("/manifest.json")
+async def get_manifest():
+    """Return the web app manifest"""
+    return _get_manifest()

main.py

@@ -1,30 +1,14 @@
-import json
 from fastapi import FastAPI
-from fastapi.responses import RedirectResponse, JSONResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.middleware.cors import CORSMiddleware
 from app.api import chat_api, conversation_api, management_api
-from app.config.log import log_config
 from loguru import logger
-from environs import Env
 from contextlib import asynccontextmanager
 from app.db.factory import db_client
 from gradio_chatbot import build_gradio_app, app_auth
 import gradio as gr
-import os
 from app.core.initial_setup.setup import InitialSetup
 
-print(log_config.get_log_level())
-
-env = Env()
-env.read_env()
-
-DB_DATABASE_TYPE = env.str("DB_DATABASE_TYPE", "mongodb")
-
-# special handling for Hugging Face Space
-IS_HF_SPACE = os.environ.get("SPACE_ID") is not None
-SPACE_URL = "https://lokumai-openai-openapi-template.hf.space" if IS_HF_SPACE else "http://localhost:7860"
-
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
@@ -115,6 +99,7 @@ app.add_middleware(
 
 app.mount("/static", StaticFiles(directory="static"), name="static")
 app.mount("/.well-known", StaticFiles(directory=".well-known"), name="well-known")
+
 app.include_router(chat_api.router)
 app.include_router(management_api.router)
 app.include_router(conversation_api.router)
@@ -124,18 +109,4 @@ demo = build_gradio_app()
 app = gr.mount_gradio_app(app, demo, path="/ui", auth=app_auth)
 
 
-@app.get("/")
-async def root():
-    """Redirect root to Gradio UI"""
-    return RedirectResponse(url="/ui")
-
-
-@app.get("/manifest.json")
-async def get_manifest():
-    """Return the web app manifest"""
-    manifest_path = os.path.join(os.path.dirname(__file__), "static", "manifest.json")
-    with open(manifest_path, "r") as f:
-        return JSONResponse(json.load(f))
-
-
 # uv run uvicorn main:app --host 0.0.0.0 --port 7860 --reload

run_all.sh

@@ -0,0 +1,37 @@
+#!/bin/bash
+
+# check if mongodb docker container is running and start with docker compose docker/mongodb-docker-compose.yml
+MONGO_CONTAINER_NAME="mongo"
+echo "Starting MongoDB container..."
+echo "Checking if MongoDB container is running..."
+if ! docker ps | grep -q "$MONGO_CONTAINER_NAME"; then
+    echo "Starting MongoDB container..."
+    docker compose -f docker/mongodb-docker-compose.yaml up -d
+    echo "MongoDB container started"
+    echo "Waiting for MongoDB container to be ready..."
+    sleep 10
+    echo "MongoDB container is ready"
+else
+    echo "MongoDB container is already running"
+fi
+
+# stop existing application if running
+echo "--------------------------------"
+echo "checking if application is running..."
+echo "Stopping existing application if running..."
+pkill -f "uvicorn main:app"
+
+# wait for the process to stop
+sleep 2
+
+# start FastAPI application with uvicorn integrated Gradio Application
+echo "--------------------------------"
+echo "Starting application..."
+echo "--------------------------------"
+uv run uvicorn main:app --host 0.0.0.0 --port 7860 --reload
+
+# catch SIGINT (Ctrl+C) to stop the application gracefully
+trap "echo 'Stopping application...'; pkill -f 'uvicorn main:app'; exit" INT
+
+# wait for the application to run
+wait