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 (Final - CodeT5-base)")
st.write("Paste your Python code below and get a high-quality 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.strip(), tokenizer, model)
            st.subheader("✅ Explanation")
            st.write(explanation)
    else:
        st.warning("Please paste some code to explain.")