Update Dockerfile

Dockerfile

@@ -1,24 +1,39 @@
 FROM python:3.9-slim
 
-# Install system dependencies
+# Install system dependencies + Java for Elasticsearch
 RUN apt-get update && apt-get install -y \
-    curl \
+    curl openjdk-11-jdk wget gnupg unzip \
     && rm -rf /var/lib/apt/lists/*
 
 # Install Python dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Install Elasticsearch
+ENV ES_VERSION=7.17.23
+RUN wget https://artifacts.elastic.co/downloads/elasticsearch/elasticsearch-${ES_VERSION}-linux-x86_64.tar.gz && \
+    tar -xzf elasticsearch-${ES_VERSION}-linux-x86_64.tar.gz && \
+    mv elasticsearch-${ES_VERSION} /usr/share/elasticsearch && \
+    rm elasticsearch-${ES_VERSION}-linux-x86_64.tar.gz
+
+ENV PATH="/usr/share/elasticsearch/bin:${PATH}"
+ENV ES_HOME=/usr/share/elasticsearch
+
 # Copy application code
 COPY . .
 
-# Expose ports (Gradio + Argilla)
+# Expose ports (Gradio + Argilla + Elasticsearch)
 EXPOSE 7860
 EXPOSE 6900
+EXPOSE 9200
 
-# Health check
+# Health check (check Argilla, not ES directly)
 HEALTHCHECK --interval=30s --timeout=30s --start-period=30s --retries=3 \
     CMD curl -f http://localhost:6900/api/docs || exit 1
 
-# Run startup script
-CMD ["python", "app.py"]
+# Start both Elasticsearch and your app
+CMD ["/bin/bash", "-c", "\
+    elasticsearch -d -Ediscovery.type=single-node && \
+    sleep 15 && \
+    python app.py \
+"]