tschm commited on
Commit
84d8880
Β·
1 Parent(s): 3f47de8

copy from scribble

Browse files
.gitignore ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ .idea
2
+ __pycache__
3
+ .venv
4
+ .ruff_cache
5
+
6
+
7
+ *.dot
8
+ *.svg
9
+ .config
10
+ *.pdf
11
+ *.png
app.py CHANGED
@@ -1,469 +1,80 @@
1
  import marimo
2
 
3
- __generated_with = "0.9.2"
4
  app = marimo.App()
5
 
6
 
7
  @app.cell
8
- def __():
9
  import marimo as mo
10
 
11
- mo.md("# Welcome to marimo! πŸŒŠπŸƒ")
12
- return (mo,)
13
 
14
-
15
- @app.cell
16
- def __(mo):
17
- slider = mo.ui.slider(1, 22)
18
- return (slider,)
19
 
20
 
21
  @app.cell
22
- def __(mo, slider):
 
23
  mo.md(
24
  f"""
25
- marimo is a **reactive** Python notebook.
26
-
27
- This means that unlike traditional notebooks, marimo notebooks **run
28
- automatically** when you modify them or
29
- interact with UI elements, like this slider: {slider}.
30
-
31
- {"##" + "πŸƒ" * slider.value}
32
  """
33
  )
34
- return
35
-
36
-
37
- @app.cell(hide_code=True)
38
- def __(mo):
39
- mo.accordion(
40
- {
41
- "Tip: disabling automatic execution": mo.md(
42
- rf"""
43
- marimo lets you disable automatic execution: just go into the
44
- notebook settings and set
45
-
46
- "Runtime > On Cell Change" to "lazy".
47
-
48
- When the runtime is lazy, after running a cell, marimo marks its
49
- descendants as stale instead of automatically running them. The
50
- lazy runtime puts you in control over when cells are run, while
51
- still giving guarantees about the notebook state.
52
- """
53
- )
54
- }
55
- )
56
- return
57
-
58
-
59
- @app.cell(hide_code=True)
60
- def __(mo):
61
- mo.md(
62
- """
63
- Tip: This is a tutorial notebook. You can create your own notebooks
64
- by entering `marimo edit` at the command line.
65
- """
66
- ).callout()
67
- return
68
-
69
-
70
- @app.cell(hide_code=True)
71
- def __(mo):
72
- mo.md(
73
- """
74
- ## 1. Reactive execution
75
-
76
- A marimo notebook is made up of small blocks of Python code called
77
- cells.
78
-
79
- marimo reads your cells and models the dependencies among them: whenever
80
- a cell that defines a global variable is run, marimo
81
- **automatically runs** all cells that reference that variable.
82
-
83
- Reactivity keeps your program state and outputs in sync with your code,
84
- making for a dynamic programming environment that prevents bugs before they
85
- happen.
86
- """
87
- )
88
- return
89
-
90
-
91
- @app.cell(hide_code=True)
92
- def __(changed, mo):
93
- (
94
- mo.md(
95
- f"""
96
- **✨ Nice!** The value of `changed` is now {changed}.
97
-
98
- When you updated the value of the variable `changed`, marimo
99
- **reacted** by running this cell automatically, because this cell
100
- references the global variable `changed`.
101
-
102
- Reactivity ensures that your notebook state is always
103
- consistent, which is crucial for doing good science; it's also what
104
- enables marimo notebooks to double as tools and apps.
105
- """
106
- )
107
- if changed
108
- else mo.md(
109
- """
110
- **🌊 See it in action.** In the next cell, change the value of the
111
- variable `changed` to `True`, then click the run button.
112
- """
113
- )
114
- )
115
- return
116
 
117
 
118
  @app.cell
