File size: 895 Bytes
ce00289
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
/**
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 * All rights reserved.
 *
 * This source code is licensed under the license found in the
 * LICENSE file in the root directory of this source tree.
 */

import React from "react"
import ReactDOM from "react-dom"

import {
  ComponentProps,
  withStreamlitConnection,
} from "streamlit-component-lib"


import ContributionGraph from "./ContributionGraph"
import Selector from "./Selector"

const LlmViewerComponent = (props: ComponentProps) => {
  switch (props.args['component']) {
    case 'graph':
      return <ContributionGraph />
    case 'selector':
      return <Selector />
    default:
      return <></>
  }
};

const StreamlitLlmViewerComponent = withStreamlitConnection(LlmViewerComponent)

ReactDOM.render(
  <React.StrictMode>
    <StreamlitLlmViewerComponent />
  </React.StrictMode>,
  document.getElementById("root")
)