Upload 2 files

Dockerfile

@@ -23,5 +23,5 @@ COPY app.py .
 # Expose the Streamlit port
 EXPOSE 8501
 
-# Start Ollama in the background, wait 5 seconds, then run Streamlit
-CMD ollama serve & sleep 5 && streamlit run app.py --server.port 8501 --server.address 0.0.0.0
+# Start Ollama in the background, log output, wait 10 seconds, then run Streamlit
+CMD ollama serve > ollama.log 2>&1 & sleep 10 && streamlit run app.py --server.port 8501 --server.address 0.0.0.0

app.py

@@ -7,6 +7,7 @@ from atomic_agents.agents.base_agent import BaseAgent, BaseAgentConfig, BaseAgen
 from dotenv import load_dotenv
 import asyncio
 import httpx
+import time
 
 # Load environment variables
 load_dotenv()
@@ -15,14 +16,20 @@ load_dotenv()
 st.title("Math Reasoning Chatbot")
 st.write("Select a provider and chat with the bot to solve math problems!")
 
-# Function to check if the Ollama server is running
-async def check_ollama_health():
-    try:
-        async with httpx.AsyncClient() as client:
-            response = await client.get("http://localhost:11434/v1")
-            return response.status_code == 200
-    except httpx.RequestError:
-        return False
+# Function to check if the Ollama server is running with retries
+async def check_ollama_health(max_retries=5, retry_delay=5):
+    for attempt in range(max_retries):
+        try:
+            async with httpx.AsyncClient() as client:
+                response = await client.get("http://localhost:11434/v1")
+                if response.status_code == 200:
+                    return True
+        except httpx.RequestError:
+            if attempt < max_retries - 1:
+                st.warning(f"Ollama server not yet available (attempt {attempt + 1}/{max_retries}). Retrying in {retry_delay} seconds...")
+                await asyncio.sleep(retry_delay)
+            continue
+    return False
 
 # Function to set up the client based on the chosen provider
 def setup_client(provider):
@@ -37,9 +44,9 @@ def setup_client(provider):
         display_model = "OpenAI (gpt-4o-mini)"
     elif provider == "ollama":
         from openai import AsyncOpenAI as OllamaClient
-        # Check if Ollama server is running
+        # Check if Ollama server is running with retries
         if not asyncio.run(check_ollama_health()):
-            st.error("Ollama server is not running or not accessible at http://localhost:11434. Please try again later or select a different provider.")
+            st.error("Ollama server is not running or not accessible at http://localhost:11434 after multiple attempts. Please try again later or select a different provider.")
             return None, None, None
         client = instructor.from_openai(
             OllamaClient(base_url="http://localhost:11434/v1", api_key="ollama"), mode=instructor.Mode.JSON