Something went wrong. Try again.
Monorepo for Aesthetic.Computer aesthetic.computer
Something went wrong. Try again.
JavaScript
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357import http from 'http';import { URL } from 'url';import { WebSocketServer } from 'ws';import { MongoClient } from 'mongodb';import dotenv from 'dotenv';import { fileURLToPath } from 'url';import { dirname, join } from 'path';import { readFile } from 'fs/promises';
// Load environment variables from vaultconst __filename = fileURLToPath(import.meta.url);const __dirname = dirname(__filename);// Try local .env first (for production), then vault (for development)const localEnvPath = join(__dirname, '.env');const vaultEnvPath = join(__dirname, '../aesthetic-computer-vault/judge/.env');dotenv.config({ path: localEnvPath });dotenv.config({ path: vaultEnvPath }); // Will only load if local doesn't exist
const OLLAMA_API = 'http://localhost:11434/api/generate';const MODEL = 'gemma2:2b';const PORT = 3000;
// MongoDB connectionconst MONGODB_CONNECTION_STRING = process.env.MONGODB_CONNECTION_STRING || 'mongodb://localhost:27017';const MONGODB_NAME = process.env.MONGODB_NAME || 'aesthetic';const COLLECTION_NAME = process.env.COLLECTION_NAME || 'chat-system';let mongoClient = null;let chatCollection = null;
async function connectMongo() { if (!mongoClient) { mongoClient = new MongoClient(MONGODB_CONNECTION_STRING, { serverSelectionTimeoutMS: 2000, // Timeout after 2 seconds connectTimeoutMS: 2000 }); await mongoClient.connect(); chatCollection = mongoClient.db(MONGODB_NAME).collection(COLLECTION_NAME); console.log('📦 Connected to MongoDB'); } return chatCollection;}
// PG-13 content filter promptconst systemPrompt = `You are a PG-13 content filter for a chat room. Respond in s-expression format.
CRITICAL: ALWAYS include the (why "...") part in your response!
ALWAYS reply with BOTH parts:(yes) (why "short lowercase explanation")OR(no) (why "short lowercase explanation")
Reply (yes) for URLs (http://, https://, www., or domain.tld patterns).
Reply (no) if the message contains:- Sexual content or innuendo- Body functions (bathroom humor, gross-out content)- Profanity or explicit language- Violence or threats- Drug references- Hate speech or slurs
Reply (yes) for:- Normal conversation- URLs and links- Questions and discussions- Greetings and casual chat
Example responses (ALWAYS include both parts):(yes) (why "normal greeting")(yes) (why "asking a question")(yes) (why "url link")(no) (why "contains profanity")(no) (why "sexual content")(no) (why "inappropriate language")
Keep all explanations lowercase and brief. NEVER skip the (why "...") part!`;
async function filterMessage(message, onChunk = null) { const prompt = `${systemPrompt}\n\nMessage: "${message}"\n\nReply (t/f):`; const response = await fetch(OLLAMA_API, { method: 'POST', headers: { 'Content-Type': 'application/json' }, body: JSON.stringify({ model: MODEL, prompt: prompt, stream: true, // Enable streaming }), });
if (!response.ok) { throw new Error(`Ollama API error: ${response.status}`); }
let fullResponse = ''; let totalDuration = 0; // Stream the response const reader = response.body.getReader(); const decoder = new TextDecoder(); while (true) { const { done, value } = await reader.read(); if (done) break; const chunk = decoder.decode(value); const lines = chunk.split('\n').filter(line => line.trim()); for (const line of lines) { try { const data = JSON.parse(line); if (data.response) { fullResponse += data.response; if (onChunk) { onChunk({ chunk: data.response, full: fullResponse }); } } if (data.total_duration) { totalDuration = data.total_duration; } } catch (e) { // Skip invalid JSON lines } } } const finalResponse = fullResponse.trim(); // Parse s-expression: (yes) or (no) and extract why const decision = finalResponse.toLowerCase().includes('(yes)') ? 't' : 'f'; // Extract (why "...") - proper s-expression format let reason = ''; const reasonMatch = finalResponse.match(/\(why\s+"((?:[^"\\]|\\.)*)"\)/is); if (reasonMatch) { reason = reasonMatch[1].toLowerCase().trim(); // Force lowercase and trim } return { decision: decision, sentiment: finalResponse, reason: reason, responseTime: totalDuration / 1e9, // Convert to seconds };}
const server = http.createServer(async (req, res) => { // CORS headers res.setHeader('Access-Control-Allow-Origin', '*'); res.setHeader('Access-Control-Allow-Methods', 'POST, OPTIONS'); res.setHeader('Access-Control-Allow-Headers', 'Content-Type');
if (req.method === 'OPTIONS') { res.writeHead(200); res.end(); return; }
if (req.method === 'GET' && (req.url === '/' || req.url === '/index.html')) { // Serve the dashboard HTML try { const html = await readFile(join(__dirname, 'index.html'), 'utf-8'); res.writeHead(200, { 'Content-Type': 'text/html' }); res.end(html); } catch (error) { res.writeHead(500, { 'Content-Type': 'application/json' }); res.end(JSON.stringify({ error: 'Failed to load dashboard' })); } } else if (req.method === 'GET' && req.url.startsWith('/api/chat-messages')) { // Fetch random sample of chat messages for auto-testing try { console.log('📡 Fetching chat messages from MongoDB...'); console.log(' Connection string:', MONGODB_CONNECTION_STRING ? 'Set ✓' : 'Not set ✗'); console.log(' Database:', MONGODB_NAME); console.log(' Collection:', COLLECTION_NAME); const collection = await connectMongo(); const messages = await collection .aggregate([ { $match: { text: { $exists: true, $ne: '' } } }, { $sample: { size: 100 } }, { $project: { text: 1, _id: 0 } } ]) .toArray(); console.log(` Found ${messages.length} messages`); res.writeHead(200, { 'Content-Type': 'application/json' }); res.end(JSON.stringify(messages.map(m => m.text))); } catch (error) { console.error('MongoDB error:', error); // Fallback to test messages if MongoDB unavailable const testMessages = [ "hello world", "how are you doing today?", "fuck you", "shit this sucks", "nice painting!", "what time is it?", "https://example.com/image.png", "i hate you so much", "beautiful colors", "damn that's cool", "check out my site www.example.com", "you're an idiot", "love this piece", "kill yourself", "great work!", "this is garbage", "can you help me?", "stupid ass program", "amazing art", "go to hell" ]; res.writeHead(200, { 'Content-Type': 'application/json' }); res.end(JSON.stringify(testMessages)); } } else if (req.method === 'POST' && req.url === '/api/filter') { let body = ''; req.on('data', chunk => { body += chunk.toString(); });
req.on('end', async () => { try { const { message } = JSON.parse(body); if (!message || typeof message !== 'string') { res.writeHead(400, { 'Content-Type': 'application/json' }); res.end(JSON.stringify({ error: 'Invalid message' })); return; }
console.log(`[${new Date().toISOString()}] Testing: "${message.substring(0, 50)}${message.length > 50 ? '...' : ''}"`); const result = await filterMessage(message); console.log(` → Decision: ${result.decision === 't' ? 'PASS' : 'FAIL'} (${(result.responseTime * 1000).toFixed(0)}ms)`);
res.writeHead(200, { 'Content-Type': 'application/json' }); res.end(JSON.stringify(result)); } catch (error) { console.error('Error:', error); res.writeHead(500, { 'Content-Type': 'application/json' }); res.end(JSON.stringify({ error: error.message })); } }); } else { res.writeHead(404, { 'Content-Type': 'application/json' }); res.end(JSON.stringify({ error: 'Not found' })); }});
// Track warmup statelet isWarmedUp = false;let warmupPromise = null;
async function warmupModel() { if (isWarmedUp) return; if (warmupPromise) return warmupPromise; warmupPromise = (async () => { console.log(`⏳ Warming up model...`); try { await filterMessage('hello'); isWarmedUp = true; console.log(`✅ Model warmed up and ready!\n`); } catch (error) { console.log(`⚠️ Model warmup failed: ${error.message}`); console.log(` First request may be slower.\n`); } })(); return warmupPromise;}
server.listen(PORT, async () => { console.log(`🛡️ Content Filter API running on http://localhost:${PORT}`); console.log(`📊 Dashboard available via Caddy on http://localhost:8080`); console.log(`🤖 Using model: ${MODEL}\n`); // Start warmup (don't block server startup) warmupModel().catch(err => console.error('❌ Warmup error:', err));});
// Handle uncaught errorsprocess.on('uncaughtException', (err) => { console.error('❌ Uncaught exception:', err);});
process.on('unhandledRejection', (err) => { console.error('❌ Unhandled rejection:', err);});
// WebSocket server for live streamingconst wss = new WebSocketServer({ server, path: '/ws' });
wss.on('connection', (ws) => { console.log('📡 WebSocket client connected'); ws.on('message', async (data) => { try { const { message } = JSON.parse(data); if (!message || typeof message !== 'string') { ws.send(JSON.stringify({ error: 'Invalid message' })); return; } // Wait for warmup to complete before processing if (!isWarmedUp) { ws.send(JSON.stringify({ type: 'warming', message: 'Model warming up...' })); await warmupPromise; } console.log(`[${new Date().toISOString()}] Streaming test: "${message.substring(0, 50)}${message.length > 50 ? '...' : ''}"`); const startTime = Date.now(); // Send start event ws.send(JSON.stringify({ type: 'start' })); // Filter with streaming const result = await filterMessage(message, (update) => { ws.send(JSON.stringify({ type: 'chunk', chunk: update.chunk, full: update.full })); }); const responseTime = Date.now() - startTime; console.log(` → Decision: ${result.decision === 't' ? 'PASS' : 'FAIL'} (${responseTime}ms)`); // Send final result ws.send(JSON.stringify({ type: 'complete', decision: result.decision, sentiment: result.sentiment, reason: result.reason, responseTime: responseTime })); } catch (error) { console.error('WebSocket error:', error); ws.send(JSON.stringify({ type: 'error', error: error.message })); } }); ws.on('close', () => { console.log('📡 WebSocket client disconnected'); });});