119
- def __():
120
- changed = False
121
- return (changed,)
122
-
123
-
124
- @app.cell(hide_code=True)
125
- def __(mo):
126
- mo.accordion(
127
- {
128
- "Tip: execution order": (
129
- """
130
- The order of cells on the page has no bearing on
131
- the order in which cells are executed: marimo knows that a cell
132
- reading a variable must run after the cell that defines it. This
133
- frees you to organize your code in the way that makes the most
134
- sense for you.
135
- """
136
- )
137
- }
138
- )
139
- return
140
-
141
-
142
- @app.cell(hide_code=True)
143
- def __(mo):
144
- mo.md(
145
- """
146
- **Global names must be unique.** To enable reactivity, marimo imposes a
147
- constraint on how names appear in cells: no two cells may define the same
148
- variable.
149
- """
150
- )
151
- return
152
-
153
-
154
- @app.cell(hide_code=True)
155
- def __(mo):
156
- mo.accordion(
157
- {
158
- "Tip: encapsulation": (
159
- """
160
- By encapsulating logic in functions, classes, or Python modules,
161
- you can minimize the number of global variables in your notebook.
162
- """
163
- )
164
- }
165
- )
166
- return
167
-
168
-
169
- @app.cell(hide_code=True)
170
- def __(mo):
171
- mo.accordion(
172
- {
173
- "Tip: private variables": (
174
- """
175
- Variables prefixed with an underscore are "private" to a cell, so
176
- they can be defined by multiple cells.
177
- """
178
- )
179
- }
180
- )
181
- return
182
-
183
-
184
- @app.cell(hide_code=True)
185
- def __(mo):
186
  mo.md(
187
- """
188
- ## 2. UI elements
189
-
190
- Cells can output interactive UI elements. Interacting with a UI
191
- element **automatically triggers notebook execution**: when
192
- you interact with a UI element, its value is sent back to Python, and
193
- every cell that references that element is re-run.
194
-
195
- marimo provides a library of UI elements to choose from under
196
- `marimo.ui`.
197
  """
198
  )
199
- return
200
-
201
-
202
- @app.cell
203
- def __(mo):
204
- mo.md("""**🌊 Some UI elements.** Try interacting with the below elements.""")
205
- return
206
-
207
-
208
- @app.cell
209
- def __(mo):
210
- icon = mo.ui.dropdown(["πŸƒ", "🌊", "✨"], value="πŸƒ")
211
- return (icon,)
212
-
213
-
214
- @app.cell
215
- def __(icon, mo):
216
- repetitions = mo.ui.slider(1, 16, label=f"number of {icon.value}: ")
217
- return (repetitions,)
218
-
219
-
220
- @app.cell
221
- def __(icon, repetitions):
222
- icon, repetitions
223
- return
224
 
225
 
226
  @app.cell
227
- def __(icon, mo, repetitions):
228
- mo.md("# " + icon.value * repetitions.value)
229
- return
230
-
231
-
232
- @app.cell(hide_code=True)
233
- def __(mo):
234
- mo.md(
235
- """
236
- ## 3. marimo is just Python
237
-
238
- marimo cells parse Python (and only Python), and marimo notebooks are
239
- stored as pure Python files β€” outputs are _not_ included. There's no
240
- magical syntax.
241
-
242
- The Python files generated by marimo are:
243
-
244
- - easily versioned with git, yielding minimal diffs
245
- - legible for both humans and machines
246
- - formattable using your tool of choice,
247
- - usable as Python scripts, with UI elements taking their default
248
- values, and
249
- - importable by other modules (more on that in the future).
250
- """
251
- )
252
- return
253
-
254
-
255
- @app.cell(hide_code=True)
256
- def __(mo):
257
  mo.md(
258
- """
259
- ## 4. Running notebooks as apps
260
-
261
- marimo notebooks can double as apps. Click the app window icon in the
262
- bottom-right to see this notebook in "app view."
263
-
264
- Serve a notebook as an app with `marimo run` at the command-line.
265
- Of course, you can use marimo just to level-up your
266
- notebooking, without ever making apps.
267
  """
268
  )
