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

# Pirinç çeşitleri
rice_varieties = ['Arborio', 'Basmati', 'Ipsala', 'Jasmine', 'Karacadag']

# Modeli yükleme
model = load_model('modelrice.h5')
model.summary()

def process_img(img):
    img = img.resize((224, 224), Image.LANCZOS)  # 200x200 piksel boyutuna dönüştürme
    img = np.array(img) / 255.0  # Normalize etme
    img = np.expand_dims(img, axis=0)  # Resme boyut ekleme
    return img

st.title("Pirinç Çeşidi Sınıflandırması :rice:")
st.write(
    "Bir pirinç resmi seçin ve modelimiz, bu resmin hangi pirinç çeşidinden olduğunu tahmin etsin. 🖼️📊\n"
    "Upload an image and the model will predict which rice variety your image shows."
)

# Stil ayarları
st.markdown("""
<style>
    .reportview-container {
        background: #F0F2F6;
    }
    .sidebar .sidebar-content {
        background: #E0E0E0;
    }
    .css-18e3th9 {
        font-size: 1.25em;
        color: #333;
    }
</style>
""", unsafe_allow_html=True)

file = st.file_uploader("Resim Yükle & Bir resim seçiniz", type=['png', 'jpg', 'jpeg'])

if file is not None:
    img = Image.open(file)
    st.image(img, caption="Yüklenen Resim", use_column_width=True)
    
    result = process_img(img)
    prediction = model.predict(result)
    prediction_class = np.argmax(prediction)  # En yüksek tahmin edilen sınıf

    # Sınıf isimleri
    result_text = rice_varieties[prediction_class]

    st.write(f"**Sonuç:** {result_text}")
else:
    st.write("Lütfen bir resim yükleyin.")