import traceback
import streamlit as st

st.title("CMBAgent")

task = st.text_area("Task", "Draw two random numbers and give me their sum")
agent = st.selectbox("Agent", ["engineer", "planner", "researcher"], index=0)
engineer_model = st.text_input("Engineer model", "gpt-4o-mini")

if st.button("Run"):
    try:
        import cmbagent
        result = cmbagent.one_shot(
            task,
            agent=agent,
            engineer_model=engineer_model,
        )
        st.write(result)
    except Exception as e:
        st.error(f"Run failed: {e}")
        st.code(traceback.format_exc())