Issue #2: fixed coloring bug

src/apps/viewer/index.html

@@ -55,7 +55,7 @@
             });
 
             // Set PDB Id here
-            var pdbId = '1afo';
+            var pdbId = '6xvf';
             function loadPdb() {
                 molstar.loadWithUNITMPMembraneRepresentation(viewer.plugin, {
                     structureUrl: `https://cs.litemol.org/${pdbId}/full`,

src/extensions/tmdet/behavior.ts

@@ -220,9 +220,11 @@ async function createStructureRepresentation(plugin: PluginUIContext, pdbtmDescr
         }
 
         console.log(`REGIONS-${chain.chain_label}`, regionsBySite);
-        const regionUpdates = plugin.build();
-        createRegionRepresentation(plugin, chain.chain_label, regionsBySite[0], regionUpdates.to(structure));
-        regionUpdates.commit();
+        regionsBySite.map((region: PDBTMRegion) => {
+            const regionUpdates = plugin.build();
+            createRegionRepresentation(plugin, chain.chain_label, region, regionUpdates.to(structure));
+            regionUpdates.commit();
+        });
 
     });
 }