cmc

Dockerfile

@@ -1,24 +1,40 @@
+# ---------- Base ----------
 FROM python:3.11-slim
 
-# install build deps
+# ---------- System Dependencies ----------
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
+    git \
+    wget \
     && rm -rf /var/lib/apt/lists/*
 
+# ---------- Working Directory ----------
 WORKDIR /code
 
-# create writable cache dir
-RUN mkdir -p /code/.cache && chmod 777 /code/.cache
+# ---------- Environment Variables ----------
+ENV PYTHONUNBUFFERED=1
+ENV PYTHONDONTWRITEBYTECODE=1
+ENV HF_HOME=/code/.cache/huggingface
+ENV TRANSFORMERS_CACHE=/code/.cache/huggingface
+ENV TORCH_HOME=/code/.cache/torch
 
-# install python deps
+# ---------- Install Python Dependencies ----------
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --upgrade pip && pip install --no-cache-dir -r requirements.txt
 
-# ---- NEW: pre-download embedding model (build-time, online) ----
+# ---------- Pre-Download Embedding Model ----------
 COPY pre_download.py .
-RUN HF_HOME=/code/.cache python pre_download.py
+RUN python pre_download.py
 
-# copy app code
+# ---------- Copy Application Code ----------
 COPY . .
 
-CMD ["gunicorn", "app:app", "-b", "0.0.0.0:7860"]
+# ---------- Ensure FAISS and Cache Dirs Exist ----------
+RUN mkdir -p /code/faiss_db /code/.cache && chmod -R 777 /code/faiss_db /code/.cache
+
+# ---------- Expose Web Port ----------
+EXPOSE 7860
+
+# ---------- Start the App ----------
+CMD ["gunicorn", "app:app", "-b", "0.0.0.0:7860", "--timeout", "300"]
+

app.py

@@ -1,36 +1,23 @@
-# app.py
-from __future__ import annotations
-import os
 from flask import Flask, request, jsonify
-from rag import ask_question
 
-# ------------------------------------------------------------------ APP INIT
 app = Flask(__name__)
 
 @app.route("/webhook", methods=["POST"])
 def webhook():
-    """
-    Handles POST requests from WhatsApp or frontend chat clients.
-    Expects JSON payload:
-        { "phone": "<number>", "question": "<message>" }
-    """
-    try:
-        payload = request.get_json(force=True)
-        phone = payload.get("phone", "").strip()
-        question = payload.get("question", "").strip()
+    data = request.get_json()
+    question = data.get("question")
+    phone = data.get("phone")
 
-        if not phone or not question:
-            return jsonify({"error": "Missing 'phone' or 'question' field"}), 400
+    # retrieve docs from FAISS or Supabase (pseudo)
+    retrieved_docs = retriever.get_relevant_documents(question)
 
-        answer, docs = ask_question(phone, question)
-        return jsonify({"answer": answer, "docs": len(docs)})
+    if not retrieved_docs:
+        return jsonify({"answer": "I couldn’t find relevant info on that yet.", "docs": 0})
 
-    except Exception as e:
-        print(f"❌ Webhook error: {e}")
-        return jsonify({"error": str(e)}), 500
+    # Generate answer
+    answer = qa_chain.invoke({"question": question, "context": retrieved_docs})
+    return jsonify({"answer": answer, "docs": len(retrieved_docs)})
 
-
-# ------------------------------------------------------------------ ENTRY POINT
 if __name__ == "__main__":
-    port = int(os.getenv("PORT", 7860))
-    app.run(host="0.0.0.0", port=port)
+    app.run(host="0.0.0.0", port=7860)
+

pre_download.py

@@ -1,4 +1,7 @@
-# pre_download.py
 from sentence_transformers import SentenceTransformer
-SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
-print("✓ Embedding model cached at build time")
+
+if __name__ == "__main__":
+    print("🔹 Pre-downloading embedding model: all-MiniLM-L6-v2 ...")
+    model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
+    model.encode(["test"])  # trigger download
+    print("✅ Embedding model cached successfully.")