Update server.js

server.js

@@ -1,71 +1,85 @@
-// File: app.js
+// File: server.js
 
-import express from "express";
-import bodyParser from "body-parser";
-import { ytmp4 } from "ruhend-scraper";
-import yts from "youtube-yts";
+import express from 'express';
+import { CAINode } from 'cainode';
 
 const app = express();
-const PORT = 7860;
-const HOST = "0.0.0.0";
-
-// Middleware
-app.use(bodyParser.json());
-
-// Function to check if input is a URL
-const isUrl = (input) => {
-  try {
-    new URL(input);
-    return true;
-  } catch {
-    return false;
-  }
-};
-
-// GET endpoint for downloading YouTube videos
-app.get("/download", async (req, res) => {
-  const { input } = req.query; // Extract single input parameter
-
-  if (!input) {
-    return res
-      .status(400)
-      .json({ error: "'input' query parameter must be provided." });
-  }
-
-  try {
-    let videoData;
-
-    if (isUrl(input)) {
-      // Input is a URL
-      videoData = await ytmp4(input);
-    } else {
-      // Input is a query, perform YouTube search
-      const searchResults = await yts(input);
-      if (!searchResults.videos || searchResults.videos.length === 0) {
-        return res.status(404).json({ error: "No videos found for the query." });
-      }
-      const firstResult = searchResults.videos[0];
-      videoData = await ytmp4(firstResult.url);
+const port = 3000;
+
+app.use(express.json());
+
+// API endpoint for chatting
+app.get('/chat', async (req, res) => {
+    const token = req.query.token || '529e24b4173b29dbc3054fef02a380e1e5b41949';
+    const characterId = req.query.characterId || 'smtV3Vyez6ODkwS8BErmBAdgGNj-1XWU73wIFVOY1hQ';
+    const chatId = req.query.chatId;
+    const voiceId = req.query.voiceId || 'f4b33b1d-2e5a-40a8-8a07-958140cd104d';
+    const message = req.query.message;
+
+    if (!message) {
+        return res.status(400).json({ error: 'Message query parameter is required.' });
     }
 
-    // Respond with video details
-    const { title, video, author, description, duration, views, upload, thumbnail } = videoData;
-    res.json({
-      title,
-      videoUrl: video,
-      author,
-      description,
-      duration,
-      views,
-      uploadDate: upload,
-      thumbnail,
-    });
-  } catch (error) {
-    console.error(error);
-    res.status(500).json({ error: "Failed to process the request." });
-  }
+    let client;
+    try {
+        client = new CAINode();
+
+        // Login to the client
+        await client.login(token);
+
+        // Connect to the character
+        await client.character.connect(characterId);
+
+        let chatSessionId = chatId;
+
+        // If no chatId provided, create a new conversation
+        if (!chatSessionId) {
+            const newConversation = await client.character.create_new_conversation();
+            chatSessionId = newConversation[0].chat.chat_id;
+        }
+
+        // Set the chat session
+        await client.chat.set_conversation_chat(chatSessionId);
+
+        // Send message and get response
+        const response = await client.character.send_message(message);
+
+        let reply = '';
+        let voiceUrl = '';
+
+        if (response.turn?.candidates?.length > 0) {
+            const messageResponse = response.turn.candidates[0].raw_content;
+            const turnId = response.turn.turn_key.turn_id;
+            const candidateId = response.turn.primary_candidate_id;
+
+            reply = messageResponse;
+
+            // Generate voice reply if voiceId is provided
+            if (voiceId) {
+                const voiceResponse = await client.character.replay_tts(turnId, candidateId, voiceId);
+                voiceUrl = voiceResponse.replayUrl;
+            }
+
+            res.json({ chatId: chatSessionId, message: reply, voiceUrl });
+        } else {
+            res.status(404).json({ error: 'No response content available.' });
+        }
+
+        // Logout and cleanup
+        await client.character.disconnect();
+        await client.logout();
+
+    } catch (error) {
+        console.error('Error:', error.message);
+        res.status(500).json({ error: error.message });
+
+        if (client) {
+            await client.logout();
+        }
+    }
 });
 
-app.listen(PORT, HOST, () => {
-  console.log(`Running on http://${HOST}:${PORT}`);
+// Start the server
+app.listen(port, () => {
+    console.log(`Server running at http://localhost:${port}`);
 });