cmbagent / app.py
JFrazUCSF's picture
Create app.py
071d4ad verified
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())