import streamlit as st
from PIL import Image

from bird_app import BirdApp

bird_app = BirdApp()

st.title("A Classifier for Israel's birds")

file_name = st.file_uploader("Upload an image of a bird")

if file_name is not None:
    col1, col2 = st.columns(2)
    image = Image.open(file_name)

    predictions = bird_app.predict(file_name)

    col1.image(image, use_column_width=True)

    col2.header(f"Predicted species: {predictions['class_name']}")
    # for p in predictions:
    #   col2.subheader(f"{ p['label'] }: { round(p['score'] * 100, 1)}%")