Upload 2 files

Dockerfile

@@ -13,7 +13,7 @@ RUN pip install --no-cache-dir -r requirements.txt
 # Copy project files into the container
 COPY . ./
 
-# Make sure the demo.launcher script is executable
+# Make the demo.launcher script executable
 RUN chmod +x demo.launcher
 
 # Add a non-root user and change ownership of necessary files
@@ -22,7 +22,7 @@ RUN useradd -ms /bin/bash user && chown -R user:user ./server.py ./client.py ./d
 # Switch to the non-root user
 USER user
 
-# Expose the port that Gradio will use
+# Expose the ports (Gradio on 7860, MCP on 8000)
 EXPOSE 7860 8000
 
 # Start the demo.launcher script to run everything

demo.launcher

@@ -1,13 +1,20 @@
 #!/bin/bash
-# Pull your Ollama model (can be moved to build if desired)
+
+# Ensure that Ollama model is pulled
+echo "Pulling Ollama model (granite3.1-moe)..."
 ollama pull granite3.1-moe
+
+# Start the Ollama API server in the background
+echo "Starting Ollama API server..."
 ollama serve &
-sleep 5
+sleep 15  # Increased wait time for Ollama to initialize
 
 # Start MCP server in the background
+echo "Starting MCP server..."
 python3 server.py &
-sleep 5  # Ensure server is fully initialized
+sleep 5  # Ensure the MCP server is fully initialized
 
-# Start Gradio client
+# Start Gradio client (wait to ensure everything is ready)
+echo "Starting Gradio client..."
 python3 client.py --server_name 0.0.0.0 --server_port 7860