pyseek / app.py
ahmadtalha's picture
Upload 2 files
3435735 verified
import cv2
import numpy as np
import streamlit as st
from PIL import Image
import streamlit_image_coordinates
def main():
prev_val = None
st.title("PySeek")
st.sidebar.title("Upload Image")
# File uploader widget
uploaded_file = st.sidebar.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
new_img = uploaded_file.name if uploaded_file is not None else None
# print(uploaded_file.file_id)
if uploaded_file is not None:
if uploaded_file.file_id not in st.session_state:
st.session_state[uploaded_file.file_id] = []
# Open and display the image
image = Image.open(uploaded_file)
img = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
# st.image(image, caption='Uploaded Image', use_column_width=True)
value = streamlit_image_coordinates.streamlit_image_coordinates(
# Image.open(uploaded_file),
img,
key="pil",
)
if value is not None and value != prev_val:
prev_val = value
st.session_state[uploaded_file.file_id].append((value['x'], value['y']))
if uploaded_file.file_id in st.session_state:
for val in st.session_state[uploaded_file.file_id]:
st.write(f"x: {val[0]}, y: {val[1]}")
if __name__ == "__main__":
main()