Add code for categories.

main.py

@@ -291,6 +291,9 @@ class PromptServer(BaseHTTPRequestHandler):
                 info['output'] = obj_class.RETURN_TYPES
                 info['name'] = x #TODO
                 info['description'] = ''
+                info['category'] = 'sd'
+                if hasattr(obj_class, 'CATEGORY'):
+                    info['category'] = obj_class.CATEGORY
                 out[x] = info
             self.wfile.write(json.dumps(out).encode('utf-8'))
         elif self.path[1:] in os.listdir(self.server.server_dir):

webshit/index.html

@@ -237,7 +237,7 @@ function onObjectInfo(json) {
         MyNode.__json_data = json[key]
 
         LiteGraph.registerNodeType(key, MyNode);
-        MyNode.category = "sd"; //TODO: proper categories
+        MyNode.category = json[key]['category'];
     };
 
     afterLoadGraph();