Spaces:

dmartincy
/

document-translation

Runtime error

App Files Files Community

dmartincy commited on Dec 23, 2024

Commit

9f97180

1 Parent(s): 481612a

Add basic Dockerfile

Browse files

Files changed (28) hide show

.eslintignore +1 -0
.eslintrc.cjs +18 -0
.gitignore +27 -0
.tool-versions +1 -0
Dockerfile +98 -0
Modelfile +2 -0
Modelfile.embeddings +1 -0
index.html +14 -0
javascript.html +40 -0
package.json +31 -0
public/Spanish.docx +0 -0
public/sample.docx +0 -0
public/sample.json +59 -0
public/script-tag.html +34 -0
public/vite.svg +1 -0
service-config.yml +20 -0
src/App.css +16 -0
src/App.tsx +48 -0
src/components/DocumentEditor.tsx +55 -0
src/index.css +18 -0
src/javascript-app.js +90 -0
src/javascript-shared-app.js +50 -0
src/main.tsx +10 -0
src/vite-env.d.ts +1 -0
start-services.sh +96 -0
tsconfig.json +25 -0
tsconfig.node.json +11 -0
vite.config.ts +72 -0

.eslintignore ADDED Viewed

	@@ -0,0 +1 @@


1	+ /public/lib/

.eslintrc.cjs ADDED Viewed

	@@ -0,0 +1,18 @@

+module.exports = {
+  root: true,
+  env: { browser: true, es2020: true },
+  extends: [
+    'eslint:recommended',
+    'plugin:@typescript-eslint/recommended',
+    'plugin:react-hooks/recommended',
+  ],
+  ignorePatterns: ['dist', '.eslintrc.cjs'],
+  parser: '@typescript-eslint/parser',
+  plugins: ['react-refresh'],
+  rules: {
+    'react-refresh/only-export-components': [
+      'warn',
+      { allowConstantExport: true },
+    ],
+  },
+}

.gitignore ADDED Viewed

	@@ -0,0 +1,27 @@

