Spaces:
Sleeping
Sleeping
import streamlit as st | |
import pandas as pd | |
from PIL import Image | |
from streamlit_drawable_canvas import st_canvas | |
from helper import load_background_image, display_canvas_data | |
# Specify canvas parameters in application | |
drawing_mode = st.sidebar.selectbox( | |
"Drawing tool:", ("point", "freedraw", "line", "rect", "circle", "transform") | |
) | |
stroke_width = st.sidebar.slider("Stroke width: ", 1, 25, 3) | |
if drawing_mode == 'point': | |
point_display_radius = st.sidebar.slider("Point display radius: ", 1, 25, 3) | |
stroke_color = st.sidebar.color_picker("Stroke color hex: ") | |
bg_color = st.sidebar.color_picker("Background color hex: ", "#eee") | |
bg_image = st.sidebar.file_uploader("Background image:", type=["png", "jpg"]) | |
realtime_update = st.sidebar.checkbox("Update in realtime", True) | |
# Load background image if provided | |
background_image = load_background_image(bg_image) | |
# Create a canvas component | |
canvas_result = st_canvas( | |
fill_color="rgba(255, 165, 0, 0.3)", # Fixed fill color with some opacity | |
stroke_width=stroke_width, | |
stroke_color=stroke_color, | |
background_color=bg_color, | |
background_image=background_image, | |
update_streamlit=realtime_update, | |
height=150, | |
drawing_mode=drawing_mode, | |
point_display_radius=point_display_radius if drawing_mode == 'point' else 0, | |
key="canvas", | |
) | |
# Process the canvas data and display it | |
display_canvas_data(canvas_result) |