Policies are now loaded from the hub

js/draw_p5js.js

@@ -225,8 +225,8 @@ function drawName(agent, scale){
     else if(agent.morphology == "fish"){
         y_pos = pos.y + agent.agent_body.AGENT_HEIGHT * 2;
     }
-    let age = agent.age == "adult" ? "" : " (" + window.lang_dict[window.get_language()]['morphologies'][agent.age] + ")";
-    text(agent.name + age, x_pos, RENDERING_VIEWER_H - y_pos);
+
+    text(agent.name, x_pos, RENDERING_VIEWER_H - y_pos);
 }
 
 /**

js/envs/multi_agents_continuous_parkour.js

@@ -91,7 +91,7 @@ class MultiAgentsContinuousParkour {
     /**
      * Creates an agent with the given parameters.
      * @param morphology {string} - Name of the morphology
-     * @param policy {{name: string, age: string, path: string}} - Name and path of the policy model
+     * @param policy {{name: string path: string}} - Name and path of the policy model
      * @param init_pos {{x: number, y: number}} - Initial position of the agent
      */
     create_agent(morphology, policy, init_pos){
@@ -99,7 +99,6 @@ class MultiAgentsContinuousParkour {
         let agent = {
             id: this.agents.length,
             name: policy.name,
-            age: policy.age,
             is_selected: false,
             morphology: morphology,
             policy: policy,

js/ui_state/components/agents_list.js

@@ -23,7 +23,6 @@ export default class AgentsList extends Component {
         let morph_dict = window.lang_dict[store.state.language]['morphologies'];
         this.element.querySelector('#agents_list_title').innerHTML = `<strong>${dict['title']}</strong>`;
         this.element.querySelector('#agents_list').innerHTML = store.state.agents.map(agent => {
-            let age = agent.age == "adult" ? "" : " (" + window.lang_dict[window.get_language()]['morphologies'][agent.age] + ")";
             // Creates a list item for each agent
             return `<li name="agent-list-item" class="list-group-item d-flex justify-content-between align-items-center px-0 py-1">
 
@@ -36,7 +35,7 @@ export default class AgentsList extends Component {
                              
                     <!-- Text field of the name of the agent -->
                     <div class="form-group">
-                        <input name="agentNameArea" type="text" class="form-control w-75 mx-1" placeholder="${agent.name + age}">
+                        <input name="agentNameArea" type="text" class="form-control w-75 mx-1" placeholder="${agent.name}">
                     </div>
                     
                     <!-- Follow switch -->         

js/ui_state/components/morphologies_list.js

@@ -76,20 +76,16 @@ export default class MorphologiesList extends Component {
                 .filter(m => m.morphology == store.state.morphologies[index].morphology)
                 .flatMap(morphology => morphology.seeds)
                 .map((seedEntry, idx) => {
-
-                    // Splits seed name between name and age
-                    let seed = seedEntry.name.split('/');
-                    let name = seed[0];
-                    let age = seed[1];
+                    let name = seedEntry.name;
 
                     // Adds options and optgroups
                     if(seed_names.hasOwnProperty(name)){
-                        seed_names[name] += `<option value="${seedEntry.path}">${name + " (" + dict[age] + ")"}</option>`;
+                        seed_names[name] += `<option value="${seedEntry.path}">${name}</option>`;
 
                     }
                     else {
                         seed_names[name] = `<optgroup label=${name}>
-                                                <option value="${seedEntry.path}">${name + " (" + dict[age] + ")"}</option>`;
+                                                <option value="${seedEntry.path}">${name}</option>`;
                     }
                 });
             let options = [];
@@ -111,11 +107,10 @@ export default class MorphologiesList extends Component {
         this.element.querySelectorAll('button[name="addAgentButton"]').forEach((span, index) => {
             span.addEventListener('click', () => {
                 let morph = store.state.morphologies[index];
-                let seed = morph.seeds[store.state.currentSeedsIdx[morph.morphology]].name.split('/');
+                let name = morph.seeds[store.state.currentSeedsIdx[morph.morphology]].name;
                 store.dispatch('addAgent', {
                     morphology: morph.morphology,
-                    name: seed[0],
-                    age: seed[1],
+                    name: name,
                     path: morph.seeds[store.state.currentSeedsIdx[morph.morphology]].path,
                     init_pos: null
                 });

js/ui_state/store/actions.js

@@ -52,7 +52,6 @@ export default {
             context.commit('addAgent', {
                 morphology: agent.morphology,
                 name: agent.name,
-                age: agent.age,
                 path: agent.path,
                 init_pos: agent.init_pos
             });

js/ui_state/store/mutations.js

@@ -155,7 +155,6 @@ export default {
             policies: state.agents.map(a => {
                 return {
                     name: a.name,
-                    age: a.age,
                     path: a.path
                 };
             }),
@@ -200,13 +199,13 @@ export default {
     /**
      * Adds the given agent to the environment and renders it.
      * @param state {Object} - UI state
-     * @param payload {{morphology: string, name: string, age: string, path: string, init_pos: {x: number, y: number}}}
+     * @param payload {{morphology: string, name: string, path: string, init_pos: {x: number, y: number}}}
      * @return {Object} - UI state
      */
     addAgent(state, payload) {
         state.agents.push(payload);
         if(window.game != null){
-            window.game.env.add_agent(payload.morphology, {name: payload.name, age: payload.age, path: payload.path}, payload.init_pos);
+            window.game.env.add_agent(payload.morphology, {name: payload.name, path: payload.path}, payload.init_pos);
             window.game.env.render();
         }
         return state;

ui.js

@@ -286,24 +286,44 @@ globalElementsInstance.render();
 /*
  * Fetches the available morphologies and policies from the JSON file
  */
-fetch('./policies.json')
+fetch('https://huggingface.co/api/models?filter=teach-my-agent-parkour')
     .then(resp => resp.text().then(body => {
-        window.agent_policies = JSON.parse(body);
-        return window.agent_policies;
+        return JSON.parse(body);
     }))
-    .then(types => {
-        types.forEach(type => {
-            type["morphologies"].forEach(morphology => {
+    .then(models => {
+        let morphologies = {}
+        let promises = []
+        models.forEach(model => {
+            promises.push(
+                fetch('https://huggingface.co/'+model.id+'/raw/main/ta-config.json')
+                .then(result => result.text().then(body => {
+                    let policy = JSON.parse(body);
+                    if(!(policy["body"] in morphologies)){
+                        morphologies[policy["body"]] = {
+                            "morphology": policy["body"],
+                            "seeds": []
+                        }
+                    }
+
+                    morphologies[policy["body"]]["seeds"].push({
+                        "seed": policy["seed"],
+                        "name": policy["name"],
+                        "path": 'https://huggingface.co/'+model.id+'/resolve/main'
+                    })
+                }))
+            ); 
+        });
 
-                // Adds the morphology with its compatible policies to the list of morphologies
+        Promise.all(promises).then(() => {
+            for (const [key, value] of Object.entries(morphologies)) {
                 store.dispatch('addMorphology', {
-                    morphology: morphology["morphology"],
-                    seeds: morphology["seeds"].map((seed, index) => {
+                    morphology: value["morphology"],
+                    seeds: value["seeds"].map((seed, index) => {
                         seed["idx"] = index;
                         return seed;
                     })
                 });
-            });
+            }
         });
     });