Prepare for Hugging Face deployment: add Dockerfile, FastAPI /health endpoint, and sync workflow

.github/workflows/hf_sync.yml

@@ -0,0 +1,18 @@
+name: Sync to Hugging Face hub
+on:
+  push:
+    branches: [main]
+  workflow_dispatch:
+
+jobs:
+  sync-to-hub:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v3
+        with:
+          fetch-depth: 0
+          lfs: true
+      - name: Push to hub
+        env:
+          HF_TOKEN: ${{ secrets.HF_TOKEN }}
+        run: git push --force https://AUXteam:$HF_TOKEN@huggingface.co/spaces/AUXteam/tiny_factory main

.hfignore

@@ -0,0 +1,6 @@
+.git
+__pycache__
+.venv
+.env
+*.log
+.pytest_cache

Dockerfile

@@ -0,0 +1,32 @@
+FROM python:3.11-slim
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    build-essential \
+    curl \
+    software-properties-common \
+    git \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install uv
+COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/
+
+# Set up a non-root user
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+# Copy requirements and install
+COPY --chown=user requirements.txt .
+RUN uv pip install --system -r requirements.txt
+
+# Copy the rest of the application
+COPY --chown=user . .
+
+# Expose the port
+EXPOSE 7860
+
+# Command to run the application
+CMD ["python", "app.py"]

README.md

@@ -3,9 +3,8 @@ title: Tiny Factory
 emoji: 💻
 colorFrom: yellow
 colorTo: gray
-sdk: gradio
-sdk_version: 6.3.0
-app_file: app.py
+sdk: docker
+app_port: 7860
 pinned: false
 ---
 

app.py

@@ -3,6 +3,8 @@ import os
 import gradio as gr
 import json
 from tinytroupe.factory import TinyPersonFactory
+from fastapi import FastAPI
+import uvicorn
 
 # --- CHANGE 1: The function now accepts an optional API key. ---
 def generate_personas(business_description, customer_profile, num_personas, blablador_api_key=None):
@@ -82,5 +84,14 @@ with gr.Blocks() as demo:
         api_name="generate_personas"
     )
 
+app = FastAPI()
+
+@app.get("/health")
+def health():
+    return {"status": "ok"}
+
+# Mount Gradio app to FastAPI
+app = gr.mount_gradio_app(app, demo, path="/")
+
 if __name__ == "__main__":
-    demo.queue().launch()
+    uvicorn.run(app, host="0.0.0.0", port=7860)

requirements.txt

@@ -22,3 +22,5 @@ textdistance
 scipy
 transformers==4.38.2
 huggingface-hub==0.22.2
+fastapi
+uvicorn