tweak

app.py

@@ -1,112 +1,12 @@
 import gradio as gr
-import bpy
-from tqdm import tqdm
-from math import pi
-import tempfile
-import molecularnodes as mn
-import os
 
-def enable_GPUS():
-    bpy.data.scenes[0].render.engine = "CYCLES" #"CYCLES"
-    # Set the device_type
-    bpy.context.preferences.addons[
-        "cycles"
-    ].preferences.compute_device_type = "CUDA"  # or "OPENCL"
+def greet(name, intensity):
+    return "Hello22, " + name + "!" * int(intensity)
 
-    # Set the device and feature set
-    bpy.context.scene.cycles.device = "GPU"
+demo = gr.Interface(
+    fn=greet,
+    inputs=["text", "slider"],
+    outputs=["text"],
+)
 
-    for scene in bpy.data.scenes:
-        scene.cycles.device = "GPU"
-
-    bpy.context.preferences.addons["cycles"].preferences.get_devices()
-    print(bpy.context.preferences.addons["cycles"].preferences.compute_device_type)
-    for d in bpy.context.preferences.addons["cycles"].preferences.devices:
-        d["use"] = True  # Using all devices, include GPU and CPU
-        print(d["name"])
-
-
-enable_GPUS()
-
-window = bpy.context.window
-screen = window.screen
-
-style = 'cartoon'
-
-nodes_to_append = ["MN_color_set", 
-                   "MN_color_common", 
-                   "MN_color_attribute_random",
-                   mn.nodes.styles_mapping[style]]
-
-def get_areas(type):
-    return [area for area in screen.areas if area.type == type]
-
-def get_regions(areas):
-    return [region for region in areas[0].regions if region.type == 'WINDOW']
-
-for node in nodes_to_append:
-    bpy.ops.wm.append(
-                'INVOKE_DEFAULT',
-                directory = os.path.join(mn.nodes.mn_data_file, 'NodeTree'), 
-                filename = node, 
-                link = False
-            )
-
-def generate(progress=gr.Progress(track_tqdm=True)):
-    area_type = 'VIEW_3D' 
-    areas  = get_areas(area_type)
-    with bpy.context.temp_override(window=window, area=areas[0], region=get_regions(areas)[0], screen=screen):
-        
-        
-        for obj in bpy.context.scene.objects:
-            if obj.type == 'MESH':
-                bpy.data.objects.remove(obj, do_unlink=True)
-
-        molecule = mn.load.molecule_rcsb("7TYG", starting_style=style, center_molecule=True)
-        molecule.select_set(True)
-        bpy.context.view_layer.objects.active = molecule
-
-        bpy.ops.view3d.camera_to_view_selected()
-        camera = bpy.data.objects["Camera"]
-        camera.data.dof.use_dof = True
-        camera.data.dof.focus_distance = 5
-        camera.data.dof.aperture_fstop = 4
-        camera.data.angle = pi / 3
-        camera.data.type = "PERSP"
-
-        with tempfile.NamedTemporaryFile(suffix=".JPEG", delete=False) as f:
-            bpy.context.scene.render.resolution_y = 400
-            bpy.context.scene.render.resolution_x = 600
-            bpy.context.scene.render.image_settings.file_format = "JPEG"
-            bpy.context.scene.render.filepath = f.name
-
-            with tqdm() as pbar:
-
-                def elapsed(dummy):
-                    pbar.update()
-
-                bpy.app.handlers.render_stats.append(elapsed)
-                bpy.context.scene.frame_set(1)
-                bpy.context.scene.frame_current = 1
-                bpy.ops.render.render(animation=False, write_still=True)
-
-                bpy.data.images["Render Result"].save_render(
-                    filepath=bpy.context.scene.render.filepath
-                )
-                bpy.app.handlers.render_stats.clear()
-                return f.name
-
-with gr.Blocks() as demo:
-    with gr.Row():
-        with gr.Column():
-            render_btn = gr.Button("Render")
-        with gr.Column(scale=3):
-            image = gr.Image(type="filepath")
-
-    render_btn.click(
-        generate,
-        outputs=[image],
-    )
-
-demo.queue(concurrency_count=1)
-demo.launch(debug=True, inline=True)
+demo.launch()

hi.ipynb

@@ -0,0 +1,90 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/Users/jan-hendrik/projects/bpy-nvidia-T4/.venv/lib/python3.11/site-packages/tqdm/auto.py:21: TqdmWarning: IProgress not found. Please update jupyter and ipywidgets. See https://ipywidgets.readthedocs.io/en/stable/user_install.html\n",
+      "  from .autonotebook import tqdm as notebook_tqdm\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Running on local URL:  http://127.0.0.1:7860\n",
+      "\n",
+      "To create a public link, set `share=True` in `launch()`.\n"
+     ]
+    },
+    {
+     "data": {
+      "text/html": [
+       "<div><iframe src=\"http://127.0.0.1:7860/\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
+      ],
+      "text/plain": [
+       "<IPython.core.display.HTML object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "text/plain": []
+     },
+     "execution_count": 1,
+     "metadata": {},
+     "output_type": "execute_result"
+    }
+   ],
+   "source": [
+    "import gradio as gr\n",
+    "\n",
+    "def greet(name, intensity):\n",
+    "    return \"Hello, \" + name + \"!\" * int(intensity)\n",
+    "\n",
+    "demo = gr.Interface(\n",
+    "    fn=greet,\n",
+    "    inputs=[\"text\", \"slider\"],\n",
+    "    outputs=[\"text\"],\n",
+    ")\n",
+    "\n",
+    "demo.launch()"
+   ]
+  },
+  {
+   "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.11.10"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

requirements.txt

@@ -1,2 +1 @@
-bpy
-molecularnodes
+bpy