+# Logs
+logs
+*.log
+npm-debug.log*
+yarn-debug.log*
+yarn-error.log*
+pnpm-debug.log*
+lerna-debug.log*
+node_modules
+dist
+dist-ssr
+*.local
+/public/lib/
+package-lock.json
+# Editor directories and files
+.vscode/*
+!.vscode/extensions.json
+.idea
+.DS_Store
+*.suo
+*.ntvs*
+*.njsproj
+*.sln
+*.sw?

.tool-versions ADDED Viewed

	@@ -0,0 +1 @@


1	+ nodejs 20.9.0

Dockerfile ADDED Viewed

	@@ -0,0 +1,98 @@

+# First stage: Get files from ai-assistant
+FROM pspdfkit/ai-assistant:nightly as ai-assistant
+# Main stage
+FROM pgvector/pgvector:pg16
+# Install Node.js and other dependencies
+RUN apt-get update && apt-get install -y \
+    curl \
+    wget \
+    nginx \
+    python3 \
+    python3-pip \
+    git \
+    unzip \
+    && curl -fsSL https://deb.nodesource.com/setup_20.x | bash - \
+    && apt-get install -y nodejs \
+    && rm -rf /var/lib/apt/lists/*
+WORKDIR /app
+# Create necessary directories and set up users
+RUN mkdir -p /base /var/log/llamafiler /var/lib/nginx /var/log/nginx /run/nginx /var/run/postgresql && \
+    useradd -m -u 1000 -s /bin/bash appuser && \
+    chown -R appuser:appuser /base /var/log/llamafiler /app && \
+    chown -R www-data:www-data /var/lib/nginx /var/log/nginx /run/nginx && \
+    mkdir -p /var/lib/postgresql/data && \
+    chmod 700 /var/lib/postgresql/data && \
+    chown -R appuser:appuser /var/lib/postgresql/data && \
+    chown -R appuser:appuser /var/run/postgresql && \
+    chmod 777 /var/run/postgresql
+# Copy entire ai-assistant to a separate directory
+COPY --from=ai-assistant /base/app /ai-assistant
+RUN chown -R appuser:appuser /ai-assistant
+# Copy our configuration files
+COPY --chown=appuser:appuser service-config.yml /app/
+COPY --chown=appuser:appuser start-services.sh /app/
+COPY --chown=appuser:appuser package*.json /app/
+# Install node dependencies for webapp
+RUN npm install
+# Copy webapp files
+COPY --chown=appuser:appuser . /app/
+# Configure nginx to route traffic
+RUN echo '\
+events { worker_connections 1024; } \n\
+http { \n\
+    server { \n\
+        listen 7860; \n\
+        location /v1 { \n\
+            proxy_pass http://localhost:8080; \n\
+        } \n\
+        location /api { \n\
+            proxy_pass http://localhost:4000; \n\
+        } \n\
+        location / { \n\
+            proxy_pass http://localhost:3000; \n\
+        } \n\
+    } \n\
+}' > /etc/nginx/nginx.conf
+# Set up environment variables
+ENV OPENAI_API_KEY="dummy-key" \
+    PGUSER=appuser \
+    PGPASSWORD=password \
+    PGDATABASE=postgres \
+    PGHOST=localhost \
+    PGPORT=5432 \
+    API_AUTH_TOKEN=secret \
+    DASHBOARD_USERNAME=dashboard \
+    DASHBOARD_PASSWORD=secret \
+    SECRET_KEY_BASE=secret-key-base \
+    NODE_OPTIONS="--experimental-vm-modules --experimental-json-modules" \
+    PATH="/usr/lib/postgresql/16/bin:${PATH}" \
+    POSTGRES_USER=appuser \
+    POSTGRES_PASSWORD=password \
+    POSTGRES_DB=postgres \
+    PGDATA=/var/lib/postgresql/data
+# Create local bin directory
+RUN mkdir -p /.local/bin && \
+    chown -R appuser:appuser /.local
+# Make start script executable
+RUN chmod +x /app/start-services.sh
+# Switch to appuser
+USER appuser
+# Expose the Space port
+EXPOSE 7860
+# Start all services
+CMD ["/app/start-services.sh"]

Modelfile ADDED Viewed

	@@ -0,0 +1,2 @@


1	+ FROM /models/gemma-2-2b-it-Q4_K_M.gguf
2	+

Modelfile.embeddings ADDED Viewed

	@@ -0,0 +1 @@


1	+ FROM /models/mxbai-embed-large-v1-f16.gguf

index.html ADDED Viewed

	@@ -0,0 +1,14 @@

+<!doctype html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <link rel="icon" type="image/svg+xml" href="/vite.svg" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>Document Authoring SDK</title>
+  </head>
+  <body>
+    <ul>
+      <li><a href='./javascript.html'>JavaScript example</a></li>
+    </ul>
+  </body>
+</html>

javascript.html ADDED Viewed

	@@ -0,0 +1,40 @@

+<!doctype html>
+<html lang="en">
+	<head>
+		<meta charset="UTF-8" />
+		<meta name="viewport" content="width=device-width, initial-scale=1.0" />
+		<title>Document Authoring SDK</title>
+		<style>
+			.spinner {
+				display: inline-block;
+				width: 16px;
+				height: 16px;
+				border: 2px solid #f3f3f3;
+				border-top: 2px solid #3498db;
+				border-radius: 50%;
+				animation: spin 1s linear infinite;
+				margin-right: 8px;
+				vertical-align: middle;
+			}
+			@keyframes spin {
+				0% { transform: rotate(0deg); }
+				100% { transform: rotate(360deg); }
+			}
+			#translateButton:disabled {
+				cursor: not-allowed;
+				opacity: 0.7;
+			}
+		</style>
+	</head>
+	<body>
+		<!--
+            IMPORTANT: An editor target element needs to have its `position` set to a value other than the default or `static`!
+                       If unsure use `relative`.
+         -->
+        <button id="translateButton">Translate to English</button>
+		<div id="editor" style="border: 1px solid black; width: 1024px; height: 600px; position: relative"></div>
+		<script type="module" src="./src/javascript-app.js"></script>
+	</body>
+</html>

package.json ADDED Viewed

	@@ -0,0 +1,31 @@

+{
+  "name": "document-authoring-sdk-example",
+  "private": true,
+  "version": "0.0.0",
+  "type": "module",
+  "author": "PSPDFKit (https://pspdfkit.com)",
+  "homepage": "https://pspdfkit.com/pdf-sdk/web/document-authoring/",
+  "scripts": {
+    "dev": "vite",
+    "build": "vite build",
+    "lint": "eslint . --ext ts,tsx --report-unused-disable-directives --max-warnings 0"
+  },
+  "dependencies": {
+    "@pspdfkit/document-authoring": "1.0.26",
+    "@wllama/wllama": "^1.16.4",
+    "react": "^18.2.0",
+    "react-dom": "^18.2.0"
+  },
+  "devDependencies": {
+    "@types/react": "^18.2.66",
+    "@types/react-dom": "^18.2.22",
+    "@typescript-eslint/eslint-plugin": "^7.2.0",
+    "@typescript-eslint/parser": "^7.2.0",
+    "@vitejs/plugin-react": "^4.2.1",
+    "eslint": "^8.57.0",
+    "eslint-plugin-react-hooks": "^4.6.0",
+    "eslint-plugin-react-refresh": "^0.4.6",
+    "typescript": "^5.2.2",
+    "vite": "^5.2.0"
+  }
+}

public/Spanish.docx ADDED Viewed

Binary file (5.24 kB). View file

public/sample.docx ADDED Viewed

Binary file (16.4 kB). View file

public/sample.json ADDED Viewed

	@@ -0,0 +1,59 @@

+{
+	"type": "https://pspdfkit.com/document-authoring/persistence/container",
+	"container": {
+		"document": {
+			"body": {
+				"sections": [
+					{
+						"elements": [
+							{
+								"type": "p",
+								"pPr": {
+									"styleId": "Heading 1",
+									"alignment": "center"
+								},
+								"elements": [
+									{
+										"type": "r",
+										"text": "Heading"
+									}
+								]
+							},
+							{
+								"type": "p",
+								"elements": [
+									{
+										"type": "r",
+										"text": "Simple paragraphs consist of one `r` typed element containing all the text."
+									}
+								]
+							},
+							{
+								"type": "p",
+								"elements": [
+									{
+										"type": "r",
+										"text": "There is "
+									},
+									{
+										"type": "r",
+										"rPr": {
+											"bold": true,
+											"color": "#ff0000",
+											"underline": true
+										},
+										"text": "formatted"
+									},
+									{
+										"type": "r",
+										"text": " text in this paragraph."
+									}
+								]
+							}
+						]
+					}
+				]
+			}
+		}
+	}
+}

public/script-tag.html ADDED Viewed

	@@ -0,0 +1,34 @@

+<!doctype html>
+<html lang="en">
+	<head>
+		<meta charset="UTF-8" />
+		<meta name="viewport" content="width=device-width, initial-scale=1.0" />
+		<title>Document Authoring SDK</title>
+	</head>
+	<body>
+		<!--
+            IMPORTANT: An editor target element needs to have its `position` set to a value other than the default or `static`!
+                       If unsure use `relative`.
+         -->
+		<div id="editor" style="border: 1px solid black; width: 1024px; height: 600px; position: relative"></div>
+		<script src="https://document-authoring-cdn.pspdfkit.com/releases/document-authoring-1.0.26-umd.js"></script>
+		<script>
+			(async ()=>{
+				const docAuthSystem = await DocAuth.createDocAuthSystem({
+					// assets: { base: '<CUSTOM_DOCUMENT_AUTHORING_ASSETS_LOCATION>' },
+					// licenseKey: '<YOUR_LICENSE_KEY>',
+				})
+				const editor = await docAuthSystem.createEditor(document.getElementById('editor'),{
+					document: await docAuthSystem.createDocumentFromPlaintext('Hi there!'),
+				});
+				// Dev exports.
+				window.DocAuth = DocAuth;
+				window.docAuthSystem = docAuthSystem
+				window.editor = editor;
+			})()
+		</script>
+	</body>
+</html>

public/vite.svg ADDED Viewed

service-config.yml ADDED Viewed

	@@ -0,0 +1,20 @@

+version: '1'
+aiServices:
+  chat:
+    provider:
+      name: 'openai-compat'
+      baseUrl: http://localhost:8080/v1
+    model: 'gemma-2b'
+  textEmbeddings:
+    provider:
+      name: 'openai-compat'
+      baseUrl: http://localhost:8080/v1
+    model: 'all-MiniLM-L6-v2'
+  headless:
+    - provider:
+        name: 'openai-compat'
+        baseUrl: http://localhost:8080/v1
+      model:
+        name: 'gemma-2b'
+        id: 'gemma-2b'

src/App.css ADDED Viewed

	@@ -0,0 +1,16 @@

+.App {
+    display: flex;
+    flex-direction: column;
+    width: 100%;
+    height: 100vh;
+    overflow: hidden;
+}
+.Header {
+    border-bottom: solid 1px black;
+    button {
+        margin: 10px;
+        padding: 3px;
+    }
+}

src/App.tsx ADDED Viewed

	@@ -0,0 +1,48 @@

+import DocumentEditor from './components/DocumentEditor'
+import { DocAuthEditor, DocAuthSystem } from '@pspdfkit/document-authoring'
+import { useCallback, useState } from 'react'
+import './App.css'
+function App() {
+  const [editor,setEditor] = useState<DocAuthEditor|null>(null)
+  const clear = async () => {
+    if( editor ) {
+      const docAuthSystem = editor.docAuthSystem()
+      editor.setCurrentDocument(await docAuthSystem.createDocumentFromPlaintext(''))
+    }
+  }
+  const loadDocx = async () => {
+    if( editor ) {
+      const docAuthSystem = editor.docAuthSystem()
+      editor.setCurrentDocument(await docAuthSystem.importDOCX(fetch('./sample.docx')))
+    }
+  }
+  const loadJson = async () => {
+    if( editor ) {
+      const docAuthSystem = editor.docAuthSystem()
+      editor.setCurrentDocument(await docAuthSystem.loadDocument(fetch('./sample.json')))
+    }
+  }
+  const createInitialDocument = useCallback((docAuthSystem:DocAuthSystem) => {
+    return docAuthSystem.createDocumentFromPlaintext('Hi there!')
+  },[])
+  return (
+    <div className='App'>
+        <>
+          <div className='Header'>
+            <button onClick={clear}>Clear</button>
+            <button onClick={loadDocx}>Load example.docx</button>
+            <button onClick={loadJson}>Load example.json</button>
+          </div>
+          <DocumentEditor initialDocument={createInitialDocument} onEditor={setEditor} />
+        </>
+    </div>
+  )
+}
+export default App

src/components/DocumentEditor.tsx ADDED Viewed

	@@ -0,0 +1,55 @@

+import { useEffect, useRef } from 'react';
+import { DocAuthSystem, DocAuthDocument, DocAuthEditor, createDocAuthSystem } from '@pspdfkit/document-authoring';
+const DocumentEditor = (props:{
+  initialDocument?: (docAuthSystem:DocAuthSystem)=>Promise<DocAuthDocument>,
+  onEditor?: (editor:DocAuthEditor)=>void,
+}) => {
+  const {initialDocument,onEditor} = props
+  const containerRef = useRef<HTMLDivElement|null>(null)
+  useEffect(()=>{
+    const target = document.createElement('div') as HTMLDivElement
+    let removed = false
+    const docAuthSystemAndEditorPromise = createDocAuthSystem({
+      // assets: { base: '<CUSTOM_DOCUMENT_AUTHORING_ASSETS_LOCATION>' },
+      // licenseKey: '<YOUR_LICENSE_KEY>',
+    }).then(async (docAuthSystem)=>{
+      const doc = await (initialDocument ? initialDocument(docAuthSystem) : undefined)
+      const editor = await docAuthSystem.createEditor(target,{
+        document: doc,
+      })
+      if( !removed ) {
+        if( containerRef.current ) {
+          containerRef.current.append(target)
+        }
+        if( onEditor ) {
+          onEditor(editor)
+        }
+      }
+      return {
+        docAuthSystem,
+        editor,
+      }
+    })
+    return ()=>{
+      target.remove()
+      removed = true
+      docAuthSystemAndEditorPromise.then(({docAuthSystem,editor})=>{
+        editor.destroy()
+        docAuthSystem.destroy()
+      })
+    }
+  },[onEditor,initialDocument])
+  return <div ref={containerRef} style={{ flexGrow: '1', position: 'relative' }} />
+}
+export default DocumentEditor

src/index.css ADDED Viewed

	@@ -0,0 +1,18 @@

+:root {
+  font-family: Inter, system-ui, Avenir, Helvetica, Arial, sans-serif;
+  font-weight: 400;
+  font-synthesis: none;
+  text-rendering: optimizeLegibility;
+  -webkit-font-smoothing: antialiased;
+  -moz-osx-font-smoothing: grayscale;
+}
+body {
+  margin: 0;
+  padding: 0;
+}
+#root {
+  display: flex;
+}

src/javascript-app.js ADDED Viewed

	@@ -0,0 +1,90 @@

+import DocAuth from "@pspdfkit/document-authoring";
+const docAuthSystem = await DocAuth.createDocAuthSystem({
+  // assets: { base: '<CUSTOM_DOCUMENT_AUTHORING_ASSETS_LOCATION>' },
+  // licenseKey: '<YOUR_LICENSE_KEY>',
+});
+const editor = await docAuthSystem.createEditor(
+  document.getElementById("editor"),
+  {
+    document: await docAuthSystem.importDOCX(fetch("./Spanish.docx")),
+  }
+);
+// Translation function.
+async function translate(content, lang = "English") {
+  try {
+    const response = await fetch('/api/aia/v1/chat/completions', {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        'Authorization': 'Token token=eyJhbGciOiJSUzI1NiIsInR5cCI6IkpXVCJ9.eyJpYXQiOjE3MzMyNjQ2MDcsImV4cCI6MTczMzUyMzgwN30.WWyks6NOU43_2OcZX82z3P4QUwQiUA61SeTZsNobIGoVg3uDTkedXWqISIB7QqTnRyf_D92t_pGr5g_GzqBaOTqaXYJJPco9mtZjtAyUpdWr1UHu4jx-zoKUpCeguZQZbHG6vqYzu1BNV1iPdij2OIoUrr_DY--G-6m5-31E0LcDZb7ysUyZL1Ai_q7PPRwTS8mEtvuF4QO6Cw_a77X7Qh9x_V8a_WwNMFwQMRkIUAsRx3IOI10q0hdwP-k8xYtTbHQO-z66pRl6I71ATclR7FS5FraoLLmWlq8lqNKfzeOgER8Kcf8uB87zHQbfk0QR8AfJGmZDE8pG5F3QG29MeA'
+      },
+      body: JSON.stringify({
+        messages: [
+          {
+            role: "system",
+            content: "You are a translator. Only provide the translation, no explanations or additional text."
+          },
+          {
+            role: "user",
+            content: `Translate the following text to ${lang}: ${content}`
+          }
+        ],
+        model: "gemma-2b",
+        stream: false
+      })
+    });
+    if (!response.ok) {
+      throw new Error('Translation request failed');
+    }
+    const data = await response.json();
+    console.log('Translation API response:', data);
+    return data.choices[0].message.content;
+  } catch (error) {
+    console.error("Translation error:", error);
+    throw new Error("Failed to generate translation");
+  }
+}
+// Function to recursively translate text in the JSON structure.
+async function translateRecursive(obj) {
+  for (let key in obj) {
+    if (typeof obj[key] === "string" && key === "text" && obj[key].length > 0) {
+      obj[key] = await translate(obj[key]);
+    } else if (typeof obj[key] === "object") {
+      await translateRecursive(obj[key]);
+    }
+  }
+}
+// Function to translate the document.
+async function translateDocument() {
+  try {
+    translateButton.disabled = true;
+    translateButton.innerHTML = '<span class="spinner"></span> Translating...';
+    const jsonDoc = await editor.currentDocument().saveDocument();
+    await translateRecursive(jsonDoc.container.document);
+    const translatedDoc = await docAuthSystem.loadDocument(jsonDoc);
+    editor.setCurrentDocument(translatedDoc);
+  } catch (error) {
+    console.error("Translation error:", error);
+  } finally {
+    translateButton.disabled = false;
+    translateButton.innerHTML = 'Translate to English';
+  }
+}
+// Add click event listener to the translate button.
+const translateButton = document.getElementById("translateButton");
+translateButton.addEventListener("click", translateDocument);
+// Dev exports.
+window.DocAuth = DocAuth;
+window.docAuthSystem = docAuthSystem;
+window.editor = editor;

src/javascript-shared-app.js ADDED Viewed

	@@ -0,0 +1,50 @@

+// Load the JS module.
+import DocAuth from '@pspdfkit/document-authoring'
+// Load the base Document Authoring system. The WASM and font code is initialized here.
+// The `docAuthSystem` can be re-used between multiple editor instances or used to
+// export PDFs or import DOCX documents without the need to instantiate the visual
+// editor.
+const docAuthSystem = await DocAuth.createDocAuthSystem({
+    // assets: { base: '<CUSTOM_DOCUMENT_AUTHORING_ASSETS_LOCATION>' },
+    // licenseKey: '<YOUR_LICENSE_KEY>',
+});
+// A PDF can be created without the need for an editor instance.
+document.getElementById('headless').onclick = async () => {
+    const createdDocument = await docAuthSystem.createDocumentFromPlaintext(`Hello World!\n\nISO date: ${new Date().toISOString()}`);
+    const pdfBlob = await createdDocument.exportPDF();
+    window.open(URL.createObjectURL(new Blob([pdfBlob], { type: 'application/pdf' })));
+};
+// Instantiate a visual document authoring editor.
+const editor1 = await docAuthSystem.createEditor(
+    document.getElementById('editor1'),{
+    document: await docAuthSystem.createDocumentFromPlaintext('Hi there!\n\nHow are you?'),
+});
+// Instantiate another visual document authoring editor.
+const editor2 = await docAuthSystem.createEditor(
+    document.getElementById('editor2'),
+    {
+    document: await docAuthSystem.loadDocument(await fetch('./sample.json').then((e) => e.json())),
+});
+document.getElementById('docx').onclick = async () => {
+    const doc = await docAuthSystem.importDOCX(fetch('./sample.docx'));
+    editor1.setCurrentDocument(doc)
+};
+// We can copy the document from `editor1` to `editor2` by exporting and importing it as either
+// a JSON string (`exportDocumentJSON`) or an equivalent object (`exportDocument`).
+document.getElementById('copy1to2').onclick = async () => {
+    // "Clone" the document by saving and loading it again.
+    editor2.setCurrentDocument(await docAuthSystem.loadDocument(editor1.currentDocument().saveDocument()))
+}
+// Dev exports.
+window.DocAuth = DocAuth;
+window.docAuthSystem = docAuthSystem;
+window.editor1 = editor1;
+window.editor2 = editor2;

src/main.tsx ADDED Viewed

	@@ -0,0 +1,10 @@

+import React from 'react'
+import ReactDOM from 'react-dom/client'
+import App from './App.tsx'
+import './index.css'
+ReactDOM.createRoot(document.getElementById('root')!).render(
+  <React.StrictMode>
+    <App />
+  </React.StrictMode>,
+)

src/vite-env.d.ts ADDED Viewed

	@@ -0,0 +1 @@


1	+ /// <reference types="vite/client" />

start-services.sh ADDED Viewed

	@@ -0,0 +1,96 @@

+#!/bin/sh
+set -e  # Exit on error
+# Create download directory
+mkdir -p /tmp/downloads
+cd /tmp/downloads
+# Download models
+echo "Downloading models..."
+wget -q https://huggingface.co/bartowski/gemma-2-2b-it-GGUF/resolve/main/gemma-2-2b-it-Q4_K_M.gguf -O /base/gemma-2b.gguf
+wget -q https://huggingface.co/leliuga/all-MiniLM-L6-v2-GGUF/resolve/main/all-MiniLM-L6-v2.F16.gguf -O /base/embeddings.gguf
+# Download and setup llamafiler
+echo "Setting up llamafiler..."
+wget --no-check-certificate https://github.com/Mozilla-Ocho/llamafile/releases/download/0.8.17/llamafile-0.8.17.zip
+unzip llamafile-0.8.17.zip
+mkdir -p /.local/bin
+cp llamafile-0.8.17/bin/llamafiler /.local/bin/
+chmod +x /.local/bin/llamafiler
+# Clean up downloads
+cd /app
+rm -rf /tmp/downloads
+# Initialize PostgreSQL if needed
+echo "Initializing PostgreSQL..."
+if [ ! -f "$PGDATA/PG_VERSION" ]; then
+    initdb -D "$PGDATA" --auth=trust
+    # Configure PostgreSQL
+    echo "Configuring PostgreSQL..."
+    cat >> "$PGDATA/postgresql.conf" << EOF
+shared_preload_libraries = 'pgvector'
+listen_addresses = '*'
+port = 5432
+unix_socket_directories = '/var/run/postgresql'
+EOF
+    # Configure client authentication
+    echo "host all all all trust" >> "$PGDATA/pg_hba.conf"
+fi
+# Start PostgreSQL
+echo "Starting PostgreSQL..."
+pg_ctl -D "$PGDATA" -l "$PGDATA/logfile" start
+# If PostgreSQL fails to start, show the logs
+if [ $? -ne 0 ]; then
+    echo "PostgreSQL failed to start. Here are the logs:"
+    cat "$PGDATA/logfile"
+    echo "Contents of postgresql.conf:"
+    cat "$PGDATA/postgresql.conf"
+    echo "Contents of pg_hba.conf:"
+    cat "$PGDATA/pg_hba.conf"
+    exit 1
+fi
+# Wait for PostgreSQL to be ready
+until pg_isready -h localhost; do
+    echo "Waiting for PostgreSQL to start..."
+    echo "PostgreSQL log output:"
+    cat "$PGDATA/logfile"
+    sleep 2
+done
+echo "PostgreSQL is ready"
+# Create extension if not exists
+psql -h localhost -d postgres -c 'CREATE EXTENSION IF NOT EXISTS vector;'
+# Start nginx
+echo "Starting nginx..."
+nginx
+# Start llamafiler instances
+echo "Starting llamafiler services..."
+/.local/bin/llamafiler --model /base/gemma-2b.gguf --listen 0.0.0.0:8082 &
+GEMMA_PID=$!
+/.local/bin/llamafiler --model /base/embeddings.gguf --listen 0.0.0.0:8081 &
+EMBEDDINGS_PID=$!
+# Start the AI assistant
+echo "Starting AI assistant..."
+cd /ai-assistant && ./bin/entrypoint.sh node app/main.bundle.js &
+AI_ASSISTANT_PID=$!
+# Start the webapp
+echo "Starting webapp..."
+cd /app && node --experimental-modules ./node_modules/vite/bin/vite.js serve --host 0.0.0.0 --port 3000 &
+WEBAPP_PID=$!
+# Wait for any process to exit
+wait -n
+# Exit with status of process that exited first
+exit $?

tsconfig.json ADDED Viewed

	@@ -0,0 +1,25 @@

+{
+  "compilerOptions": {
+    "target": "ES2020",
+    "useDefineForClassFields": true,
+    "lib": ["ES2020", "DOM", "DOM.Iterable"],
+    "module": "ESNext",
+    "skipLibCheck": true,
+    /* Bundler mode */
+    "moduleResolution": "bundler",
+    "allowImportingTsExtensions": true,
+    "resolveJsonModule": true,
+    "isolatedModules": true,
+    "noEmit": true,
+    "jsx": "react-jsx",
+    /* Linting */
+    "strict": true,
+    "noUnusedLocals": true,
+    "noUnusedParameters": true,
+    "noFallthroughCasesInSwitch": true
+  },
+  "include": ["src"],
+  "references": [{ "path": "./tsconfig.node.json" }]
+}

