import streamlit as st
from main import run_agent

st.set_page_config(page_title="CodeReview Buddy", layout="wide")

st.title("🤖 CodeReview Buddy (Production)")

st.write("App Loaded ✅")  # DEBUG

prompt = st.text_area("Enter your prompt:")

if st.button("Run Agent"):
    st.write("Button clicked ✅")  # DEBUG

    with st.spinner("Processing..."):
        try:
            result = run_agent(prompt)
            st.write("Agent finished ✅")  # DEBUG
            st.write(result)  # SEE RAW OUTPUT

        except Exception as e:
            st.error(f"Error: {e}")
            st.stop()

    st.subheader("Final Code")
    st.code(result.get("code", ""))

    st.subheader("Review")
    st.write(result.get("review", ""))