Clean docker containers

Author: gabriel-piles

Dockerfile.gradio

@@ -31,5 +31,6 @@ WORKDIR /app
 
 COPY ./src/gradio_app.py ./src/gradio_app.py
 
+ENV PYTHONUNBUFFERED=1
 ENV PYTHONPATH "${PYTHONPATH}:/app/src"
 

Makefile

@@ -24,7 +24,7 @@ ifeq ($(HAS_GPU), 1)
 		echo "Warning: Ollama GPU container may not be fully healthy yet, but continuing..."; \
 	fi
 	@echo "Starting all services with translation support..."
-	docker compose -f docker-compose-gpu.yml up --build pdf-document-layout-analysis-gpu-translation pdf-document-layout-analysis-gui-gpu-translation
+	docker compose -f docker-compose-gpu.yml up --build pdf-document-layout-analysis-gpu pdf-document-layout-analysis-gui-gpu
 else
 	@echo "No NVIDIA GPU detected, starting with translation support (CPU Ollama)"
 	@echo "Starting Ollama container first..."
@@ -43,7 +43,7 @@ else
 		echo "Warning: Ollama container may not be fully healthy yet, but continuing..."; \
 	fi
 	@echo "Starting all services with translation support..."
-	docker compose -f docker-compose.yml up --build pdf-document-layout-analysis-translation pdf-document-layout-analysis-gui-translation
+	docker compose -f docker-compose.yml up --build pdf-document-layout-analysis pdf-document-layout-analysis-gui
 endif
 
 
@@ -66,7 +66,7 @@ start_no_gpu:
 		echo "Warning: Ollama container may not be fully healthy yet, but continuing..."; \
 	fi
 	@echo "Starting all services with translation support..."
-	docker compose up --build pdf-document-layout-analysis-translation pdf-document-layout-analysis-gui-translation
+	docker compose up --build pdf-document-layout-analysis pdf-document-layout-analysis-gui
 
 stop:
 	docker compose stop

docker-compose-gpu.yml

@@ -19,29 +19,6 @@ services:
     entrypoint: [ "gunicorn", "-k", "uvicorn.workers.UvicornWorker", "--chdir", "./src", "app:app", "--bind", "0.0.0.0:5060", "--timeout", "10000"]
     init: true
     restart: unless-stopped
-    build:
-      context: .
-      dockerfile: Dockerfile
-    ports:
-      - "5060:5060"
-    deploy:
-      resources:
-        reservations:
-          devices:
-            - driver: nvidia
-              count: 1
-              capabilities: [ gpu ]
-    environment:
-      - RESTART_IF_NO_GPU=$RESTART_IF_NO_GPU
-      - OLLAMA_HOST=http://localhost:11434
-    networks:
-      - pdf-analysis-network
-
-  pdf-document-layout-analysis-gpu-translation:
-    container_name: pdf-document-layout-analysis-gpu-translation
-    entrypoint: [ "gunicorn", "-k", "uvicorn.workers.UvicornWorker", "--chdir", "./src", "app:app", "--bind", "0.0.0.0:5060", "--timeout", "10000"]
-    init: true
-    restart: unless-stopped
     build:
       context: .
       dockerfile: Dockerfile
@@ -68,23 +45,6 @@ services:
     entrypoint: [ "python", "./src/gradio_app.py" ]
     init: true
     restart: unless-stopped
-    build:
-      context: .
-      dockerfile: Dockerfile.gradio
-    ports:
-      - "7860:7860"
-    depends_on:
-      - pdf-document-layout-analysis-gpu
-    environment:
-      - API_BASE_URL=http://pdf-document-layout-analysis-gpu:5060
-    networks:
-      - pdf-analysis-network
-
-  pdf-document-layout-analysis-gui-gpu-translation:
-    container_name: pdf-document-layout-analysis-gui-gpu-translation
-    entrypoint: [ "python", "./src/gradio_app.py" ]
-    init: true
-    restart: unless-stopped
     build:
       context: .
       dockerfile: Dockerfile.gradio
@@ -93,10 +53,10 @@ services:
     depends_on:
       ollama-gpu:
         condition: service_healthy
-      pdf-document-layout-analysis-gpu-translation:
+      pdf-document-layout-analysis-gpu:
         condition: service_started
     environment:
-      - API_BASE_URL=http://pdf-document-layout-analysis-gpu-translation:5060
+      - API_BASE_URL=http://pdf-document-layout-analysis-gpu:5060
     networks:
       - pdf-analysis-network
 

docker-compose.yml

@@ -21,21 +21,6 @@ services:
     entrypoint: [ "gunicorn", "-k", "uvicorn.workers.UvicornWorker", "--chdir", "./src", "app:app", "--bind", "0.0.0.0:5060", "--timeout", "10000"]
     init: true
     restart: unless-stopped
-    build:
-      context: .
-      dockerfile: Dockerfile
-    ports:
-      - "5060:5060"
-    environment:
-      - OLLAMA_HOST=http://localhost:11434
-    networks:
-      - pdf-analysis-network
-
-  pdf-document-layout-analysis-translation:
-    container_name: pdf-document-layout-analysis-translation
-    entrypoint: [ "gunicorn", "-k", "uvicorn.workers.UvicornWorker", "--chdir", "./src", "app:app", "--bind", "0.0.0.0:5060", "--timeout", "10000"]
-    init: true
-    restart: unless-stopped
     build:
       context: .
       dockerfile: Dockerfile
