Delete src

src/apiHandler.js

@@ -1,168 +0,0 @@
-// ==========================================
-//             API HANDLER FILE
-// ==========================================
-
-require('dotenv').config();
-const OpenAI = require('openai');
-const fs = require('fs');
-
-// Initialize OpenAI client
-const openai = new OpenAI({
-  apiKey: process.env.OPENAI_UNLIMITED_API_KEY,
-});
-
-// ------------------------------------------
-// Chat Completions (Text Generation, Reasoning, Function Calling)
-// ------------------------------------------
-
-async function createChatCompletion(requestBody, res) {
-  if (requestBody.stream) {
-    // Handle streaming response
-    try {
-      const completion = await openai.chat.completions.create({
-        ...requestBody,
-        stream: true,
-      });
-
-      res.setHeader('Content-Type', 'text/event-stream');
-      res.setHeader('Cache-Control', 'no-cache');
-      res.flushHeaders();
-
-      for await (const part of completion) {
-        const data = JSON.stringify(part);
-        res.write(`data: ${data}\n\n`);
-      }
-      res.write('data: [DONE]\n\n');
-      res.end();
-    } catch (error) {
-      console.error('Error in createChatCompletion:', error);
-      if (!res.headersSent) {
-        res.status(500).json(error.response ? error.response.data : error);
-      } else {
-        res.write(`data: ${JSON.stringify({ error: error.message })}\n\n`);
-        res.end();
-      }
-    }
-  } else {
-    // Non-streaming case
-    try {
-      const completion = await openai.chat.completions.create(requestBody);
-      return completion;
-    } catch (error) {
-      throw error.response ? error.response.data : error;
-    }
-  }
-}
-
-// ------------------------------------------
-// Image Generation
-// ------------------------------------------
-
-async function createImage(requestBody) {
-  try {
-    const response = await openai.images.generate(requestBody);
-    return response;
-  } catch (error) {
-    throw error.response ? error.response.data : error;
-  }
-}
-
-// ------------------------------------------
-// Image Editing
-// ------------------------------------------
-
-async function createImageEdit(requestBody) {
-  try {
-    const response = await openai.images.edit({
-      image: fs.createReadStream(requestBody.image),
-      mask: fs.createReadStream(requestBody.mask),
-      prompt: requestBody.prompt,
-      n: requestBody.n,
-      size: requestBody.size,
-    });
-    return response;
-  } catch (error) {
-    throw error.response ? error.response.data : error;
-  }
-}
-
-// ------------------------------------------
-// Image Variation
-// ------------------------------------------
-
-async function createImageVariation(requestBody) {
-  try {
-    const response = await openai.images.createVariation({
-      image: fs.createReadStream(requestBody.image),
-      n: requestBody.n,
-      size: requestBody.size,
-    });
-    return response;
-  } catch (error) {
-    throw error.response ? error.response.data : error;
-  }
-}
-
-// ------------------------------------------
-// Audio Transcription
-// ------------------------------------------
-
-async function createTranscription(requestBody) {
-  try {
-    const response = await openai.audio.transcriptions.create({
-      file: fs.createReadStream(requestBody.file),
-      model: requestBody.model,
-    });
-    return response;
-  } catch (error) {
-    throw error.response ? error.response.data : error;
-  }
-}
-
-// ------------------------------------------
-// Text to Speech
-// ------------------------------------------
-
-async function createSpeech(requestBody, res) {
-  try {
-    const response = await openai.audio.speech.create({
-      ...requestBody,
-    });
-
-    const arrayBuffer = await response.arrayBuffer();
-    res.setHeader('Content-Type', 'audio/mpeg');
-    res.send(Buffer.from(arrayBuffer));
-  } catch (error) {
-    console.error('Error in createSpeech:', error);
-    if (!res.headersSent) {
-      res.status(500).json(error.response ? error.response.data : error);
-    }
-  }
-}
-
-// ------------------------------------------
-// Vision (Image Analysis)
-// ------------------------------------------
-
-async function createVisionAnalysis(requestBody) {
-  try {
-    const response = await openai.chat.completions.create(requestBody);
-    return response;
-  } catch (error) {
-    throw error.response ? error.response.data : error;
-  }
-}
-
-// ------------------------------------------
-// Export Modules
-// ------------------------------------------
-
-module.exports = {
-  createChatCompletion,
-  createImage,
-  createImageEdit,
-  createImageVariation,
-  createTranscription,
-  createSpeech,
-  createVisionAnalysis,
-};

