Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
+from PIL import Image
 import numpy as np
-from PIL import ImageDraw, Image
 
 def main():
     st.title("Streamlit Whiteboard for Note-Taking")
@@ -10,43 +10,46 @@ def main():
 
     # Create a canvas as a numpy array
     canvas = np.ones((400, 800, 3), dtype=np.uint8) * 255
-    image_location = st.image(canvas, caption="Draw Here", use_column_width=True)
 
-    # Get the drawing context
-    draw = ImageDraw.Draw(image_location)
+    # Display the canvas
+    st.image(canvas, caption="Draw Here", use_column_width=True)
 
-    # Track whether the mouse is currently clicked
-    mouse_clicked = False
+    # Initialize drawing variables
+    drawing = False
+    previous_point = None
 
-    # Track the position of the last mouse click
-    last_position = None
+    # Handle mouse events
+    mouse_event = st._get_last_mouse_event()
 
-    # Function to handle mouse events
-    def handle_mouse_event(event, x, y, _):
-        nonlocal mouse_clicked, last_position
+    if mouse_event is not None:
+        event, x, y = mouse_event
         if event == "mousedown":
-            mouse_clicked = True
-            last_position = (x, y)
-        elif event == "mousemove" and mouse_clicked:
-            draw.line([last_position, (x, y)], fill=(0, 0, 0), width=5)
-            last_position = (x, y)
-            st.image(canvas)
+            drawing = True
+            previous_point = (x, y)
+        elif event == "mousemove" and drawing:
+            # Draw a line from the previous point to the current point
+            canvas = draw_line(canvas, previous_point, (x, y))
+            previous_point = (x, y)
+            st.image(canvas, caption="Draw Here", use_column_width=True)
         elif event == "mouseup":
-            mouse_clicked = False
-
-    # Set up event handling
-    image_location.mousemove(handle_mouse_event, True)
-    image_location.mousedown(handle_mouse_event, True, "left")
-    image_location.mouseup(handle_mouse_event, True, "left")
+            drawing = False
 
     # Buttons for interaction
     if st.button("Clear Canvas"):
         canvas[:] = 255
-        st.image(canvas)
+        st.image(canvas, caption="Draw Here", use_column_width=True)
 
     if st.button("Save Notes"):
         save_notes(note)
 
+def draw_line(canvas, start_point, end_point):
+    """
+    Draw a line on the canvas between two points.
+    """
+    draw = Image.fromarray(canvas)
+    draw.line([start_point, end_point], fill=(0, 0, 0), width=5)
+    return np.array(draw)
+
 def save_notes(note):
     with open("notes.txt", "w") as file:
         file.write(note)