import streamlit as st
from tensorflow.keras.models import load_model
from PIL import Image
import numpy as np


model=load_model('Rice.h5')


def procces_image(img):
    img = img.resize((170, 170), Image.Resampling.LANCZOS)
    img=np.array(img)
    img=img/255.0 #normalize ettik
    img=np.expand_dims(img,axis=0)
    return img

st.title("Pirinç Türlerinin Sınıflandırması :ear_of_rice:")
st.write("Resim Seç ve Model Yaprakların Durumunu tahmin et")

file=st.file_uploader('Bir Resim Seçin ',type=['jpg','jpeg','png'])

if file is not None:
    img = Image.open(file)
    st.image(img, caption='Yüklenen Resim')
    image = procces_image(img)
    prediction = model.predict(image)

    # Tahmin sonuçlarını kontrol et
    st.write("Tahmin Olasılıkları:", prediction)

    predicted_class = np.argmax(prediction)
    st.write("Tahmin Edilen Sınıf Indeksi:", predicted_class)

    class_names = ['Arborio','Basmati','Ipsala','Jasmine','Karacadag']
    if predicted_class < len(class_names):
        st.write("Tahmin Edilen Sınıf:", class_names[predicted_class])
    else:
        st.write("Geçersiz sınıf indeksi!")