269
- return
270
-
271
-
272
- @app.cell(hide_code=True)
273
- def __(mo):
274
- mo.md(
275
- """
276
- ## 5. The `marimo` command-line tool
277
-
278
- **Creating and editing notebooks.** Use
279
-
280
- ```
281
- marimo edit
282
- ```
283
-
284
- in a terminal to start the marimo notebook server. From here
285
- you can create a new notebook or edit existing ones.
286
-
287
-
288
- **Running as apps.** Use
289
-
290
- ```
291
- marimo run notebook.py
292
- ```
293
 
294
- to start a webserver that serves your notebook as an app in read-only mode,
295
- with code cells hidden.
296
 
297
- **Convert a Jupyter notebook.** Convert a Jupyter notebook to a marimo
298
- notebook using `marimo convert`:
299
-
300
- ```
301
- marimo convert your_notebook.ipynb > your_app.py
302
- ```
303
-
304
- **Tutorials.** marimo comes packaged with tutorials:
305
-
306
- - `dataflow`: more on marimo's automatic execution
307
- - `ui`: how to use UI elements
308
- - `markdown`: how to write markdown, with interpolated values and
309
- LaTeX
310
- - `plots`: how plotting works in marimo
311
- - `sql`: how to use SQL
312
- - `layout`: layout elements in marimo
313
- - `fileformat`: how marimo's file format works
314
- - `markdown-format`: for using `.md` files in marimo
315
- - `for-jupyter-users`: if you are coming from Jupyter
316
 
317
- Start a tutorial with `marimo tutorial`; for example,
318
 
319
- ```
320
- marimo tutorial dataflow
321
- ```
322
 
323
- In addition to tutorials, we have examples in our
324
- [our GitHub repo](https://www.github.com/marimo-team/marimo/tree/main/examples).
325
- """
 
 
 
326
  )
327
- return
328
-
329
 
330
- @app.cell(hide_code=True)
331
- def __(mo):
332
  mo.md(
333
- """
334
- ## 6. The marimo editor
335
-
336
- Here are some tips to help you get started with the marimo editor.
337
- """
338
- )
339
- return
340
-
341
-
342
- @app.cell
343
- def __(mo, tips):
344
- mo.accordion(tips)
345
- return
346
-
347
-
348
- @app.cell(hide_code=True)
349
- def __(mo):
350
- mo.md("""## Finally, a fun fact""")
351
- return
352
-
353
 
354
- @app.cell(hide_code=True)
355
- def __(mo):
356
- mo.md(
357
- """
358
- The name "marimo" is a reference to a type of algae that, under
359
- the right conditions, clumps together to form a small sphere
360
- called a "marimo moss ball". Made of just strands of algae, these
361
- beloved assemblages are greater than the sum of their parts.
362
- """
363
  )
364
- return
365
-
366
-
367
- @app.cell(hide_code=True)
368
- def __():
369
- tips = {
370
- "Saving": (
371
- """
372
- **Saving**
373
-
374
- - _Name_ your app using the box at the top of the screen, or
375
- with `Ctrl/Cmd+s`. You can also create a named app at the
376
- command line, e.g., `marimo edit app_name.py`.
377
-
378
- - _Save_ by clicking the save icon on the bottom right, or by
379
- inputting `Ctrl/Cmd+s`. By default marimo is configured
380
- to autosave.
381
- """
382
- ),
383
- "Running": (
384
- """
385
- 1. _Run a cell_ by clicking the play ( β–· ) button on the top
386
- right of a cell, or by inputting `Ctrl/Cmd+Enter`.
387
-
388
- 2. _Run a stale cell_ by clicking the yellow run button on the
389
- right of the cell, or by inputting `Ctrl/Cmd+Enter`. A cell is
390
- stale when its code has been modified but not run.
391
-
392
- 3. _Run all stale cells_ by clicking the play ( β–· ) button on
393
- the bottom right of the screen, or input `Ctrl/Cmd+Shift+r`.
394
- """
395
- ),
396
- "Console Output": (
397
- """
398
- Console output (e.g., `print()` statements) is shown below a
399
- cell.
400
- """
401
- ),
402
- "Creating, Moving, and Deleting Cells": (
403
- """
404
- 1. _Create_ a new cell above or below a given one by clicking
405
- the plus button to the left of the cell, which appears on
406
- mouse hover.
407
-
408
- 2. _Move_ a cell up or down by dragging on the handle to the
409
- right of the cell, which appears on mouse hover.
410
-
411
- 3. _Delete_ a cell by clicking the trash bin icon. Bring it
412
- back by clicking the undo button on the bottom right of the
413
- screen, or with `Ctrl/Cmd+Shift+z`.
414
- """
415
- ),
416
- "Disabling Automatic Execution": (
417
- """
418
- Via the notebook settings (gear icon) or footer panel, you
419
- can disable automatic execution. This is helpful when
420
- working with expensive notebooks or notebooks that have
421
- side-effects like database transactions.
422
- """
423
- ),
424
- "Disabling Cells": (
425
- """
426
- You can disable a cell via the cell context menu.
427
- marimo will never run a disabled cell or any cells that depend on it.
428
- This can help prevent accidental execution of expensive computations
429
- when editing a notebook.
430
- """
431
- ),
432
- "Code Folding": (
433
- """
434
- You can collapse or fold the code in a cell by clicking the arrow
435
- icons in the line number column to the left, or by using keyboard
436
- shortcuts.
437
 
438
- Use the command palette (`Ctrl/Cmd+k`) or a keyboard shortcut to
439
- quickly fold or unfold all cells.
440
- """
441
- ),
442
- "Code Formatting": (
443
- """
444
- If you have [ruff](https://github.com/astral-sh/ruff) installed,
445
- you can format a cell with the keyboard shortcut `Ctrl/Cmd+b`.
446
- """
447
- ),
448
- "Command Palette": (
449
- """
450
- Use `Ctrl/Cmd+k` to open the command palette.
451
- """
452
- ),
453
- "Keyboard Shortcuts": (
454
- """
455
- Open the notebook menu (top-right) or input `Ctrl/Cmd+Shift+h` to
456
- view a list of all keyboard shortcuts.
457
- """
458
- ),
459
- "Configuration": (
460
- """
461
- Configure the editor by clicking the gears icon near the top-right
462
- of the screen.
463
- """
464
- ),
465
- }
466
- return (tips,)
467
 
