Update start.sh

start.sh

@@ -5,7 +5,7 @@ export OMP_NUM_THREADS=4
 export MKL_NUM_THREADS=4
 export CUDA_VISIBLE_DEVICES=0
 
-echo "--- Starting Ollama, FastAPI, and Streamlit ---"
+echo "--- Starting Ollama and FastAPI ---"
 
 # Start Ollama in the background
 echo "Starting Ollama server..."
@@ -21,7 +21,7 @@ else
 fi
 
 # Wait for Ollama to start up
-max_attempts=60 # Increased attempts as model pulling can take time
+max_attempts=90 # Increased attempts as model pulling can take time on first run
 attempt=0
 echo "Waiting for Ollama to start (max $max_attempts seconds)..."
 while ! curl -s http://localhost:11434/api/tags >/dev/null; do
@@ -39,28 +39,9 @@ echo "--- Files in current directory ($PWD): ---"
 ls -l
 echo "-------------------------------------------"
 
-# Start the FastAPI server in the background
+# Start the FastAPI server
 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 &
+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..."
-
-# 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
+# The script will stay alive as long as uvicorn is running in the foreground.
+# If uvicorn exits, the container will exit.