tsconfig.node.json ADDED Viewed

	@@ -0,0 +1,11 @@

+{
+  "compilerOptions": {
+    "composite": true,
+    "skipLibCheck": true,
+    "module": "ESNext",
+    "moduleResolution": "bundler",
+    "allowSyntheticDefaultImports": true,
+    "strict": true
+  },
+  "include": ["vite.config.ts"]
+}

vite.config.ts ADDED Viewed

	@@ -0,0 +1,72 @@

+import { defineConfig } from 'vite'
+import react from '@vitejs/plugin-react'
+import { resolve } from 'path'
+// https://vitejs.dev/config/
+export default defineConfig({
+  plugins: [react()],
+  appType: 'mpa',
+  server: {
+    host: '0.0.0.0',  // Listen on all available network interfaces
+    port: 3000,
+    headers: {
+      'Cross-Origin-Embedder-Policy': 'require-corp',
+      'Cross-Origin-Opener-Policy': 'same-origin'
+    },
+    proxy: {
+      '/api/aia': {
+        target: 'http://aia:4000',
+        changeOrigin: true,
+        rewrite: (path) => {
+          console.log('Rewriting path:', path);
+          const rewritten = path.replace(/^\/api\/aia/, '/client/api');
+          console.log('Rewritten to:', rewritten);
+          return rewritten;
+        },
+        configure: (proxy, options) => {
+          console.log('Configuring proxy:', options);
+          proxy.on('error', (err, req, res) => {
+            console.log('Proxy error:', err);
+            console.log('Request:', req.method, req.url);
+          });
+          proxy.on('proxyReq', (proxyReq, req, res) => {
+            console.log('Proxying request:', {
+              method: req.method,
+              url: req.url,
+              targetUrl: proxyReq.path
+            });
+          });
+          proxy.on('proxyRes', (proxyRes, req, res) => {
+            console.log('Received response:', {
+              status: proxyRes.statusCode,
+              url: req.url,
+              headers: proxyRes.headers
+            });
+          });
+        },
+        ws: true,
+        secure: false,
+        debug: true
+      }
+    }
+  },
+  esbuild: {
+    supported: {
+      'top-level-await': true,
+    },
+  },
+  build: {
+    rollupOptions: {
+      input: {
+        'index.html': resolve(__dirname,'index.html'),
+        'react.html': resolve(__dirname,'react.html'),
+        'javascript.html': resolve(__dirname,'javascript.html'),
+        'javascript-shared.html': resolve(__dirname,'javascript-shared.html'),
+      }
+    },
+    chunkSizeWarningLimit: 2048,
+  }
+})