468
 
469
  if __name__ == "__main__":
 
1
  import marimo
2
 
3
+ __generated_with = "0.9.30"
4
  app = marimo.App()
5
 
6
 
7
  @app.cell
8
+ def __init():
9
  import marimo as mo
10
 
11
+ from pyscribble import create
 
12
 
13
+ return create, mo
 
 
 
 
14
 
15
 
16
  @app.cell
17
+ def __input_name(mo):
18
+ name = mo.ui.text(placeholder="Name...")
19
  mo.md(
20
  f"""
21
+ Enter the name of the guest: {name}
 
 
 
 
 
 
22
  """
23
  )
24
+ return (name,)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
25
 
26
 
27
  @app.cell
28
+ def __input_function(mo):
29
+ options = ["tanh((-1+2j)*z)", "sinh(3*z)", "exp((-1+2j)*z)"]
30
+ dropdown = mo.ui.dropdown(options=options, value="sinh(3*z)")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
31
  mo.md(
32
+ f"""
33
+ Enter the complex function: {dropdown}
 
 
 
 
 
 
 
 
34
  """
35
  )
36
+ return dropdown, options
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
37
 
38
 
39
  @app.cell
40
+ def __input_event(mo):
41
+ event = mo.ui.text(placeholder="Event...")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
42
  mo.md(
43
+ f"""
44
+ Enter the name of the event: {event}
 
 
 
 
 
 
 
45
  """
46
  )
47
+ return (event,)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
48
 
 
 
49
 
50
+ @app.cell
51
+ def __output(create, dropdown, event, mo, name):
52
+ from io import BytesIO
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
53
 
54
+ fig = create(name=name.value, fct=dropdown.value, event=event.value, n=100)
55
 
56
+ img = fig.to_image(format="png")
57
+ # print(img)
58
+ data = BytesIO(img)
59
 
60
+ # Create a download button for the Plotly graph
61
+ download_btn = mo.download(
62
+ data=data,
63
+ filename=f"{name.value}_{event.value}_plot.png",
64
+ label="Download",
65
+ mimetype="image/png",
66
  )
 
 
67
 
68
+ # Display the plot and download button
 
69
  mo.md(
70
+ f"""
71
+ {download_btn}
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
72
 
73
+ {mo.ui.plotly(fig)}
74
+ """
 
 
 
 
 
 
 
75
  )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
76
 
77
+ # return fig, download_btn
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
78
 
79
 
80
  if __name__ == "__main__":