src/server.js

@@ -1,148 +0,0 @@
-// ==========================================
-//             SERVER WITH ENDPOINTS
-// ==========================================
-
-require('dotenv').config();
-const express = require('express');
-const app = express();
-const port = process.env.PORT || 7860;
-const apiHandler = require('./apiHandler');
-const bodyParser = require('body-parser');
-const fs = require('fs');
-
-// Middleware to parse JSON bodies
-app.use(bodyParser.json());
-
-// Middleware to parse multipart/form-data for file uploads
-const multer = require('multer');
-const upload = multer({ dest: 'uploads/' });
-
-// ------------------------------------------
-// Chat Completions Endpoint
-// ------------------------------------------
-
-app.post('/v2/chat/completions', (req, res) => {
-  if (req.body.stream) {
-    // For streaming, the apiHandler manages the response and error handling
-    apiHandler.createChatCompletion(req.body, res);
-  } else {
-    // Non-streaming
-    apiHandler
-      .createChatCompletion(req.body)
-      .then((data) => {
-        res.json(data);
-      })
-      .catch((error) => {
-        res.status(500).json(error);
-      });
-  }
-});
-
-// ------------------------------------------
-// Image Generation Endpoint
-// ------------------------------------------
-
-app.post('/v2/images/generations', async (req, res) => {
-  try {
-    const data = await apiHandler.createImage(req.body);
-    res.json(data);
-  } catch (error) {
-    res.status(500).json(error);
-  }
-});
-
-// ------------------------------------------
-// Image Editing Endpoint
-// ------------------------------------------
-
-app.post('/v2/images/edits', upload.fields([{ name: 'image' }, { name: 'mask' }]), async (req, res) => {
-  try {
-    // Add file paths to request body
-    req.body.image = req.files['image'][0].path;
-    req.body.mask = req.files['mask'][0].path;
-
-    const data = await apiHandler.createImageEdit(req.body);
-    res.json(data);
-
-    // Clean up uploaded files
-    fs.unlinkSync(req.body.image);
-    fs.unlinkSync(req.body.mask);
-  } catch (error) {
-    res.status(500).json(error);
-  }
-});
-
-// ------------------------------------------
-// Image Variations Endpoint
-// ------------------------------------------
-
-app.post('/v2/images/variations', upload.single('image'), async (req, res) => {
-  try {
-    // Add file path to request body
-    req.body.image = req.file.path;
-
-    const data = await apiHandler.createImageVariation(req.body);
-    res.json(data);
-
-    // Clean up uploaded file
-    fs.unlinkSync(req.body.image);
-  } catch (error) {
-    res.status(500).json(error);
-  }
-});
-
-// ------------------------------------------
-// Audio Transcription Endpoint
-// ------------------------------------------
-
-app.post('/v2/audio/transcriptions', upload.single('file'), async (req, res) => {
-  try {
-    // Add file path to request body
-    req.body.file = req.file.path;
-
-    const data = await apiHandler.createTranscription(req.body);
-    res.json(data);
-
-    // Clean up uploaded file
-    fs.unlinkSync(req.body.file);
-  } catch (error) {
-    res.status(500).json(error);
-  }
-});
-
-// ------------------------------------------
-// Text to Speech Endpoint
-// ------------------------------------------
-
-app.post('/v2/audio/speech', async (req, res) => {
-  try {
-    await apiHandler.createSpeech(req.body, res);
-  } catch (error) {
-    if (!res.headersSent) {
-      res.status(500).json(error);
-    } else {
-      console.error('Error after headers sent:', error);
-    }
-  }
-});
-
-// ------------------------------------------
-// Vision Endpoint
-// ------------------------------------------
-
-app.post('/v2/vision', async (req, res) => {
-  try {
-    const data = await apiHandler.createVisionAnalysis(req.body);
-    res.json(data);
-  } catch (error) {
-    res.status(500).json(error);
-  }
-});
-
-// ------------------------------------------
-// Start the Server
-// ------------------------------------------
-
-app.listen(port, () => {
-  console.log(`Server running on port ${port}`);
-});