Update start.sh

start.sh

@@ -5,31 +5,62 @@ export OMP_NUM_THREADS=4
 export MKL_NUM_THREADS=4
 export CUDA_VISIBLE_DEVICES=0
 
+echo "--- Starting Ollama, FastAPI, and Streamlit ---"
+
 # Start Ollama in the background
+echo "Starting Ollama server..."
 ollama serve &
 
 # Pull the model if not already present
-# --- CHANGE THIS LINE ---
-if ! ollama list | grep -q "krishna_choudhary/AI_Assistant_Chatbot"; then
-    ollama pull krishna_choudhary/AI_Assistant_Chatbot
+MODEL_TO_PULL="krishna_choudhary/AI_Assistant_Chatbot"
+if ! ollama list | grep -q "$MODEL_TO_PULL"; then
+    echo "Pulling Ollama model: $MODEL_TO_PULL"
+    ollama pull "$MODEL_TO_PULL"
+else
+    echo "Ollama model $MODEL_TO_PULL already present."
 fi
 
 # Wait for Ollama to start up
-max_attempts=30
+max_attempts=60 # Increased attempts as model pulling can take time
 attempt=0
+echo "Waiting for Ollama to start (max $max_attempts seconds)..."
 while ! curl -s http://localhost:11434/api/tags >/dev/null; do
     sleep 1
     attempt=$((attempt + 1))
     if [ $attempt -eq $max_attempts ]; then
-        echo "Ollama failed to start within 30 seconds. Exiting."
+        echo "Ollama failed to start within $((max_attempts)) seconds. Exiting."
         exit 1
     fi
 done
-
 echo "Ollama is ready."
 
-# Print the API URL
-echo "API is running on: http://0.0.0.0:7860"
+# --- DEBUGGING: List files in current directory ---
+echo "--- Files in current directory ($PWD): ---"
+ls -l
+echo "-------------------------------------------"
+
+# Start the FastAPI server in the background
+echo "Starting FastAPI server..."
+# Ensure 'main:app' correctly points to your main.py file and the 'app' object within it.
+uvicorn main:app --host 0.0.0.0 --port 7860 --workers 1 --limit-concurrency 20 &
+
+# Store the PID of the FastAPI server
+FASTAPI_PID=$!
+echo "FastAPI server started with PID: $FASTAPI_PID"
+
+# Wait a moment for FastAPI to start (optional, but good practice)
+sleep 5
+
+# Start the Streamlit server in the background
+echo "Starting Streamlit app..."
+streamlit run streamlit_app.py --server.port 8501 --server.address 0.0.0.0 &
+
+# Store the PID of the Streamlit server
+STREAMLIT_PID=$!
+echo "Streamlit app started with PID: $STREAMLIT_PID"
+
+echo "All services initiated. Keeping container alive..."
 
-# Start the FastAPI server
-uvicorn app:app --host 0.0.0.0 --port 7860 --workers 4 --limit-concurrency 20
+# Keep the script running indefinitely, so the Docker container doesn't exit.
+# This is more robust than `wait -n` if one background process exits unexpectedly.
+tail -f /dev/null