Update server.js

server.js

@@ -1,14 +1,12 @@
-const express = require('express');
+Pakai ini const express = require('express');
 const axios = require('axios');
 const fs = require('fs');
 const path = require('path');
-const swaggerUi = require('swagger-ui-express');
-const swaggerJsDoc = require('swagger-jsdoc');
 
 const app = express();
 app.set('json spaces', 2);
 
-const geminiApiKey = 'YOUR_GEMINI_API_KEY';
+const geminiApiKey = 'AIzaSyCRpX4SnNQl__BNLcVWqgzKdbIlt06vod0';
 const geminiApiUrl = 'https://generativelanguage.googleapis.com/v1/models/gemini-pro:generateContent';
 
 const geminiTmpDir = path.join(require('os').tmpdir(), 'gemini_sessions');
@@ -16,125 +14,76 @@ const deepseekTmpDir = path.join(require('os').tmpdir(), 'deepseek_sessions');
 if (!fs.existsSync(geminiTmpDir)) fs.mkdirSync(geminiTmpDir);
 if (!fs.existsSync(deepseekTmpDir)) fs.mkdirSync(deepseekTmpDir);
 
-const swaggerOptions = {
-    swaggerDefinition: {
-        openapi: '3.0.0',
-        info: {
-            title: 'scnario AI',
-            version: '1.0.0',
-            description: 'Free AI.'
-        },
-        servers: [{ url: 'https://scnario-gemini.hf.space' }],
-    },
-    apis: [__filename],
-};
-
-const swaggerDocs = swaggerJsDoc(swaggerOptions);
-app.use('/', swaggerUi.serve, swaggerUi.setup(swaggerDocs));
-
 function getSessionFile(sessionId, model) {
-    const dir = model === 'gemini' ? geminiTmpDir : deepseekTmpDir;
-    return path.join(dir, `${sessionId}.json`);
+const dir = model === 'gemini' ? geminiTmpDir : deepseekTmpDir;
+return path.join(dir, ${sessionId}.json);
 }
 
 async function gemini(sessionId, prompt) {
-    const sessionFile = getSessionFile(sessionId, 'gemini');
-    let history = fs.existsSync(sessionFile) ? JSON.parse(fs.readFileSync(sessionFile, 'utf8')) : [];
-    history.push({ role: 'user', content: { parts: [{ text: prompt }] } });
+const sessionFile = getSessionFile(sessionId, 'gemini');
+let history = [];
+if (fs.existsSync(sessionFile)) {
+history = JSON.parse(fs.readFileSync(sessionFile, 'utf8'));
+}
+history.push({ role: 'user', content: { parts: [{ text: prompt }] } });
+
+const response = await axios.post(`${geminiApiUrl}?key=${geminiApiKey}`, {  
+    contents: history.map(entry => ({  
+        role: entry.role === 'assistant' ? 'model' : 'user',  
+        parts: entry.content.parts  
+    }))  
+}, {  
+    headers: { 'Content-Type': 'application/json' }  
+});  
+  
+const reply = response.data.candidates?.[0]?.content?.parts?.[0]?.text || 'No response';  
+history.push({ role: 'assistant', content: { parts: [{ text: reply }] } });  
+fs.writeFileSync(sessionFile, JSON.stringify(history, null, 2));  
+return reply;
 
-    const response = await axios.post(`${geminiApiUrl}?key=${geminiApiKey}`, {
-        contents: history.map(entry => ({
-            role: entry.role,
-            parts: entry.content.parts
-        }))
-    }, {
-        headers: { 'Content-Type': 'application/json' }
-    });
-    
-    const reply = response.data.candidates?.[0]?.content?.parts?.[0]?.text || 'No response';
-    history.push({ role: 'assistant', content: { parts: [{ text: reply }] } });
-    fs.writeFileSync(sessionFile, JSON.stringify(history, null, 2));
-    return reply;
 }
 
 async function deepseek(sessionId, prompt) {
-    const sessionFile = getSessionFile(sessionId, 'deepseek');
-    let history = fs.existsSync(sessionFile) ? JSON.parse(fs.readFileSync(sessionFile, 'utf8')) : [];
-    history.push({ role: 'user', content: prompt });
+const sessionFile = getSessionFile(sessionId, 'deepseek');
+let history = [];
+if (fs.existsSync(sessionFile)) {
+history = JSON.parse(fs.readFileSync(sessionFile, 'utf8'));
+}
+history.push({ role: 'user', content: prompt });
+
+let { data } = await axios.post("https://ai.clauodflare.workers.dev/chat", {  
+    "model": "@cf/deepseek-ai/deepseek-r1-distill-qwen-32b",  
+    "messages": history  
+}).catch(e => e.response);  
+  
+if (!data.success) throw new Error(JSON.stringify(data, null, 2));  
+let response = data.data.response  
+history.push({ role: 'assistant', content: response });  
+fs.writeFileSync(sessionFile, JSON.stringify(history, null, 2));  
+return response;
 
-    let { data } = await axios.post("https://ai.clauodflare.workers.dev/chat", {
-        "model": "@cf/deepseek-ai/deepseek-r1-distill-qwen-32b",
-        "messages": history
-    }).catch(e => e.response);
-    
-    if (!data.success) throw new Error(JSON.stringify(data, null, 2));
-    let response = data.data.response
-    history.push({ role: 'assistant', content: response });
-    fs.writeFileSync(sessionFile, JSON.stringify(history, null, 2));
-    return response;
 }
 
