Update server.js

server.js

@@ -8,265 +8,88 @@ app.use(cors());
 app.use(express.json({ limit: '50mb' }));
 
 const PORT = process.env.PORT || 7860;
-
-// Ollama Configuration - use 127.0.0.1 instead of localhost
-const OLLAMA_HOST = process.env.OLLAMA_URL || 'http://127.0.0.1:11434';
+const OLLAMA_HOST = '127.0.0.1';
+const OLLAMA_PORT = 11434;
 const MODEL = process.env.MODEL || 'qwen3.5:0.8b';
 
-// Parse OLLAMA_URL to get hostname and port
-const OLLAMA_URL_OBJ = new URL(OLLAMA_HOST);
-const OLLAMA_HOSTNAME = OLLAMA_URL_OBJ.hostname;
-const OLLAMA_PORT = OLLAMA_URL_OBJ.port || 11434;
-
-// Tera V2 System Prompt
-const SYSTEM_PROMPT = `You are Tera V2, an advanced AI document intelligence assistant.
-
-Your capabilities:
-- Analyze complex documents, contracts, and research papers
-- Extract key insights and summarize content  
-- Answer questions based on provided documents
-- Generate podcast scripts and audio content
-
-Response Style:
-- Professional yet approachable
-- Clear and concise
-- Cite specific parts of documents when answering
-- Format responses for readability`;
+console.log('Tera V2 Starting...');
 
-console.log('╔════════════════════════════════════════╗');
-console.log('║         Tera V2 Starting...            ║');
-console.log('╚════════════════════════════════════════╝');
-console.log(`Model: ${MODEL}`);
-console.log(`Ollama: ${OLLAMA_HOST}:${OLLAMA_PORT}\n`);
-
-// Root route - API info
+// Root route
 app.get('/', (req, res) => {
-  res.json({
-    name: 'Tera V2 API',
-    version: '2.0.0',
-    model: MODEL,
-    status: 'running',
-    ollama: `${OLLAMA_HOST}:${OLLAMA_PORT}`,
-    endpoints: {
-      health: '/health',
-      chat: '/api/chat',
-      audio: '/api/generate-audio-script'
-    }
-  });
+    res.json({ name: 'Tera V2 API', version: '2.0.0', model: MODEL, status: 'running' });
 });
 
 // Health check
