qwen real

Dockerfile

@@ -1,19 +1,15 @@
-FROM python:3.9-slim
+FROM chromadb/chroma:latest
 
 WORKDIR /app
 
-# Install requirements
-COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+# Copy the start script
+COPY start.sh .
 
-# Copy all files
-COPY . .
-
-# Make start.sh executable
+# Make it executable
 RUN chmod +x start.sh
 
-EXPOSE 8000
+# Expose ChromaDB port
 EXPOSE 7860
 
-# Run the startup script which handles both ChromaDB and FastAPI
-CMD ["./start.sh"]
+# Run the start script
+CMD ["./start.sh"]

start.sh

@@ -1,26 +1,24 @@
 #!/bin/bash
 
-echo "=== Starting ChromaDB and FastAPI ==="
+echo "=== [HF Space] Starting ChromaDB with Persistent Storage ==="
+echo "Time: $(date)"
 
-# Define the persistent directory
+# Define the persistent directory inside the mounted HF Bucket
 PERSIST_DIR="/data/chroma_db"
+
+# Create the directory if it doesn't exist
+# HF Buckets are mounted at /data, so this ensures the folder exists
 mkdir -p $PERSIST_DIR
 
-# Start ChromaDB in background
-echo "🚀 Starting ChromaDB on port 8000..."
+echo "📂 Persistence directory: $PERSIST_DIR"
+echo "✅ Directory contents:"
+ls -la $PERSIST_DIR || echo "Directory is empty or new."
+
+# Set environment variables for ChromaDB
 export CHROMA_SERVER_HOST=0.0.0.0
-export CHROMA_SERVER_HTTP_PORT=8000
+export CHROMA_SERVER_HTTP_PORT=7860
 export PERSIST_DIRECTORY=$PERSIST_DIR
-chroma run --path $PERSIST_DIR --host 0.0.0.0 --port 8000 &
-CHROMA_PID=$!
-
-# Wait for ChromaDB to start
-sleep 3
-
-# Start FastAPI app
-echo "🌐 Starting FastAPI app on port 7860..."
-python3 app.py &
-APP_PID=$!
 
-# Keep both processes alive
-wait $CHROMA_PID $APP_PID
+echo "🚀 Starting ChromaDB Server..."
+# 'chroma run' is the command in the official image to start the server
+exec chroma run --path $PERSIST_DIR --host 0.0.0.0 --port 7860