diff --git a/docker-compose.yml b/docker-compose.yml
index fa101f7..5724ff3 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,23 +1,23 @@
-version: '3.8'
-
 services:
   codegenerator:
-    build: ./CodeGenerator
+    build:
+      context: .
+      dockerfile: Dockerfile
     container_name: codegenerator
-    ports:
-      - "8000:8000"
-    environment:
-      - OLLAMA_URL=http://host.docker.internal:11434
-    volumes:
-      - /var/run/docker.sock:/var/run/docker.sock  # Docker-in-Docker via host socket
-    restart: unless-stopped
-
-  open-webui:
-    image: ghcr.io/open-webui/open-webui:main
-    container_name: open-webui
-    ports:
-      - "3000:8080"
     environment:
       - OLLAMA_BASE_URL=http://host.docker.internal:11434
-      - WEBUI_SECRET_KEY=change-me  # changer pour la production
+      - DOCKER_HOST=unix:///var/run/docker.sock
+    volumes:
+      - /var/run/docker.sock:/var/run/docker.sock
+      - codegen-data:/app/data
+      - /tmp/docker-data:/tmp/docker-data
+    ports:
+      - "3000:3000"
+    extra_hosts:
+      - "host.docker.internal:host-gateway"
+    privileged: true
     restart: unless-stopped
+
+volumes:
+  codegen-data:
+