Update retrievers/community_diffusion_models.js

retrievers/community_diffusion_models.js

@@ -1,4 +1,3 @@
-const express = require('express');
 const fs = require('fs');
 // Read and Parse JSON Data
 const data = JSON.parse(fs.readFileSync('diffusers-unfiltered.json', 'utf8'));
@@ -29,26 +28,26 @@ function filterOptimizedOnly(data) {
   return data
 }
 
-async function queryCommunityModels(req, res) {
+async function queryCommunityModels(queryParams) {
   let result = data;
 
-  if (req.query.sortBy) {
-    result = sortData(result, req.query.sortBy);
+  if (queryParams.sortBy) {
+    result = sortData(result, queryParams.sortBy);
   }
 
-  if (req.query.filterBy && req.query.threshold) {
-    result = filterData(result, req.query.filterBy,  req.query.threshold ?  
-        parseFloat(req.query.threshold)
+  if (queryParams.filterBy && queryParams.threshold) {
+    result = filterData(result, queryParams.filterBy,  queryParams.threshold ?  
+        parseFloat(queryParams.threshold)
         :
         0.5
     );
   }
 
-  if (req.query.optimized_only === 'true') {
+  if (queryParams.optimized_only === 'true') {
     result = filterOptimizedOnly(result);
   }
   console.log(result.length, " items returned");
-  res.json(makeReadable(result));
+  return(makeReadable(result));
 }
 
 module.exports={queryCommunityModels}