-/**
- * @swagger
- * /gemini:
- *   get:
- *     summary: Get response from Gemini AI
- *     parameters:
- *       - in: query
- *         name: question
- *         required: true
- *         schema:
- *           type: string
- *       - in: query
- *         name: sessionId
- *         required: true
- *         schema:
- *           type: string
- *     responses:
- *       200:
- *         reply: Lorem ipsum dolor sit amet
- */
 app.get('/gemini', async (req, res) => {
-    try {
-        const { question, sessionId } = req.query;
-        if (!question || !sessionId) return res.status(400).json({ error: 'Missing parameters' });
-        const reply = await gemini(sessionId, question);
-        res.json({ reply });
-    } catch (error) {
-        res.status(500).json({ error: error.response?.data || error.message });
-    }
+try {
+const { question, sessionId } = req.query;
+if (!question || !sessionId) return res.status(400).json({ error: 'Missing parameters' });
+const reply = await gemini(sessionId, question);
+res.json({ reply });
+} catch (error) {
+res.status(500).json({ error: error.response?.data || error.message });
+}
 });
 
-/**
- * @swagger
- * /deepseek-ai/r1:
- *   get:
- *     summary: Get response from DeepSeek AI
- *     parameters:
- *       - in: query
- *         name: question
- *         required: true
- *         schema:
- *           type: string
- *       - in: query
- *         name: sessionId
- *         required: true
- *         schema:
- *           type: string
- *     responses:
- *       200:
- *         reply: Lorem ipsum dolor sit amet
- *         think: Lorem ipsum dolor sit amet
- */
-app.get('/deepseek-ai/r1', async (req, res) => {
-    try {
-        const { question, sessionId } = req.query;
-        if (!question || !sessionId) return res.status(400).json({ error: 'Missing parameters' });
-        const reply = await deepseek(sessionId, question);
-        res.json({ reply });
-    } catch (error) {
-        res.status(500).json({ error: error.response?.data || error.message });
-    }
+app.get('/deepseek-ai/deepseek-r1-distill-qwen-32b', async (req, res) => {
+try {
+const { question, sessionId } = req.query;
+if (!question || !sessionId) return res.status(400).json({ error: 'Missing parameters' });
+const reply = await deepseek(sessionId, question);
+res.json({ reply: reply.split("</think>")?.pop()?.trim() || reply.split("</think>")[1]?.trim(), think: reply.split('</think>')[0]?.trim() });
+} catch (error) {
+res.status(500).json({ error: error.response?.data || error.message });
+}
 });
 
-app.listen(7860, () => console.log('Server running on port 7860'));
+app.listen(7860);