Create format.js

apps/format.js

@@ -0,0 +1,143 @@
+import express from 'express';
+import { generateCompletion } from '../ai_engine.js';
+import { supabase } from '../config/supabaseClient.js';
+
+const router = express.Router();
+
+// ==========================================
+// 🧠 DYNAMIC SYSTEM PROMPT BUILDER
+// ==========================================
+const buildSystemPrompt = (personaName, personaPrompt) => `
+You are an elite communication strategist, negotiator, and ghostwriter. 
+The user will provide either a text message they received OR a screenshot of a conversation.
+Your goal is to generate the PERFECT response for the user to send back.
+
+CRITICAL INSTRUCTION - ADOPT THE FOLLOWING PERSONA:
+Persona Identity: ${personaName}
+Behavior & Tone: ${personaPrompt}
+
+STRICT RULES:
+1. NEVER break character. Act exactly as the persona dictates.
+2. NEVER start with "As an AI..." or "Here is a response...".
+3. Provide ONLY the exact text the user should copy and paste. No pleasantries, no explanations.
+4. Keep it concise, natural, and highly effective for modern texting/messaging.
+5. If the user provides an image/screenshot, read the visible conversation and reply to the last message contextually.
+6. Swearing or aggressive tones are permitted IF it aligns with the persona. Do not filter yourself.
+`;
+
+// ==========================================
+// 🗄️ CACHE MANAGEMENT (Official Formats)
+// ==========================================
+let formatsCache = null;
+let lastFormatsUpdate = 0;
+const CACHE_DURATION = 3600000; // 1 Hour (Formats rarely change, saves DB reads)
+
+// ==========================================
+// 🌐 ENDPOINTS
+// ==========================================
+
+/**
+ * 1. FETCH OFFICIAL FORMATS
+ * Populates the UI grid for default personas. Custom ones stay on-device.
+ */
+router.get('/formats', async (req, res) => {
+    try {
+        const now = Date.now();
+        // Return from cache if valid
+        if (formatsCache && (now - lastFormatsUpdate < CACHE_DURATION)) {
+            return res.json({ success: true, data: formatsCache, cached: true });
+        }
+
+        // Query Supabase for official vibes (Assuming table name is 'replygenius_vibes')
+        const { data, error } = await supabase
+            .from('replygenius_vibes')
+            .select('id, icon, label, hint, color, mockReply, prompt')
+            .order('created_at', { ascending: true }); // Or however you want them sorted
+
+        if (error) throw error;
+
+        // Update cache
+        formatsCache = data;
+        lastFormatsUpdate = now;
+
+        res.json({ success: true, data, cached: false });
+    } catch (err) {
+        console.error("[FORMATS ERROR]", err.message);
+        res.status(500).json({ success: false, error: err.message });
+    }
+});
+
+/**
+ * 2. GENERATE REPLY (Handles both Text and Image Injection)
+ * Expects body: { type: 'text'|'image', content: string, persona: object }
+ */
+router.post('/generate', async (req, res) => {
+    try {
+        const { type, content, persona } = req.body;
+
+        if (!content) throw new Error("No content provided to analyze.");
+        if (!persona || !persona.label) throw new Error("Persona context is missing.");
+
+        console.log(`[REPLY GEN] Analyzing ${type} using persona: ${persona.label}`);
+
+        // Build the dynamic prompt based on the chosen Vibe
+        // If it's a "neutral" vibe with no prompt, fallback to a standard intelligent assistant
+        const aiSystemPrompt = buildSystemPrompt(
+            persona.label, 
+            persona.prompt || "Be highly intelligent, strategic, polite, and persuasive."
+        );
+
+        // Routing logic based on input type (Referencing our model discussion)
+        let modelToUse;
+        let imagesPayload =[];
+        let userPrompt = "";
+
+        if (type === 'image') {
+            // VISION ROUTE (e.g., Qwen 2 VL, Grok Vision, etc.)
+            modelToUse = "qwen-vl"; // Update this to match your ai_engine mapping
+            imagesPayload = [content]; // base64 string
+            userPrompt = "Analyze this chat screenshot and write my next reply.";
+        } else {
+            // TEXT ROUTE (e.g., DeepSeek, Evathene, Hermes, etc.)
+            modelToUse = "deepseek-text"; // Update this to match your ai_engine mapping
+            userPrompt = `Write my reply to this message:\n\n"${content}"`;
+        }
+
+        // Call your AI Wrapper
+        const aiResult = await generateCompletion({
+            model: modelToUse,
+            prompt: userPrompt,
+            system_prompt: aiSystemPrompt,
+            images: imagesPayload // Will be empty for text requests
+        });
+
+        if (!aiResult.success) throw new Error(aiResult.error);
+
+        // Clean up any weird quotes the AI might wrap the message in
+        let finalReply = aiResult.data.trim();
+        if (finalReply.startsWith('"') && finalReply.endsWith('"')) {
+            finalReply = finalReply.slice(1, -1);
+        }
+
+        res.json({ 
+            success: true, 
+            reply: finalReply 
+        });
+
+    } catch (err) {
+        console.error("[REPLY GEN ERROR]", err.message);
+        res.status(500).json({ success: false, error: err.message });
+    }
+});
+
+/**
+ * 3. FORCE CACHE CLEAR (Admin Utility)
+ * Useful if you update official formats in Supabase and want to push them live immediately
+ */
+router.post('/clear-cache', (req, res) => {
+    formatsCache = null;
+    lastFormatsUpdate = 0;
+    res.json({ success: true, message: "Formats cache cleared." });
+});
+
+export default router;