websocket api

Dockerfile

@@ -0,0 +1,38 @@
+# 🌐 Base image légère avec Python 3.12
+FROM python:3.12-slim
+
+# 🧱 Installer les dépendances système minimales
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/*
+
+# 👤 Créer un utilisateur non-root recommandé pour Hugging Face
+RUN useradd -m -u 1000 appuser
+
+# 📁 Définir le répertoire de travail
+WORKDIR /app
+
+# 📦 Copier les dépendances
+COPY requirements.txt .
+
+# 🐍 Créer un environnement virtuel dédié
+RUN python -m venv /opt/venv
+ENV PATH="/opt/venv/bin:$PATH"
+
+# 📥 Installer les paquets Python dans le venv
+RUN pip install --no-cache-dir -r requirements.txt
+
+# 📄 Copier le reste du code source
+COPY . .
+
+# 🔐 Définir les permissions
+RUN chown -R appuser:appuser /app
+
+# 👤 Passer à l'utilisateur sécurisé
+USER appuser
+
+# 🔊 Exposer le port attendu par Hugging Face
+EXPOSE 7860
+
+# 🚀 Commande de démarrage de l'application FastAPI
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

main.py

@@ -0,0 +1,39 @@
+from fastapi import FastAPI, WebSocket
+from fastapi.middleware.cors import CORSMiddleware
+import httpx
+import os
+
+app = FastAPI()
+
+# Autoriser l'accès depuis ton app mobile (optionnel selon déploiement)
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # à restreindre en prod
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+LANGSERVE_URL = os.getenv("LANGSERVE_URL", "https://thibautmodrin-vitizen-chat.hf.space/chat/invoke")
+
+@app.websocket("/ws/chat")
+async def websocket_endpoint(websocket: WebSocket):
+    await websocket.accept()
+    async with httpx.AsyncClient() as client:
+        while True:
+            try:
+                data = await websocket.receive_text()
+                request_body = {
+                    "input": {"question": data},
+                    "stream": False
+                }
+
+                response = await client.post(LANGSERVE_URL, json=request_body)
+                response.raise_for_status()
+                json_data = response.json()
+
+                answer = json_data.get("output", {}).get("answer", "[no response]")
+                await websocket.send_text(answer)
+
+            except Exception as e:
+                await websocket.send_text(f"[error] {str(e)}")

requirements.txt

@@ -0,0 +1,3 @@
+fastapi
+uvicorn
+httpx