Spaces:
Sleeping
Sleeping
File size: 1,970 Bytes
58e0d8e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import streamlit as st
from streamlit import session_state as ss
import db_utils
from pg_agents import PageAgents
from pg_tasks import PageTasks
from pg_crews import PageCrews
from pg_tools import PageTools
from pg_crew_run import PageCrewRun
from pg_export_crew import PageExportCrew
from dotenv import load_dotenv
import os
def pages():
return {
'Crews': PageCrews(),
'Tools': PageTools(),
'Agents': PageAgents(),
'Tasks': PageTasks(),
'Kickoff!': PageCrewRun(),
'Import/export': PageExportCrew()
}
def load_data():
ss.agents = db_utils.load_agents()
ss.tasks = db_utils.load_tasks()
ss.crews = db_utils.load_crews()
ss.tools = db_utils.load_tools()
ss.enabled_tools = db_utils.load_tools_state()
def draw_sidebar():
with st.sidebar:
st.image("img/crewai_logo.png")
if 'page' not in ss:
ss.page = 'Crews'
selected_page = st.radio('Page', list(pages().keys()), index=list(pages().keys()).index(ss.page),label_visibility="collapsed")
if selected_page != ss.page:
ss.page = selected_page
st.rerun()
def main():
st.set_page_config(page_title="CrewAI Studio", page_icon="img/favicon.ico", layout="wide")
load_dotenv()
if (str(os.getenv('AGENTOPS_ENABLED')).lower() in ['true', '1']) and not ss.get('agentops_failed', False):
try:
import agentops
agentops.init(api_key=os.getenv('AGENTOPS_API_KEY'),auto_start_session=False)
except ModuleNotFoundError as e:
ss.agentops_failed = True
print(f"Error initializing AgentOps: {str(e)}")
db_utils.initialize_db()
load_data()
draw_sidebar()
PageCrewRun.maintain_session_state() #this will persist the session state for the crew run page so crew run can be run in a separate thread
pages()[ss.page].draw()
if __name__ == '__main__':
main()
|