Spaces:
Runtime error
Runtime error
File size: 4,211 Bytes
16e9d92 c39c430 16e9d92 c39c430 16e9d92 c39c430 fc17971 c39c430 16e9d92 c39c430 16e9d92 c39c430 16e9d92 c39c430 16e9d92 c39c430 16e9d92 fc17971 16e9d92 |
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 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 |
{
"cells": [
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"import bpy\n",
"from IPython.display import display\n",
"from anywidget_3dviewer import Viewer\n",
"import solara\n",
"from pathlib import Path\n",
"import molecularnodes as mn\n",
"\n",
"\n",
"\n",
"\n",
"def make_gltf_model():\n",
" bpy.ops.preferences.addon_enable(module=\"io_scene_gltf2\")\n",
"\n",
" # Delete all mesh objects from the scene\n",
" bpy.ops.object.select_all(action=\"DESELECT\")\n",
" bpy.ops.object.select_by_type(type=\"MESH\")\n",
" bpy.ops.object.delete()\n",
"\n",
" obj = mn.load.molecule_rcsb(\"1YAR\", starting_style=\"cartoon\", center_molecule=True)\n",
" obj.scale = [10, 10, 10]\n",
"\n",
" bpy.context.view_layer.objects.active = obj\n",
" obj.modifiers[\"MolecularNodes\"].node_group.nodes[\"MN_color_attribute_random\"].inputs[\"Lightness\"].default_value = 0.5\n",
" obj.modifiers[\"MolecularNodes\"].node_group.nodes[\"MN_color_attribute_random\"].inputs[\"Saturation\"].default_value = 1\n",
" obj.modifiers[\"MolecularNodes\"].node_group.nodes[\"MN_color_attribute_random\"].inputs[\"Seed\"].default_value = 1\n",
"\n",
" for modifier in obj.modifiers:\n",
" bpy.ops.object.modifier_apply(modifier=modifier.name)\n",
" bpy.ops.geometry.color_attribute_render_set(name=\"Color\")\n",
"\n",
"\n",
" # Exporting the model to .glb format for Three.js\n",
" # Specify the path where you want to save the exported model\n",
" gltf_path = Path() / \"_MY_MODEL_DATA.gltf\"\n",
"\n",
" bpy.ops.geometry.color_attribute_render_set(name=\"Color\")\n",
"\n",
" # Select only the object\n",
" bpy.ops.object.select_all(action=\"DESELECT\")\n",
" obj.select_set(True)\n",
"\n",
" # Export the selected obj object to a .glb file\n",
" bpy.ops.export_scene.gltf(\n",
" filepath=str(gltf_path), export_format=\"GLTF_EMBEDDED\", use_selection=True\n",
" )\n",
"\n",
" data = gltf_path.read_text()\n",
"\n",
" gltf_path.unlink()\n",
"\n",
" return data\n",
"\n",
"\n",
"data = make_gltf_model()\n",
" \n",
"\n",
"do_render = solara.reactive(False)\n",
"\n",
"data = make_gltf_model()\n",
"\n",
"@solara.component\n",
"def Page():\n",
"\n",
" def render():\n",
" if do_render.value:\n",
" return data\n",
" \n",
" result = solara.use_thread(render, [do_render.value])\n",
" if not do_render.value:\n",
" solara.Button(\"Start Rendering\", on_click=lambda: do_render.set(True), color=\"orange\")\n",
" else:\n",
" if result.state == solara.ResultState.RUNNING:\n",
" solara.Info(\"Rendering in progress...\")\n",
" solara.ProgressLinear()\n",
" elif result.state == solara.ResultState.ERROR:\n",
" solara.Error(\"Rendering failed!: %s\" % result.error)\n",
" elif result.state == solara.ResultState.FINISHED:\n",
" with solara.Column():\n",
" solara.Success(\"Rendering complete. \")\n",
" if result.value:\n",
" viewer = Viewer(gltf_data=result.value)\n",
" solara.FileDownload(data, filename=\"model_cube.gltf\", label=\"Download model.gltf file\")\n",
"\n",
" display(viewer)\n",
"Page()"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {
"kernelspec": {
"display_name": ".venv",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.10.13"
}
},
"nbformat": 4,
"nbformat_minor": 2
}
|