@@ -54,23 +39,6 @@ services:
     entrypoint: [ "python", "./src/gradio_app.py" ]
     init: true
     restart: unless-stopped
-    build:
-      context: .
-      dockerfile: Dockerfile.gradio
-    ports:
-      - "7860:7860"
-    depends_on:
-      - pdf-document-layout-analysis
-    environment:
-      - API_BASE_URL=http://pdf-document-layout-analysis:5060
-    networks:
-      - pdf-analysis-network
-
-  pdf-document-layout-analysis-gui-translation:
-    container_name: pdf-document-layout-analysis-gui-translation
-    entrypoint: [ "python", "./src/gradio_app.py" ]
-    init: true
-    restart: unless-stopped
     build:
       context: .
       dockerfile: Dockerfile.gradio
@@ -79,10 +47,10 @@ services:
     depends_on:
       ollama:
         condition: service_healthy
-      pdf-document-layout-analysis-translation:
+      pdf-document-layout-analysis:
         condition: service_started
     environment:
-      - API_BASE_URL=http://pdf-document-layout-analysis-translation:5060
+      - API_BASE_URL=http://pdf-document-layout-analysis:5060
     networks:
       - pdf-analysis-network
 

justfile

@@ -75,7 +75,7 @@ start:
 			echo "Warning: Ollama GPU container may not be fully healthy yet, but continuing..."
 		fi
 		echo "Starting all services with translation support..."
-		docker compose -f docker-compose-gpu.yml up --build pdf-document-layout-analysis-gpu-translation pdf-document-layout-analysis-gui-gpu-translation
+		docker compose -f docker-compose-gpu.yml up --build pdf-document-layout-analysis-gpu pdf-document-layout-analysis-gui-gpu
 	else
 		echo "No NVIDIA GPU detected, starting with translation support (CPU Ollama)"
 		echo "Starting Ollama container first..."
@@ -95,7 +95,7 @@ start:
 			echo "Warning: Ollama container may not be fully healthy yet, but continuing..."
 		fi
 		echo "Starting all services with translation support..."
-		docker compose -f docker-compose.yml up --build pdf-document-layout-analysis-translation pdf-document-layout-analysis-gui-translation
+		docker compose -f docker-compose.yml up --build pdf-document-layout-analysis pdf-document-layout-analysis-gui
 	fi
 
 start_no_gpu:
@@ -119,7 +119,7 @@ start_no_gpu:
 		echo "Warning: Ollama container may not be fully healthy yet, but continuing..."
 	fi
 	echo "Starting all services with translation support..."
-	docker compose up --build pdf-document-layout-analysis-translation pdf-document-layout-analysis-gui-translation
+	docker compose up --build pdf-document-layout-analysis pdf-document-layout-analysis-gui
 
 start_no_translation:
 	mkdir -p ./models

src/gradio_app.py

@@ -5,10 +5,44 @@
 import tempfile
 import base64
 import zipfile
+import time
 
 API_BASE_URL = os.environ.get("API_BASE_URL", "http://localhost:5060")
 
 
+def wait_for_backend(max_retries: int = 60, retry_interval: int = 2) -> bool:
+    print("\n" + "=" * 80, flush=True)
+    print("⏳ Waiting for PDF Analysis backend service to be ready...".center(80), flush=True)
+    print("=" * 80 + "\n", flush=True)
+
+    for attempt in range(1, max_retries + 1):
+        try:
+            response = requests.get(f"{API_BASE_URL}/info", timeout=5)
+            if response.status_code == 200:
+                print("=" * 80, flush=True)
+                print("✅  PDF DOCUMENT LAYOUT ANALYSIS UI IS READY!".center(80), flush=True)
+                print("=" * 80, flush=True)
+                print("", flush=True)
+                print("🌐 Access the UI at:", flush=True)
+                print("   → http://localhost:7860", flush=True)
+                print("", flush=True)
+                print(f"🔗 Backend API: {API_BASE_URL}", flush=True)
+                print("", flush=True)
+                print("=" * 80, flush=True)
+                print("=" * 80 + "\n", flush=True)
+                return True
+        except (requests.exceptions.ConnectionError, requests.exceptions.Timeout):
+            time.sleep(retry_interval)
+        except Exception as e:
+            print(f"   Unexpected error while checking backend: {str(e)}", flush=True)
+            time.sleep(retry_interval)
+
+    print("\n" + "=" * 80, flush=True)
+    print("❌ Backend service did not become ready in time!".center(80), flush=True)
+    print("=" * 80 + "\n", flush=True)
+    return False
+
+
 def validate_pdf_file(pdf_file):
     """Validate that the uploaded file is a valid PDF"""
     if pdf_file is None:
@@ -901,7 +935,8 @@ def convert_to_html(
 
 
 if __name__ == "__main__":
-    print("Starting Gradio interface...")
-    print(f"API Base URL: {API_BASE_URL}")
+    if not wait_for_backend():
+        print("⚠️  Starting UI anyway, but backend may not be available.", flush=True)
+        print("   Please check that the API service is running.\n", flush=True)
 
     app.launch(server_name="0.0.0.0", server_port=7860, share=False, show_error=True, max_file_size="100mb")