from vitessce import VitessceConfig, AnnDataWrapper, Component as cm
from urllib.parse import quote_plus
import json

zarr_path = r"D:\data\h5\adata_processed.zarr"

vc = VitessceConfig(schema_version="1.0.15", name='Habib et al', description='COVID-19 Healthy Donor Brain')
dataset = vc.add_dataset(name='Brain').add_object(
    AnnDataWrapper(
        adata_path=zarr_path,
        obs_embedding_paths=["obsm/X_umap"],
        obs_embedding_names=["UMAP"],
        obs_set_paths=["obs/leiden"],
        obs_set_names=["Leiden"],
        obs_feature_matrix_path="X",
        feature_filter_path="var/highly_variable",
        obs_locations_path="obsm/spatial"
    ))

# 创建 spatial 视图
spatial = vc.add_view(cm.SPATIAL, dataset=dataset, props={"cellRadius": 100, "fill": "Leiden", "fillOpacity": 1})


scatterplot = vc.add_view(cm.SCATTERPLOT, dataset=dataset, mapping="UMAP", props={"xAxisColumn": "0", "yAxisColumn": "1"})
cell_sets = vc.add_view(cm.OBS_SETS, dataset=dataset)
genes = vc.add_view(cm.FEATURE_LIST, dataset=dataset)
heatmap = vc.add_view(cm.HEATMAP, dataset=dataset)
vc.layout((scatterplot | (cell_sets / genes)) / heatmap)
vc.layout(
    (spatial | (scatterplot | (cell_sets / genes))) / heatmap
)
config_dict = vc.export(to='files', base_url='http://localhost:3000', out_dir='./test')
vitessce_url = "http://vitessce.io/?url=data:," + quote_plus(json.dumps(config_dict))
import webbrowser

webbrowser.open(vitessce_url)