import streamlit as st
from model.model_utils import load_model, generate_explanation

st.set_page_config(page_title="Code Explainer", layout="centered")

st.title("🧠 Code Explainer (CodeT5-small)")
st.write("Paste your Python code below and get a natural language explanation:")

code_input = st.text_area("Paste Python Code", height=200)

if st.button("Explain"):
    if code_input.strip():
        with st.spinner("Generating explanation..."):
            tokenizer, model = load_model()
            explanation = generate_explanation(code_input, tokenizer, model)
            st.subheader("✅ Explanation")
            st.write(explanation)
    else:
        st.warning("Please paste some code to explain.")