# app.py
import streamlit as st
from tensorflow.keras.models import load_model
from tensorflow.keras.preprocessing import image
import numpy as np
from PIL import Image

# --- Modeli Yükle ---
model = load_model("src/heads_tails_vgg16.keras")

st.title("Heads or Tails Predictor")

# --- Kullanıcıdan Resim Al ---
uploaded_file = st.file_uploader("Upload a coin image", type=["jpg", "png"])
if uploaded_file:
    img = Image.open(uploaded_file).resize((224, 224))
    st.image(img, caption="Uploaded Image", use_column_width=True)

    # --- Tahmin ---
    x = np.expand_dims(np.array(img)/255.0, axis=0)
    pred = model.predict(x)[0][0]

    # --- Sonuç ---
    st.write("Prediction:", "Heads" if pred > 0.5 else "Tails")
    st.write("Probability of Heads:", round(float(pred), 3))