Updated

app.py

@@ -3,57 +3,40 @@ import numpy as np
 import google.generativeai as genai
 import os
 import streamlit as st
-from PIL import Image, ImageDraw
+from PIL import Image
+from streamlit_drawable_canvas import st_canvas
 
 # Set up environment variables and configurations
 genai.configure(api_key=os.environ['GOOGLE_API_KEY'])
 
-# Initialize session state for drawing
-if 'points' not in st.session_state:
-    st.session_state.points = []
-
-# Function to draw on the canvas
-def draw_on_image(image, points):
-    draw = ImageDraw.Draw(image)
-    for stroke in points:
-        if len(stroke) > 1:
-            draw.line(stroke, fill=(255, 255, 255), width=5)
-
 def main():
     st.title("Virtual Math Calculator")
 
     col1, col2 = st.columns([3, 1])
 
     with col1:
-        # Create a canvas for drawing
-        canvas_image = Image.new("RGB", (800, 600), (0, 0, 0))
-
-        # Display the canvas
-        canvas_placeholder = st.empty()
-
-        # Instructions and buttons
+        # Create a drawing canvas
         st.write("**Draw your equation below:**")
-        if st.button("Reset"):
-            st.session_state.points = []
-        if st.button("Submit"):
-            draw_on_image(canvas_image, st.session_state.points)
-            canvas_image.save('drawing.png')
-
-            # Send the image to Gemini and get the response
-            response = send_to_gemini('drawing.png')
-            st.session_state.result = response
+        canvas_result = st_canvas(
+            fill_color="rgb(255, 255, 255)",  # Fill color for drawing
+            stroke_width=5,
+            stroke_color="rgb(0, 0, 0)",
+            background_color="rgb(255, 255, 255)",
+            height=300,
+            width=600,
+            drawing_mode="freedraw",
+            key="canvas",
+        )
 
-        # Interactive canvas for drawing
-        st.write("**Use your mouse to draw:**")
-        mouse_down = st.button("Hold to Draw")
-        if mouse_down:
-            st.session_state.points.append(st.mouse_position())
-
-        # Draw on the image
-        draw_on_image(canvas_image, st.session_state.points)
+        if st.button("Submit"):
+            if canvas_result.image_data is not None:
+                # Convert to image
+                drawing_image = Image.fromarray(canvas_result.image_data.astype('uint8'), 'RGBA').convert('RGB')
+                drawing_image.save('drawing.png')
 
-        # Display the updated image
-        canvas_placeholder.image(np.array(canvas_image))
+                # Send the image to Gemini and get the response
+                response = send_to_gemini('drawing.png')
+                st.session_state.result = response
 
     with col2:
         st.header("Instructions")