Upload folder using huggingface_hub

server.js

@@ -1,115 +1,120 @@
+console.log("--> SERVER STARTING...");
 import express from 'express';
-import { SystemMessage, HumanMessage, AIMessage, LlamaCppLLM } from './src/index.js';
+// import { SystemMessage, HumanMessage, AIMessage, LlamaCppLLM } from './src/index.js'; // CÓ THỂ LỖI Ở ĐÂY
+// Để chắc chắn, tôi sẽ import trực tiếp từ file thực tế nếu cần, 
+// nhưng trước hết hãy thử wrap import để bắt lỗi
+let SystemMessage, HumanMessage, AIMessage, LlamaCppLLM;
+
+try {
+    const module = await import('./src/index.js');
+    SystemMessage = module.SystemMessage;
+    HumanMessage = module.HumanMessage;
+    AIMessage = module.AIMessage;
+    LlamaCppLLM = module.LlamaCppLLM;
+    console.log("--> Modules loaded successfully");
+} catch (e) {
+    console.error("--> FATAL IMPORT ERROR:", e);
+    // Vẫn giữ server chạy để xem log
+}
+
 import bodyParser from 'body-parser';
 import fs from 'fs';
 import path from 'path';
 
 // --- CONFIGURATION ---
 const PORT = 7860;
-const LOG_FILE = 'activity.log';
 const MODEL_PATH = './models/Qwen3-1.7B-Q8_0.gguf';
 
-// --- LOGGING UTILS ---
-function log(type, content) {
-    const timestamp = new Date().toISOString();
-    const entry = `[${timestamp}] [${type}] ${JSON.stringify(content)}
-`;
-    console.log(entry.trim());
-    fs.appendFileSync(LOG_FILE, entry);
-}
+// ... (Các phần code khác giữ nguyên nhưng thêm check null) ...
+
+// --- LOAD PROMPTS ---
+const PROMPTS = {
+    "default": "You are a helpful AI assistant. Answer concisely.",
+    "talk2people": fs.existsSync('MssterPrompt_Talk2People.txt') 
+        ? fs.readFileSync('MssterPrompt_Talk2People.txt', 'utf-8') 
+        : "Default Talk2People prompt (File missing)",
+    "coder": "You are an expert programmer."
+};
+
+console.log("--> Prompts loaded. Talk2People length:", PROMPTS['talk2people'].length);
+
+const WELCOME_MESSAGE = "Xin chào đây là hệ thống bot AI hoạt động riêng tư được đào tạo bởi Dạ Hành Studio , Ko phụ thuộc vào Google";
 
 // --- AI ENGINE ---
 let llm = null;
-let defaultSystemPrompt = "You are a helpful and intelligent AI assistant. You answer concisely and accurately.";
 
 async function initModel() {
+    if (!LlamaCppLLM) {
+        console.error("--> Cannot init model because LlamaCppLLM class is missing.");
+        return;
+    }
     try {
-        log('SYSTEM', 'Loading model...');
+        console.log('--> Loading model...');
         if (!fs.existsSync(MODEL_PATH)) {
-            log('ERROR', `Model not found at ${MODEL_PATH}`);
+            console.error(`--> Model not found at ${MODEL_PATH}`);
+            // Check if models folder exists
+            if (fs.existsSync('./models')) {
+                console.log("--> Models dir content:", fs.readdirSync('./models'));
+            } else {
+                console.log("--> Models dir MISSING!");
+            }
             return;
         }
 
         llm = new LlamaCppLLM({
             modelPath: MODEL_PATH,
-            temperature: 0.7, // Higher for creativity
-            maxTokens: 1024   // Longer responses
+            temperature: 0.7, 
+            maxTokens: 1024
         });
         
-        // Warmup
         await llm.invoke("Hello");
-        log('SYSTEM', 'Model loaded successfully. Ready to serve.');
+        console.log('--> Model loaded successfully.');
     } catch (err) {
-        log('FATAL', err.message);
+        console.error('--> FATAL MODEL ERROR:', err.message);
     }
 }
 
 // --- SERVER SETUP ---
 const app = express();
 app.use(bodyParser.json());
+app.use(express.static('public'));
 
-// 1. CHAT ENDPOINT (Main Interaction)
 app.post('/chat', async (req, res) => {
-    if (!llm) return res.status(503).json({ error: "System initializing..." });
+    if (!llm) return res.status(503).json({ reply: "Hệ thống đang khởi động hoặc lỗi model. Vui lòng chờ..." });
 
     try {
-        const { message, history = [], system_instruction } = req.body;
+        const { message, history = [], role = 'default' } = req.body;
+        const systemInstruction = PROMPTS[role] || PROMPTS['default'];
         
-        // Construct Message Chain
         const messages = [];
-        
-        // 1. System Prompt (Training/Instruction)
-        messages.push(new SystemMessage(system_instruction || defaultSystemPrompt));
+        messages.push(new SystemMessage(systemInstruction));
 
-        // 2. Chat History (Short-term memory)
         history.forEach(msg => {
             if (msg.role === 'user') messages.push(new HumanMessage(msg.content));
             if (msg.role === 'ai') messages.push(new AIMessage(msg.content));
         });
 
-        // 3. Current Message
         messages.push(new HumanMessage(message));
+        console.log(`[${role}] User: ${message}`);
 
-        log('REQUEST', { message, instruction: system_instruction || "default" });
-
-        // Invoke AI
         const response = await llm.invoke(messages);
-        
-        log('RESPONSE', response.content);
+        console.log(`[${role}] AI: ${response.content}`);
         res.json({ reply: response.content });
 
     } catch (error) {
-        log('ERROR', error.message);
+        console.error(error);
         res.status(500).json({ error: error.message });
     }
 });
 
-// 2. CONFIG ENDPOINT (Set default behavior)
-app.post('/config', (req, res) => {
-    const { system_prompt, temperature } = req.body;
-    if (system_prompt) {
-        defaultSystemPrompt = system_prompt;
-        log('CONFIG', `Updated default system prompt to: ${system_prompt.substring(0, 50)}...`);
-    }
-    // Note: Temperature change requires LLM reload in current simple wrapper, skipping for now
-    res.json({ success: true, current_system_prompt: defaultSystemPrompt });
-});
-
-// 3. LOGS ENDPOINT (Realtime Monitoring)
-app.get('/logs', (req, res) => {
-    if (fs.existsSync(LOG_FILE)) {
-        const logs = fs.readFileSync(LOG_FILE, 'utf-8');
-        // Return last 100 lines
-        const lines = logs.trim().split('\n').slice(-100).join('\n');
-        res.send(`<pre>${lines}</pre>`);
-    } else {
-        res.send('No logs yet.');
-    }
+app.get('/info', (req, res) => {
+    res.json({
+        welcome: WELCOME_MESSAGE,
+        roles: Object.keys(PROMPTS)
+    });
 });
 
-app.get('/', (req, res) => res.send('AI Agent v2 (Chat Mode) Running.'));
-
 app.listen(PORT, '0.0.0.0', () => {
-    console.log(`Server listening on ${PORT}`);
+    console.log(`--> Server listening on ${PORT}`);
     initModel();
-});
+});