Update Dockerfile

Dockerfile

@@ -1,24 +1,33 @@
 # 1. Base Image
 FROM python:3.9-slim
 
-# 2. Install OpenJDK 21 and system libs for RDKit
+# 2. System dependencies (Added locales and libstdc++6)
 RUN apt-get update && apt-get install -y --no-install-recommends \
     openjdk-21-jre-headless \
     libxrender1 \
     libxext6 \
+    libstdc++6 \
+    locales \
     gcc \
     python3-dev \
     && rm -rf /var/lib/apt/lists/*
 
-# 3. Environment Flags for C3 (Sapphire Rapids) Optimization
-# TF_CPP_MIN_LOG_LEVEL=2 silences INFO/WARNING logs about CPU guards
-# TF_ENABLE_ONEDNN_OPTS=1 enables Intel's high-speed math engine
+# Generate locales to prevent character conversion crashes
+RUN sed -i -e 's/# en_US.UTF-8 UTF-8/en_US.UTF-8 UTF-8/' /etc/locale.gen && \
+    locale-gen
+ENV LANG=en_US.UTF-8 \
+    LANGUAGE=en_US:en \
+    LC_ALL=en_US.UTF-8
+
+# 3. Environment Variables (Critical Fixes for SIGSEGV)
 ENV JAVA_HOME=/usr/lib/jvm/java-21-openjdk-amd64 \
     PATH=$JAVA_HOME/bin:$PATH \
     TF_ENABLE_ONEDNN_OPTS=1 \
     TF_CPP_MIN_LOG_LEVEL=2 \
-    OMP_NUM_THREADS=8 \
-    KMP_AFFINITY=granularity=fine,compact,1,0 \
+    # Forces C++ libs to resolve early and avoid conflicts
+    LD_BIND_NOW=1 \
+    # Force the standard C++ lib to be pre-loaded
+    LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libstdc++.so.6 \
     HOME=/home/user
 
 # 4. Hugging Face User Setup
@@ -27,8 +36,7 @@ USER user
 WORKDIR /home/user/app
 ENV PATH=/home/user/.local/bin:$PATH
 
-# 5. Install Intel-Optimized TensorFlow and STOUT
-# 'intel-tensorflow' replaces standard tensorflow for AVX512/VNNI support
+# 5. Install Python Stack
 RUN pip install --no-cache-dir wheel && \
     pip install --no-cache-dir \
     intel-tensorflow \
@@ -37,16 +45,15 @@ RUN pip install --no-cache-dir wheel && \
 # 6. Create app.py
 RUN cat <<EOF > app.py
 import os
-# Set log levels before importing TF to ensure silence
-os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
-
+import uvicorn
 from fastapi import FastAPI, HTTPException, Body
 from STOUT import translate_forward, translate_reverse
 from rdkit import Chem
 from typing import List
 
-app = FastAPI(title="STOUT V2 Optimized API")
+app = FastAPI(title="STOUT V2 SIGSEGV-Fixed API")
 
+# Helper to ensure RDKit doesn't break
 def clean_smi(s):
     try:
         m = Chem.MolFromSmiles(s)
@@ -54,8 +61,7 @@ def clean_smi(s):
     except: return None
 
 @app.get("/")
-def root():
-    return {"message": "STOUT V2 - Optimized for Intel Sapphire Rapids"}
+def root(): return {"message": "STOUT V2 - Stable Build"}
 
 @app.api_route("/smiles_to_iupac", methods=["GET", "POST"])
 def s2i(smiles: str):
@@ -63,25 +69,19 @@ def s2i(smiles: str):
     if not s: raise HTTPException(400, "Invalid SMILES")
     return {"iupac": translate_forward(s)}
 
-@app.api_route("/iupac_to_smiles", methods=["GET", "POST"])
-def i2s(iupac: str):
-    try: return {"smiles": translate_reverse(iupac)}
-    except: raise HTTPException(400, "Invalid IUPAC")
-
-@app.post("/batch/smiles_to_iupac")
-def batch_s2i(smiles_list: List[str] = Body(...)):
-    return [{"smiles": s, "iupac": translate_forward(clean_smi(s))} for s in smiles_list if clean_smi(s)]
-
 @app.get("/health")
 def health(): return {"status": "healthy"}
+
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)
 EOF
 
-# 7. Final Config
+# 7. Expose Port and Start
 EXPOSE 7860
 
-# 4 workers is optimal for 8 vCPUs on C3 to balance memory and AMX utilization
+# Reduce workers to 2 or 3 if memory crashes persist; 4 is the limit for 32GB
 CMD ["gunicorn", "app:app", \
-     "--workers", "4", \
+     "--workers", "3", \
      "--worker-class", "uvicorn.workers.UvicornWorker", \
      "--bind", "0.0.0.0:7860", \
      "--timeout", "300"]