Hugging Face's logo

Spaces:
omnitool-ai
/
omnitool_on_hf
Running

App Files Community
omnitool_on_hf
File size: 1,369 Bytes 
b39afbe
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
 
/**
 * Copyright (c) 2023 MERCENARIES.AI PTE. LTD.
 * All rights reserved.
 */

import DOMPurify from 'dompurify';
import { ChatRenderer } from 'omni-client-services';
import { OAIComponent31 } from 'omni-sockets';

// An extension to render an omnitool component object

class OmniComponentRenderer extends ChatRenderer {
  constructor(id?: string, opts?: any) {
    super({ id: id ?? 'omni/component' }, opts);
  }

  render(content: { type: string; value: any }): string {
    const componentJson = JSON.parse(content.value);

    let component;
    if (componentJson.type === 'OAIComponent31') {
      component = OAIComponent31.fromJSON(componentJson);
    }

    if (!component) {
      //WorkflowComponentRegistry.getSingleton().add([component])
      return `Can not render missing component ${content.value}`;
    }

    return `
      <div class='component block rounded-lg px-4 py-2 text-sm w-full text-gray-500 hover:bg-gray-50 hover:text-gray-700  cursor-pointer'
        x-data='{}' x-tooltip='${component?.name}'
        x-on:click="window.client.runScript('add',['${DOMPurify.sanitize(component.name)}'])">
        <div

          class='font-semibold max-w-md'>${DOMPurify.sanitize(component?.title)}
        </div>

        <div>
        ${DOMPurify.sanitize(component?.description)}</div>

      </div>`;
  }
}

export default OmniComponentRenderer;