import streamlit as st
import pandas as pd
import pickle

# from huggingface_hub import login

# # Use your Hugging Face token
# login(token="hf_XmkhAdKiaTYaQbgMoGTYRqBFDFVAjvbTI")


model = pickle.load(open('model.pkl', 'rb'))

# from huggingface_hub import create_repo

# # Replace 'your_model_name' with the name you want for your model
# repo_url = create_repo(name='Almond Classification', private=False)

st.title('Almond Classification')
st.write('This web app classifies almonds based on your input features.')


# # Input untuk setiap fitur
# length_major_axis = st.number_input('Length (major axis)', min_value=269.356903, max_value=279.879883)
# width_minor_axis = st.number_input('Width (minor axis)', min_value=176.023636, max_value=227.940628)
# thickness_depth = st.number_input('Thickness (depth)', min_value=0.0, max_value=279.879883)
# area = st.number_input('Area', min_value=0.0, max_value=279.879883)
# perimeter = st.number_input('Perimeter', min_value=0.0, max_value=279.879883)
# roundness = st.slider('Roundness', min_value=0.0, max_value=1.0, step=0.01)
# solidity = st.slider('Solidity', min_value=0.0, max_value=1.0, step=0.01)
# compactness =  st.slider('Compactness', min_value=0.0, max_value=1.0, step=0.01)
# aspect_ratio =  st.slider('Aspect Ratio', min_value=0.0, max_value=5.0, step=0.01)
# eccentricity = st.slider('Eccentricity', min_value=0.0, max_value=1.0, step=0.01)
# extent = st.slider('Extent', min_value=0.0, max_value=1.0, step=0.01)
# convex_area = st.number_input('Convex hull (convex area)', min_value=0.0, step=0.01)


length_major_axis = st.number_input('Length (major axis)', min_value=269.356903, max_value=279.879883)
width_minor_axis = st.number_input('Width (minor axis)', min_value=176.023636, max_value=227.940628)
thickness_depth = st.number_input('Thickness (depth)', min_value=107.253448, max_value=127.795132)
area = st.number_input('Area', min_value=18471.5, max_value=36683.0)
perimeter = st.number_input('Perimeter', min_value=551.688379, max_value=887.310743)
roundness = st.slider('Roundness', min_value=0.472718, max_value=0.643761, step=0.01)
solidity = st.slider('Solidity', min_value=0.931800, max_value=0.973384, step=0.01)
compactness = st.slider('Compactness', min_value=1.383965, max_value=1.764701, step=0.01)
aspect_ratio = st.slider('Aspect Ratio', min_value=1.530231, max_value=1.705716, step=0.01)
eccentricity = st.slider('Eccentricity', min_value=0.75693, max_value=0.81012, step=0.01)
extent = st.slider('Extent', min_value=0.656535, max_value=0.725739, step=0.01)
convex_area = st.number_input('Convex hull (convex area)', min_value=18068.0, max_value=36683.0, step=0.01)


# Tombol untuk memprediksi
if st.button('Predict'):
    input_features = [[length_major_axis, width_minor_axis, thickness_depth, area,
                       perimeter, roundness, solidity, compactness, aspect_ratio,
                       eccentricity, extent, convex_area]]
    prediction = model.predict(input_features)
    st.write(f'The predicted class is: {prediction[0]}')