Feature NMH-65: Collect and use statistics on downloads of model

JohnSnowLabs · Feb 24, 2022 · 5637cf9 · 5637cf9
1 parent 374ed40
commit 5637cf9
Show file tree

Hide file tree

Showing 3 changed files with 41 additions and 13 deletions.
diff --git a/docs/_frontend/src/models/Sidebar/Sidebar.js b/docs/_frontend/src/models/Sidebar/Sidebar.js
@@ -1,7 +1,7 @@
-import Combobox from '../Combobox';
-import Radio from '../Radio/Radio';
-import RadioGroup from '../RadioGroup';
-import styles from './Sidebar.module.css';
+import Combobox from "../Combobox";
+import Radio from "../Radio/Radio";
+import RadioGroup from "../RadioGroup";
+import styles from "./Sidebar.module.css";
 
 const Sidebar = ({ meta, params, onSubmit }) => {
   let tags = [];
@@ -17,20 +17,20 @@ const Sidebar = ({ meta, params, onSubmit }) => {
     <div className={styles.root}>
       <div className={styles.control}>
         <RadioGroup
-          value={params.type || ''}
+          value={params.type || ""}
           onChange={(value) => {
             onSubmit({ type: value });
           }}
         >
-          <Radio value="">All</Radio>
-          <Radio value="model">Models</Radio>
-          <Radio value="pipeline">Pipelines</Radio>
+          <Radio value=''>All</Radio>
+          <Radio value='model'>Models</Radio>
+          <Radio value='pipeline'>Pipelines</Radio>
         </RadioGroup>
       </div>
 
       <div className={styles.control}>
         <Combobox
-          label="Assigned tags"
+          label='Assigned tags'
           items={tags}
           initialSelectedItems={params.tags || []}
           onChange={(values) => {
@@ -41,7 +41,7 @@ const Sidebar = ({ meta, params, onSubmit }) => {
 
       <div className={styles.control}>
         <Combobox
-          label="Entities"
+          label='Entities'
           items={predictedEntities}
           initialSelectedItems={params.predicted_entities || []}
           onChange={(values) => {
@@ -52,14 +52,23 @@ const Sidebar = ({ meta, params, onSubmit }) => {
 
       <div className={styles.control}>
         <RadioGroup
-          value={params.sort || 'date'}
+          value={params.sort || "date"}
           onChange={(value) => {
             onSubmit({ sort: value });
           }}
         >
           <label>Sort By:</label>
-          <Radio value="date">Date</Radio>
-          <Radio value="views">Views</Radio>
+          <div className={styles.radioContainers}>
+            <Radio value='date' className={styles.radio}>
+              Date
+            </Radio>
+            <Radio value='views' className={styles.radio}>
+              Views
+            </Radio>
+            <Radio value='downloads' className={styles.radio}>
+              Downloads
+            </Radio>
+          </div>
         </RadioGroup>
       </div>
     </div>

diff --git a/docs/_frontend/src/models/Sidebar/Sidebar.module.css b/docs/_frontend/src/models/Sidebar/Sidebar.module.css
@@ -14,3 +14,11 @@
 .control:not(:last-child) {
   margin-bottom: 20px;
 }
+
+.radioContainers {
+  flex-direction: column;
+}
+
+.radioContainers .radio {
+  display: block;
+}
diff --git a/docs/_plugins/search_index.rb b/docs/_plugins/search_index.rb
@@ -304,6 +304,15 @@ def is_latest?(group, model)
             },
             "url": {
               "type": "keyword"
+            },
+            "download_link": {
+              "type": "keyword"
+            },
+            "views": {
+              "type": "integer"
+            },
+            "downloads": {
+              "type": "integer"
             }
         }
       }
@@ -341,6 +350,8 @@ def is_latest?(group, model)
       model[:type] = models_json[model[:url]][:type]
       # Add predicted entities to search models
       model[:predicted_entities] = models_json[model[:url]][:predicted_entities]
+      # Add download_link to search models
+      model[:download_link] = models_json[model[:url]][:download_link]
 
       if client
         if force_reindex || uniq_for_indexing.include?(uniq)