pyscribble/__init__.py ADDED
@@ -0,0 +1 @@
 
 
1
+ from ._plot import create
pyscribble/_letter.py ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import numpy as np
2
+
3
+
4
+ def letter(x):
5
+ # I have copied this data straight from
6
+ # https://github.com/asgeirbirkis/chebfun/blob/master/scribble.m
7
+ __letters = {
8
+ "A": [0, 0.4 + 1j, 0.8, 0.6 + 0.5j, 0.2 + 0.5j],
9
+ "B": [0, 1j, 0.8 + 0.9j, 0.8 + 0.6j, 0.5j, 0.8 + 0.4j, 0.8 + 0.1j, 0],
10
+ "C": [0.8 + 1j, 0.8j, 0.2j, 0.8],
11
+ "D": [0, 0.8 + 0.1j, 0.8 + 0.9j, 1j, 0],
12
+ "E": [0.8 + 1j, 1j, 0.5j, 0.5j + 0.7, 0.5j, 0, 0.8],
13
+ "F": [0.8 + 1j, 1j, 0.5j, 0.5j + 0.7, 0.5j, 0],
14
+ "G": [0.8 + 1j, 0.8j, 0.2j, 0.6, 0.6 + 0.5j, 0.4 + 0.5j, 0.8 + 0.5j],
15
+ "H": [0, 1j, 0.5j, 0.5j + 0.8, 0.8 + 1j, 0.8],
16
+ "I": [0, 0.8, 0.4, 0.4 + 1j, 1j, 0.8 + 1j],
17
+ "J": [0, 0.4, 0.4 + 1j, 1j, 0.8 + 1j],
18
+ "K": [0, 1j, 0.5j, 0.8 + 1j, 0.5j, 0.8],
19
+ "L": [1j, 0, 0.8],
20
+ "M": [0, 0.1 + 1j, 0.4, 0.7 + 1j, 0.8],
21
+ "N": [0, 1j, 0.8, 0.8 + 1j],
22
+ "O": [0, 1j, 0.8 + 1j, 0.8, 0],
23
+ "Q": [0, 1j, 0.8 + 1j, 0.8, 0.6 + 0.2j, 0.9 - 0.1j, 0.8, 0],
24
+ "P": [0, 1j, 0.8 + 1j, 0.8 + 0.5j, 0.5j],
25
+ "R": [0, 1j, 0.8 + 1j, 0.8 + 0.6j, 0.5j, 0.8],
26
+ "S": [0.8 + 1j, 0.9j, 0.6j, 0.8 + 0.4j, 0.8 + 0.1j, 0],
27
+ "T": [0.4, 0.4 + 1j, 1j, 0.8 + 1j],
28
+ "U": [1j, 0.1, 0.7, 0.8 + 1j],
29
+ "V": [1j, 0.4, 0.8 + 1j],
30
+ "W": [1j, 0.2, 0.4 + 1j, 0.6, 0.8 + 1j],
31
+ "X": [1j, 0.8, 0.4 + 0.5j, 0.8 + 1j, 0],
32
+ "Y": [1j, 0.4 + 0.5j, 0.8 + 1j, 0.4 + 0.5j, 0.4],
33
+ "Z": [1j, 0.8 + 1j, 0, 0.8],
34
+ " ": [],
35
+ }
36
+
37
+ return np.array(__letters[x.upper()])
pyscribble/_plot.py ADDED
@@ -0,0 +1,114 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import plotly.graph_objs as go
2
+ from plotly.subplots import make_subplots
3
+
4
+ from ._scribble import series
5
+
6
+
7
+ # Create the figure with subplots: 2 rows, 1 column
8
+ def __create_fig():
9
+ fig = make_subplots(
10
+ rows=2,
11
+ cols=1,
12
+ row_heights=[0.5, 0.5], # Top half and bottom half should be equal
13
+ vertical_spacing=0.1, # Some spacing between the top and bottom sections
14
+ subplot_titles=["", ""], # No title for the top subplot
15
+ )
16
+ return fig
17
+
18
+
19
+ def __create_annotation(fig, word):
20
+ # Add the upside-down word as an annotation in the top subplot
21
+ fig.add_annotation(
22
+ x=0.5, # X position (center)
23
+ y=1, # Y position at the top of the subplot
24
+ text=word, # The word to display
25
+ showarrow=False, # No arrow needed
26
+ font=dict(size=20, color="blue"), # Text styling: large, blue font
27
+ textangle=180, # Rotate text by 180 degrees (upside down)
28
+ align="center", # Center alignment
29
+ valign="middle", # Vertical alignment (centered)
30
+ row=1,
31
+ col=1, # Place in the top subplot
32
+ )
33
+
34
+
35
+ def __remove_axis(fig):
36
+ # Update layout settings
37
+ fig.update_layout(
38
+ plot_bgcolor="white", # White background for clean look
39
+ paper_bgcolor="white", # White paper background
40
+ showlegend=False, # No legend needed
41
+ margin=dict(l=30, r=30, t=30, b=30), # Margins around the plot
42
+ )
43
+
44
+ # Update axes settings for the top subplot (hide grid, ticks, and lines)
45
+ fig.update_xaxes(
46
+ row=1,
47
+ col=1,
48
+ showgrid=False,
49
+ zeroline=False,
50
+ showticklabels=False,
51
+ showline=False,
52
+ )
53
+ fig.update_yaxes(
54
+ row=1,
55
+ col=1,
56
+ showgrid=False,
57
+ zeroline=False,
58
+ showticklabels=False,
59
+ showline=False,
60
+ )
61
+
62
+ # Update axes settings for the bottom subplot (show gridlines and labels)
63
+ fig.update_xaxes(
64
+ row=2,
65
+ col=1,
66
+ showgrid=True,
67
+ zeroline=False,
68
+ showticklabels=False,
69
+ showline=False,
70
+ )
71
+ fig.update_yaxes(
72
+ row=2,
73
+ col=1,
74
+ showgrid=True,
75
+ zeroline=False,
76
+ showticklabels=False,
77
+ showline=False,
78
+ )
79
+ return fig
80
+
81
+
82
+ def __plot_letters(fig, d):
83
+ # Create traces for each segment
84
+ traces = []
85
+
86
+ for segment in d:
87
+ trace = go.Scatter(
88
+ x=segment.real, # Real part of the complex number
89
+ y=segment.imag, # Imaginary part of the complex number
90
+ mode="markers+lines", # Display both markers and lines
91
+ showlegend=False,
92
+ line={"width": 3, "color": "blue"}, # Line style
93
+ marker={"size": 3, "color": "blue"}, # Marker style
94
+ )
95
+ traces.append(trace)
96
+
97
+ # Add all traces to the bottom subplot
98
+ for trace in traces:
99
+ fig.add_trace(trace, row=2, col=1)
100
+
101
+
102
+ def create(name, fct, event, n=100):
103
+ # Create the figure with subplots: 2 rows, 1 column
104
+ fig = __create_fig()
105
+ __create_annotation(fig, word=f"{name}<br>{fct}<br>{event}")
106
+
107
+ segments = list(series(name, n=n, str=fct))
108
+
109
+ # d is now a list of list. Flatten it
110
+ # d = list(chain.from_iterable(d))
111
+
112
+ __plot_letters(fig, segments)
113
+ __remove_axis(fig)
114
+ return fig
pyscribble/_scribble.py ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from itertools import chain
2
+
3
+ import numexpr as ne
4
+ import numpy as np
5
+
6
+ from ._letter import letter
7
+
8
+
9
+ def series(string, n, str):
10
+ segments = []
11
+ for i, _letter in enumerate(string):
12
+ # move pts to the correction position in a word
13
+ pts = letter(_letter) + i
14
+
15
+ # move pts to unit square
16
+ pts = 2 * pts / len(string) - 1
17
+
18
+ segments.append(ne.evaluate(str, local_dict={"z": list(__segment(pts, n=n))}))
19
+
20
+ # segments is a list of list, flatten it
21
+ return list(chain.from_iterable(segments))
22
+
23
+
24
+ def __segment(points, n=100):
25
+ """
26
+ Each letter is a represented by a bunch of points a,b,c,d...
27
+ There are straight segments between two adjacent points
28
+ We represent each such segment as a collection of n auxilliary points
29
+ """
30
+ for a, b in zip(points[:-1], points[1:]):
31
+ yield np.linspace(a.real, b.real, n) + 1j * np.linspace(a.imag, b.imag, n)
requirements.txt CHANGED
@@ -1,5 +1,4 @@
1
- marimo
2
- # Or a specific version
3
- # marimo>=0.9.0
4
-
5
- # Add other dependencies as needed
 
1
+ numpy==2.1.3
2
+ plotly==5.24.1
3
+ numexpr==2.10.2
4
+ kaleido