Update server.js

server.js

@@ -13,6 +13,16 @@ const openaiKey = process.env.OPENAI_KEY;
 const port = 7860;
 const baseUrl = getExternalUrl(process.env.SPACE_ID);
 
+const allowedModels = [
+    'gpt-3.5-turbo-0125',
+    'gpt-3.5-turbo',
+    'gpt-3.5-turbo-1106',
+    'gpt-3.5-turbo-instruct',
+    'gpt-3.5-turbo-16k',
+    'gpt-3.5-turbo-0613',
+    'gpt-3.5-turbo-16k-0613'
+];
+
 let requestDetails = [];
 let apiKeyDetails = {}; // Track API key usage and first use timestamp
 
@@ -71,8 +81,16 @@ function authenticateApiKey(req, res, next) {
 }
 
 app.use('/api', authenticateApiKey, (req, res, next) => {
+    const urlParts = req.url.split('/');
+    const modelId = urlParts.length > 1 ? urlParts[1] : null;
+
+    // Check if the model ID is part of the allowed models.
+    if (!modelId || !allowedModels.includes(modelId)) {
+        return res.status(403).send("Access Denied: This model is not allowed.");
+    }
+
     const timestamp = moment().tz("Asia/Tbilisi").format("DD-MMM-YYYY HH:mm");
-    const displayKey = req.apiKey.substring(0, 4);
+    const displayKey = req.apiKey.substring(0, 10);
     const requestData = {
         requestNumber: apiKeyDetails[req.apiKey].count,
         apiKey: displayKey,
@@ -81,7 +99,7 @@ app.use('/api', authenticateApiKey, (req, res, next) => {
     };
     requestDetails.push(requestData);
 
-    const detailedLog = `Request ${requestData.requestNumber} on ${requestData.timestamp} from API Key ${displayKey}... with method "${req.method}" and text "${requestData.text}"`;
+    const detailedLog = `Request ${requestData.requestNumber} on ${requestData.timestamp} from API Key ${displayKey}... with method "${req.method}" and URL line "${req.url}"`;
     logAndEmit(detailedLog, true);
 
     next();