-app.get('/health', async (req, res) => {
-  try {
-    const options = {
-      hostname: OLLAMA_HOSTNAME,
-      port: OLLAMA_PORT,
-      path: '/api/tags',
-      method: 'GET',
-      timeout: 5000
-    };
-
-    const request = http.request(options, (response) => {
-      let data = '';
-      response.on('data', chunk => data += chunk);
-      response.on('end', () => {
-        try {
-          const models = JSON.parse(data).models || [];
-          const hasModel = models.some(m => m.name.includes(MODEL));
-          res.json({ 
-            status: 'ok', 
-            service: 'Tera V2',
-            model: MODEL,
-            modelLoaded: hasModel,
-            ollamaConnected: true
-          });
-        } catch (e) {
-          res.status(503).json({ 
-            status: 'error', 
-            message: 'Ollama not responding correctly',
-            ollamaConnected: false
-          });
-        }
-      });
-    });
-    
-    request.on('error', (err) => {
-      console.error('Health check error:', err.message);
-      res.status(503).json({ 
-        status: 'error', 
-        message: 'Ollama not running',
-        solution: 'Wait for container to finish starting',
-        ollamaConnected: false
-      });
-    });
-    
-    request.setTimeout(5000, () => {
-      request.destroy();
-      res.status(503).json({ 
-        status: 'error', 
-        message: 'Ollama timeout',
-        ollamaConnected: false
-      });
-    });
-
-    request.end();
-    
-  } catch (error) {
-    res.status(503).json({ 
-      status: 'error', 
-      message: error.message,
-      ollamaConnected: false
-    });
-  }
+app.get('/health', (req, res) => {
+    res.json({ status: 'ok', service: 'Tera V2', model: MODEL });
 });
 
-// Chat endpoint with streaming
+// SIMPLE CHAT (Non-streaming - works better)
 app.post('/api/chat', async (req, res) => {
-  try {
-    const { messages, sources } = req.body;
-    
-    // Build context from sources
-    let context = '';
-    if (sources && sources.length > 0) {
-      context = '\n\nDOCUMENTS:\n' + sources.map((s, i) => 
-        `[${i + 1}] ${s.name}:\n${s.content?.substring(0, 3000) || '[Image]'}`
-      ).join('\n\n');
-    }
-
-    const lastMessage = messages[messages.length - 1];
-    const prompt = `${SYSTEM_PROMPT}${context}\n\nUser: ${lastMessage.content}\n\nTera V2:`;
-
-    res.setHeader('Content-Type', 'text/event-stream');
-    res.setHeader('Cache-Control', 'no-cache');
-
-    const requestData = JSON.stringify({
-      model: MODEL,
-      prompt: prompt,
-      stream: true,
-      options: { temperature: 0.7, num_predict: 4096 }
-    });
-
-    const options = {
-      hostname: OLLAMA_HOSTNAME,
-      port: OLLAMA_PORT,
-      path: '/api/generate',
-      method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-        'Content-Length': Buffer.byteLength(requestData)
-      },
-      timeout: 120000
-    };
-
-    const proxyReq = http.request(options, (proxyRes) => {
-      proxyRes.on('data', (chunk) => {
-        const lines = chunk.toString().split('\n');
-        for (const line of lines) {
-          if (line.trim()) {
-            try {
-              const data = JSON.parse(line);
-              if (data.response) {
-                res.write(`data: {"content": ${JSON.stringify(data.response)}}\n\n`);
-              }
-              if (data.done) {
-                res.write('data: {"done": true}\n\n');
-              }
-            } catch (e) {}
-          }
+    try {
+        const { messages, sources } = req.body;
+        
+        // Build prompt
+        let context = '';
+        if (sources && sources.length > 0) {
+            context = '\n\nDOCUMENTS:\n' + sources.map(s => s.name + ':\n' + (s.content?.substring(0, 2000) || '')).join('\n\n');
         }
-      });
-      
-      proxyRes.on('end', () => {
-        res.write('data: {"done": true}\n\n');
-        res.end();
-      });
-    });
-
-    proxyReq.on('error', (error) => {
-      console.error('Ollama error:', error.message);
-      res.write(`data: {"error": "Ollama connection failed: ${error.message}"}\n\n`);
-      res.end();
-    });
-
-    proxyReq.write(requestData);
-    proxyReq.end();
-
-  } catch (error) {
-    console.error('Chat error:', error.message);
-    res.status(500).json({ error: error.message });
-  }
-});
-
-// Audio script generation
-app.post('/api/generate-audio-script', async (req, res) => {
-  try {
-    const { sources, format, instructions } = req.body;
-    
-    let content = '';
-    if (sources && sources.length > 0) {
-      content = sources.map(s => s.content?.substring(0, 4000) || '').join('\n\n');
+        
+        const prompt = `You are Tera V2, an AI assistant.${context}\n\nUser: ${messages[messages.length-1].content}\n\nTera V2:`;
+        
+        console.log('Processing:', messages[messages.length-1].content.substring(0, 50));
+        
+        // Call Ollama
+        const requestData = JSON.stringify({
+            model: MODEL,
+            prompt: prompt,
+            stream: false,
+            options: { temperature: 0.7, num_predict: 2048 }
+        });
+        
+        const options = {
+            hostname: OLLAMA_HOST,
+            port: OLLAMA_PORT,
+            path: '/api/generate',
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                'Content-Length': Buffer.byteLength(requestData)
+            },
+            timeout: 120000
+        };
+        
+        const proxyReq = http.request(options, (proxyRes) => {
+            let data = '';
+            proxyRes.on('data', chunk => data += chunk);
+            proxyRes.on('end', () => {
+                try {
+                    const response = JSON.parse(data);
+                    // Return in same format as before for compatibility
+                    res.json({ 
+                        content: response.response,
+                        done: true 
+                    });
+                } catch (e) {
+                    res.status(500).json({ error: 'Failed to parse response' });
+                }
+            });
+        });
+        
+        proxyReq.on('error', (e) => {
+            console.error('Ollama error:', e.message);
+            res.status(500).json({ error: e.message });
+        });
+        
+        proxyReq.write(requestData);
+        proxyReq.end();
+        
+    } catch (error) {
+        console.error('Error:', error.message);
+        res.status(500).json({ error: error.message });
     }
-
-    const formatPrompts = {
-      deepdive: 'Create an in-depth podcast discussion between two hosts analyzing the content.',
-      briefing: 'Create a brief news-style summary of the key points.',
-      story: 'Create a narrative-driven exploration of the content.'
-    };
-
-    const customInstr = instructions ? ` ${instructions}` : '';
-    const prompt = `${SYSTEM_PROMPT}\n\nYou are creating a podcast script. ${formatPrompts[format] || formatPrompts.deepdive}${customInstr}\n\nDocuments:\n${content}\n\nScript:`;
-
-    const requestData = JSON.stringify({
-      model: MODEL,
-      prompt: prompt,
-      stream: false,
-      options: { temperature: 0.8, num_predict: 4096 }
-    });
-
-    const options = {
-      hostname: OLLAMA_HOSTNAME,
-      port: OLLAMA_PORT,
-      path: '/api/generate',
-      method: 'POST',
-      headers: {
-        'Content-Type': 'application/json',
-        'Content-Length': Buffer.byteLength(requestData)
-      }
-    };
-
-    const proxyReq = http.request(options, (proxyRes) => {
-      let data = '';
-      proxyRes.on('data', chunk => data += chunk);
-      proxyRes.on('end', () => {
-        try {
-          const response = JSON.parse(data);
-          res.json({ script: response.response, success: true });
-        } catch (e) {
-          res.status(500).json({ error: 'Invalid response from Ollama' });
-        }
-      });
-    });
-
-    proxyReq.on('error', (error) => {
-      res.status(500).json({ error: error.message });
-    });
-
-    proxyReq.write(requestData);
-    proxyReq.end();
-
-  } catch (error) {
-    res.status(500).json({ error: error.message });
-  }
 });
 
 app.listen(PORT, () => {
-  console.log('╔════════════════════════════════════════╗');
-  console.log('║     Tera V2 is Ready!                  ║');
-  console.log('╚════════════════════════════════════════╝');
-  console.log(`Server: http://localhost:${PORT}`);
-  console.log(`Ollama: ${OLLAMA_HOST}:${OLLAMA_PORT}`);
-  console.log(`Model: ${MODEL}`);
-  console.log('');
-  console.log('Waiting for Ollama to be ready...');
-  console.log('╔════════════════════════════════════════╗\n');
+    console.log('Tera V2 Ready on port', PORT);
 });