Update start.sh

start.sh

@@ -1,21 +1,19 @@
 #!/bin/bash
 
-# Startup script to run both FastAPI and Streamlit
+# Startup script optimized for Hugging Face Spaces
+# HF only exposes port 7860, so we need to choose one service
 
 echo "🚀 Starting HTML to PDF Converter..."
 
-# Start FastAPI in the background on port 7860
-echo "📡 Starting FastAPI API on port 7860..."
-python3 api.py &
-API_PID=$!
-
-# Wait a moment for API to start
-sleep 3
-
-# Start Streamlit on port 8501
-echo "🎨 Starting Streamlit UI on port 8501..."
-streamlit run src/streamlit_app.py --server.port=8501 --server.address=0.0.0.0 &
-STREAMLIT_PID=$!
+# Check if we should run API or Streamlit
+if [ "$RUN_MODE" = "api" ]; then
+    echo "📡 Starting FastAPI API on port 7860..."
+    python3 api.py
+else
+    # Default: Run Streamlit (Hugging Face Spaces work best with Streamlit)
+    echo "🎨 Starting Streamlit UI on port 7860..."
+    streamlit run src/streamlit_app.py --server.port=7860 --server.address=0.0.0.0
+fi
 
 # Function to handle shutdown
 cleanup() {