feat(frontend): create fastapi backend and JS frontend for interacting with tuned model

Author: dineshpinto

Dockerfile

@@ -1,20 +1,43 @@
-# Stage 1: Build Backend
+# Stage 1: Build Frontend
+FROM node:18-alpine AS frontend-builder
+WORKDIR /frontend
+COPY chat-ui/ .
+RUN npm install
+RUN npm run build
+
+# Stage 2: Build Backend
 FROM ghcr.io/astral-sh/uv:python3.12-bookworm-slim AS backend-builder
 ADD . /flare-ai-social
 WORKDIR /flare-ai-social
 RUN uv sync --frozen
 
-# Stage 2: Final Image
+# Stage 3: Final Image
 FROM ghcr.io/astral-sh/uv:python3.12-bookworm-slim
 
+# Install nginx
+RUN apt-get update && apt-get install -y nginx supervisor curl && \
+    rm -rf /var/lib/apt/lists/*
+
 WORKDIR /app
 COPY --from=backend-builder /flare-ai-social/.venv ./.venv
 COPY --from=backend-builder /flare-ai-social/src ./src
 COPY --from=backend-builder /flare-ai-social/pyproject.toml .
 COPY --from=backend-builder /flare-ai-social/README.md .
 
+# Copy frontend files
+COPY --from=frontend-builder /frontend/build /usr/share/nginx/html
+
+# Copy nginx configuration
+COPY nginx.conf /etc/nginx/sites-enabled/default
+
+# Setup supervisor configuration
+COPY supervisord.conf /etc/supervisor/conf.d/supervisord.conf
+
 # Allow workload operator to override environment variables
-LABEL "tee.launch_policy.allow_env_override"="GEMINI_API_KEY,GEMINI_MODEL"
+LABEL "tee.launch_policy.allow_env_override"="GEMINI_API_KEY,TUNED_MODEL_NAME,SIMULATE_ATTESTATION"
 LABEL "tee.launch_policy.log_redirect"="always"
 
-CMD ["uv", "run", "start-social"]
+EXPOSE 80
+
+# Start supervisor (which will start both nginx and the backend)
+CMD ["/usr/bin/supervisord", "-c", "/etc/supervisor/conf.d/supervisord.conf"]

README.md

@@ -4,9 +4,7 @@ Flare AI Kit template for Social AI Agents.
 
 ## 🏗️ Build & Run Instructions
 
-### Fine tuning a model over a dataset
-
-1. **Prepare the Environment File:**  
+**Prepare the Environment File:**  
    Rename `.env.example` to `.env` and update the variables accordingly.
    Some parameters are specific to model fine-tuning:
 
@@ -18,25 +16,48 @@ Flare AI Kit template for Social AI Agents.
    | `batch_size`          | Number of examples to use in each training batch.                          | `4`                                  |
    | `learning_rate`       | Step size multiplier for the gradient updates.                             | `0.001`                              |
 
-2. **Prepare a dataset:**
+### Fine tuning a model over a dataset
+
+1. **Prepare a dataset:**
    An example dataset is provided in `src/data/training_data.json`, which consists of tweet from
    [Hugo Philion's X](https://x.com/HugoPhilion) account. You can use any publicly available dataset
    for model fine-tuning.
 
-3. **Tune a new model**
+2. **Tune a new model**
    Set the name of the new tuned model in `src/flare_ai_social/tune_model.py`, then:
 
    ```bash
    uv run start-tuning
    ```
 
-4. **Observe loss parameters:**
+3. **Observe loss parameters:**
    After tuning in complete, a training loss PNG will be saved in the root folder corresponding to the new model.
    Ideally the loss should minimize to near 0 after several training epochs.
 
-5. **Test the new model**
-   Select the new tuned model and test it against a set of prompts:
+4. **Test the new model**
+   Select the new tuned model and compare it against a set of prompting techniques (zero-shot, few-shot and chain-of-thought):
+
+   ```bash
+   uv run start-compare
+   ```
+
+### Build using Docker (Recommended)
+
+**Note:** You can only perform this step once you have finishing training a new model.
+
+The Docker setup mimics a TEE environment and includes an Nginx server for routing, while Supervisor manages both the backend and frontend services in a single container.
+
+1. **Build the Docker Image:**
+
+   ```bash
+   docker build -t flare-ai-social .
+   ```
+
+2. **Run the Docker Container:**
 
    ```bash
-   uv run start-social
+   docker run -p 80:80 -it --env-file .env flare-ai-social
    ```
+
+3. **Access the Frontend:**  
+   Open your browser and navigate to [http://localhost:80](http://localhost:80) to interact with the Chat UI.

chat-ui/.gitignore

@@ -0,0 +1,23 @@
+# See https://help.github.com/articles/ignoring-files/ for more about ignoring files.
+
+# dependencies
+/node_modules
+/.pnp
+.pnp.js
+
+# testing
+/coverage
+
+# production
+/build
+
+# misc
+.DS_Store
+.env.local
+.env.development.local
+.env.test.local
+.env.production.local
+
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*

chat-ui/README.md

@@ -0,0 +1,70 @@
+# Getting Started with Create React App
+
+This project was bootstrapped with [Create React App](https://github.com/facebook/create-react-app).
+
+## Available Scripts
+
+In the project directory, you can run:
+
+### `npm start`
+
+Runs the app in the development mode.\
+Open [http://localhost:3000](http://localhost:3000) to view it in your browser.
+
+The page will reload when you make changes.\
+You may also see any lint errors in the console.
+
+### `npm test`
+
+Launches the test runner in the interactive watch mode.\
+See the section about [running tests](https://facebook.github.io/create-react-app/docs/running-tests) for more information.
+
+### `npm run build`
+
+Builds the app for production to the `build` folder.\
+It correctly bundles React in production mode and optimizes the build for the best performance.
+
+The build is minified and the filenames include the hashes.\
+Your app is ready to be deployed!
+
+See the section about [deployment](https://facebook.github.io/create-react-app/docs/deployment) for more information.
+
+### `npm run eject`
+
+**Note: this is a one-way operation. Once you `eject`, you can't go back!**
+
+If you aren't satisfied with the build tool and configuration choices, you can `eject` at any time. This command will remove the single build dependency from your project.
+
+Instead, it will copy all the configuration files and the transitive dependencies (webpack, Babel, ESLint, etc) right into your project so you have full control over them. All of the commands except `eject` will still work, but they will point to the copied scripts so you can tweak them. At this point you're on your own.
+
+You don't have to ever use `eject`. The curated feature set is suitable for small and middle deployments, and you shouldn't feel obligated to use this feature. However we understand that this tool wouldn't be useful if you couldn't customize it when you are ready for it.
+
+## Learn More
+
+You can learn more in the [Create React App documentation](https://facebook.github.io/create-react-app/docs/getting-started).
+
+To learn React, check out the [React documentation](https://reactjs.org/).
+
+### Code Splitting
+
+This section has moved here: [https://facebook.github.io/create-react-app/docs/code-splitting](https://facebook.github.io/create-react-app/docs/code-splitting)
+
+### Analyzing the Bundle Size
+
+This section has moved here: [https://facebook.github.io/create-react-app/docs/analyzing-the-bundle-size](https://facebook.github.io/create-react-app/docs/analyzing-the-bundle-size)
+
+### Making a Progressive Web App
+
+This section has moved here: [https://facebook.github.io/create-react-app/docs/making-a-progressive-web-app](https://facebook.github.io/create-react-app/docs/making-a-progressive-web-app)
+
+### Advanced Configuration
+
+This section has moved here: [https://facebook.github.io/create-react-app/docs/advanced-configuration](https://facebook.github.io/create-react-app/docs/advanced-configuration)
+
+### Deployment
+
+This section has moved here: [https://facebook.github.io/create-react-app/docs/deployment](https://facebook.github.io/create-react-app/docs/deployment)
+
+### `npm run build` fails to minify
+
+This section has moved here: [https://facebook.github.io/create-react-app/docs/troubleshooting#npm-run-build-fails-to-minify](https://facebook.github.io/create-react-app/docs/troubleshooting#npm-run-build-fails-to-minify)