Spaces:
Runtime error
Runtime error
| import streamlit as st | |
| from utils import rnd_id | |
| from my_tools import TOOL_CLASSES | |
| from streamlit import session_state as ss | |
| import db_utils | |
| class PageTools: | |
| def __init__(self): | |
| self.name = "Tools" | |
| self.available_tools = TOOL_CLASSES | |
| def create_tool(self, tool_name): | |
| tool_class = self.available_tools[tool_name] | |
| tool_instance = tool_class(rnd_id()) | |
| if 'tools' not in ss: | |
| ss.tools = [] | |
| ss.tools.append(tool_instance) | |
| db_utils.save_tool(tool_instance) # Save tool to database | |
| def remove_tool(self, tool_id): | |
| ss.tools = [tool for tool in ss.tools if tool.tool_id != tool_id] | |
| db_utils.delete_tool(tool_id) | |
| st.rerun() | |
| def set_tool_parameter(self, tool_id, param_name, value): | |
| if value == "": | |
| value = None | |
| for tool in ss.tools: | |
| if tool.tool_id == tool_id: | |
| tool.set_parameters(**{param_name: value}) | |
| db_utils.save_tool(tool) | |
| break | |
| def get_tool_display_name(self, tool): | |
| first_param_name = tool.get_parameter_names()[0] if tool.get_parameter_names() else None | |
| first_param_value = tool.parameters.get(first_param_name, '') if first_param_name else '' | |
| return f"{tool.name} ({first_param_value if first_param_value else tool.tool_id})" | |
| def draw_tools(self): | |
| c1,c2 = st.columns([1, 3]) | |
| #st.write("Available Tools:") | |
| with c1: | |
| for tool_name in self.available_tools.keys(): | |
| tool_class = self.available_tools[tool_name] | |
| tool_instance = tool_class() | |
| tool_description = tool_instance.description | |
| if st.button(f"{tool_name}", key=f"enable_{tool_name}", help=tool_description): | |
| self.create_tool(tool_name) | |
| with c2: | |
| if 'tools' in ss: | |
| st.write("##### Enabled Tools") | |
| for tool in ss.tools: | |
| display_name = self.get_tool_display_name(tool) | |
| is_complete = tool.is_valid() | |
| expander_title = display_name if is_complete else f"❗ {display_name}" | |
| with st.expander(expander_title): | |
| st.write(tool.description) | |
| for param_name in tool.get_parameter_names(): | |
| param_value = tool.parameters.get(param_name, "") | |
| placeholder = "Required" if tool.is_parameter_mandatory(param_name) else "Optional" | |
| new_value = st.text_input(f"{param_name}", value=param_value, key=f"{tool.tool_id}_{param_name}", placeholder=placeholder) | |
| if new_value != param_value: | |
| self.set_tool_parameter(tool.tool_id, param_name, new_value) | |
| if st.button(f"Remove", key=f"remove_{tool.tool_id}"): | |
| self.remove_tool(tool.tool_id) | |
| def draw(self): | |
| st.subheader(self.name) | |
| self.draw_tools() | |