Add Gradio UI service and update Docker config for dual API/UI setup

Dockerfile

@@ -21,6 +21,7 @@ ENV HOME=/home/user  \
 
 # Exposer le port sur lequel l'application va tourner
 EXPOSE 7860
+EXPOSE 8000
 
 
 # RUN --mount=type=secret,id=api_read,mode=0444,required=true \
@@ -30,4 +31,4 @@ EXPOSE 7860
 #   git clone $(cat /run/secrets/api_read)
 
 # Commande pour lancer l'application
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]

app.py

@@ -0,0 +1,24 @@
+# Interface Gradio pour afficher la liste des modèles chargés
+import gradio as gr
+import asyncio
+from db.models import fetch_models_for_group
+from models.loader import load_models, model_pipelines
+from config.settings import RESOURCE_GROUP
+
+async def init_models():
+    """Charger les modèles au démarrage"""
+    models_data = await fetch_models_for_group(RESOURCE_GROUP)
+    await load_models(models_data)
+
+# Initialisation des modèles
+asyncio.run(init_models())
+
+def get_model_list():
+    """Retourner la liste des modèles chargés"""
+    return list(model_pipelines.keys())
+
+with gr.Blocks() as demo:
+    demo.title("Tamis AI - Liste des modèles chargés")
+    gr.JSON(get_model_list)
+
+demo.launch()

docker-compose.yml

@@ -1,11 +1,22 @@
 version: '3.8'
 
 services:
-  inference-lamp-api:
+  api:
     build: .
+    command: uvicorn main:app --host 0.0.0.0 --port 8000
     ports:
       - "8000:8000"
     volumes:
       - .:/app
+    env_file:
+      - .env
+
+  ui:
+    build: .
+    command: python app.py
+    ports:
+      - "7860:7860"
+    volumes:
+      - .:/app
     env_file:
       - .env

requirements.txt

@@ -8,4 +8,5 @@ huggingface_hub
 torch
 numpy
 httpx
-asyncpg
+asyncpg
+gradio