Update app.py
Browse files
app.py
CHANGED
|
@@ -86,10 +86,21 @@ def load_tools(tool_names):
|
|
| 86 |
tool_names = [
|
| 87 |
"Chris4K/random-character-tool",
|
| 88 |
"Chris4K/text-generation-tool",
|
| 89 |
-
|
| 90 |
-
"Chris4K/
|
| 91 |
-
|
| 92 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 93 |
]
|
| 94 |
|
| 95 |
# Create tool loader instance
|
|
@@ -112,6 +123,9 @@ def handle_submission(user_message, selected_tools):
|
|
| 112 |
|
| 113 |
st.title("Hugging Face Agent and tools")
|
| 114 |
|
|
|
|
|
|
|
|
|
|
| 115 |
if "messages" not in st.session_state:
|
| 116 |
st.session_state.messages = []
|
| 117 |
|
|
@@ -119,7 +133,6 @@ for message in st.session_state.messages:
|
|
| 119 |
with st.chat_message(message["role"]):
|
| 120 |
st.markdown(message["content"])
|
| 121 |
|
| 122 |
-
tool_checkboxes = [st.checkbox(f"{tool.name} --- {tool.description} ") for tool in tool_loader.tools]
|
| 123 |
|
| 124 |
with st.chat_message("assistant"):
|
| 125 |
st.markdown("Hello there! How can I assist you today?")
|
|
@@ -154,15 +167,15 @@ if user_message := st.chat_input("Enter message"):
|
|
| 154 |
st.table(response.iloc[0:10])
|
| 155 |
elif isinstance(response, dict):
|
| 156 |
st.json(response)
|
| 157 |
-
elif isinstance(response,
|
| 158 |
st.altair_chart(response)
|
| 159 |
-
elif isinstance(response,
|
| 160 |
st.bokeh_chart(response)
|
| 161 |
-
elif isinstance(response,
|
| 162 |
st.graphviz_chart(response)
|
| 163 |
-
elif isinstance(response,
|
| 164 |
st.plotly_chart(response)
|
| 165 |
-
elif isinstance(response,
|
| 166 |
st.pydeck_chart(response)
|
| 167 |
elif isinstance(response, matplotlib.figure.Figure):
|
| 168 |
st.pyplot(response)
|
|
|
|
| 86 |
tool_names = [
|
| 87 |
"Chris4K/random-character-tool",
|
| 88 |
"Chris4K/text-generation-tool",
|
| 89 |
+
"Chris4K/sentiment-tool",
|
| 90 |
+
"Chris4K/token-counter-tool",
|
| 91 |
+
"Chris4K/most-downloaded-model",
|
| 92 |
+
"Chris4K/rag-tool",
|
| 93 |
+
"Chris4K/word-counter-tool",
|
| 94 |
+
"Chris4K/sentence-counter-tool",
|
| 95 |
+
"Chris4K/EmojifyTextTool",
|
| 96 |
+
"Chris4K/NamedEntityRecognitionTool",
|
| 97 |
+
"Chris4K/TextDownloadTool",
|
| 98 |
+
"Chris4K/source-code-retriever-tool",
|
| 99 |
+
"Chris4K/text-to-image",
|
| 100 |
+
"Chris4K/text-to-video",
|
| 101 |
+
"Chris4K/image-transformation",
|
| 102 |
+
"Chris4K/latent-upscaler-tool"
|
| 103 |
+
# More cool tools to come
|
| 104 |
]
|
| 105 |
|
| 106 |
# Create tool loader instance
|
|
|
|
| 123 |
|
| 124 |
st.title("Hugging Face Agent and tools")
|
| 125 |
|
| 126 |
+
tool_checkboxes = [st.checkbox(f"{tool.name} --- {tool.description} ") for tool in tool_loader.tools]
|
| 127 |
+
|
| 128 |
+
|
| 129 |
if "messages" not in st.session_state:
|
| 130 |
st.session_state.messages = []
|
| 131 |
|
|
|
|
| 133 |
with st.chat_message(message["role"]):
|
| 134 |
st.markdown(message["content"])
|
| 135 |
|
|
|
|
| 136 |
|
| 137 |
with st.chat_message("assistant"):
|
| 138 |
st.markdown("Hello there! How can I assist you today?")
|
|
|
|
| 167 |
st.table(response.iloc[0:10])
|
| 168 |
elif isinstance(response, dict):
|
| 169 |
st.json(response)
|
| 170 |
+
elif isinstance(response, st.graphics_altair.AltairChart):
|
| 171 |
st.altair_chart(response)
|
| 172 |
+
elif isinstance(response, st.graphics_bokeh.BokehChart):
|
| 173 |
st.bokeh_chart(response)
|
| 174 |
+
elif isinstance(response, st.graphics_graphviz.GraphvizChart):
|
| 175 |
st.graphviz_chart(response)
|
| 176 |
+
elif isinstance(response, st.graphics_plotly.PlotlyChart):
|
| 177 |
st.plotly_chart(response)
|
| 178 |
+
elif isinstance(response, st.graphics_pydeck.PydeckChart):
|
| 179 |
st.pydeck_chart(response)
|
| 180 |
elif isinstance(response, matplotlib.figure.Figure):
|
| 181 |
st.pyplot(response)
|