start ollama and pull from bash

entrypoint.sh

@@ -4,18 +4,15 @@ set -x # Enable command tracing for debugging
 
 export PATH="$HOME/.local/bin:$PATH"
 
-# # Start Ollama server in the background
-# ollama serve &
+# Start Ollama server in the background
+if ! nc -z localhost 7860; then
+  ollama serve &
+fi
 
-# # Wait for the server to be ready
-# while ! nc -z localhost 7860; do
-#     echo "Waiting for Ollama server to start..."
-#     sleep 1
-# done
-
-# # Pull the model
-# echo "Pulling the model..."
-# ollama pull gemma3:1b
+# Pull the model
+timeout 15 bash -c 'until nc -z localhost 7860; do echo "Waiting for Ollama server..."; sleep 1; done'
+echo "Pulling the model..."
+ollama pull codellama:7b
 
 # Start streamlit app
 streamlit run mycoder/app.py --server.port=8501 --server.enableCORS=false

mycoder/app.py

@@ -35,9 +35,4 @@ if st.button("Run Query2Plot"):
     if os.path.exists("output/plot.png"):
         st.image("output/plot.png", caption="Generated Figure")
 
-if st.button("DEMO START OLLAMA"):
-    from ollamaini import start_ollama, start_model
-    start_ollama()
-    import time
-    time.sleep(20)  # Wait for Ollama to start
-    start_model()
+