diff --git a/packages/base/src/annotations/components/Annotation.tsx b/packages/base/src/annotations/components/Annotation.tsx index f3aa8196d..a4118f30b 100644 --- a/packages/base/src/annotations/components/Annotation.tsx +++ b/packages/base/src/annotations/components/Annotation.tsx @@ -9,37 +9,25 @@ import { showDialog, Dialog } from '@jupyterlab/apputils'; import { Button } from '@jupyterlab/ui-components'; import React, { useMemo, useState } from 'react'; -import { IControlPanelModel } from '@/src/types'; import { Message } from './Message'; export interface IAnnotationProps { itemId: string; annotationModel: IAnnotationModel; - rightPanelModel?: IControlPanelModel; + jgisModel?: IJupyterGISModel; children?: JSX.Element[] | JSX.Element; } const Annotation: React.FC = ({ itemId, annotationModel, - rightPanelModel, + jgisModel, children, }) => { const [messageContent, setMessageContent] = useState(''); - const [jgisModel, setJgisModel] = useState( - rightPanelModel?.jGISModel, - ); - const annotation = annotationModel.getAnnotation(itemId); const contents = useMemo(() => annotation?.contents ?? [], [annotation]); - /** - * Update the model when it changes. - */ - rightPanelModel?.documentChanged.connect((_, widget) => { - setJgisModel(widget?.model); - }); - const handleSubmit = () => { annotationModel.addContent(itemId, messageContent); setMessageContent(''); @@ -98,7 +86,7 @@ const Annotation: React.FC = ({ - {rightPanelModel && ( + {jgisModel && (