File size: 5,961 Bytes
32a4065 b8b0b89 32a4065 b8b0b89 72b06f2 3896508 72b06f2 a601250 7330cbd 2fae0bb a601250 9d4e149 7330cbd 9d4e149 72b06f2 b8b0b89 c9ed3c5 3896508 72b06f2 b7d0dc8 b8b0b89 b7d0dc8 72b06f2 b7d0dc8 b8b0b89 b7d0dc8 72b06f2 b7d0dc8 b8b0b89 b7d0dc8 72b06f2 b7d0dc8 b8b0b89 b7d0dc8 72b06f2 b8b0b89 c9ed3c5 72b06f2 b8b0b89 72b06f2 b8b0b89 72b06f2 b8b0b89 72b06f2 b8b0b89 72b06f2 b8b0b89 72b06f2 b8b0b89 72b06f2 b8b0b89 72b06f2 b8b0b89 72b06f2 b8b0b89 72b06f2 b8b0b89 72b06f2 b8b0b89 72b06f2 b8b0b89 72b06f2 b8b0b89 72b06f2 a601250 72b06f2 |
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 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
import streamlit as st
import pandas as pd
import matplotlib.figure
from PIL import Image
from pydub import AudioSegment
def app_chat(controller):
agent_config = controller.agent_config
# Chat code (user input, agent responses, etc.)
if "messages" not in st.session_state:
st.session_state.messages = []
st.markdown("Hello there! How can I assist you today?")
for message in st.session_state.messages:
with st.chat_message(message["role"]):
st.markdown(message["content"])
if user_message := st.chat_input("Enter message"):
st.chat_message("user").markdown(user_message)
st.session_state.messages.append({"role": "user", "content": user_message, "avatar": "🤗"})
response = ""
chat_respone = ""
with st.spinner('Please stand by ...'):
response = controller.handle_submission(user_message)
with st.chat_message("assistant"):
if response is None:
chat_respone = controller.handle_submission_chat(user_message, response)
st.write(chat_respone)
# st.warning("The agent's response is None. Please try again. Generate an image of a flying uncormn.")
elif isinstance(response, Image.Image):
agent_config.image = response
chat_respone = controller.handle_submission_chat(user_message, "No context . Created an image.")
st.write(chat_respone)
st.image(response)
elif isinstance(response, AudioSegment):
agent_config.audio = response
chat_respone = controller.handle_submission_chat(user_message, "Agent Tool created audio file.")
st.write(chat_respone)
st.audio(response)
elif isinstance(response, int):
chat_respone = controller.handle_submission_chat(user_message, response)
st.write(chat_respone)
st.markdown(response)
elif isinstance(response, str):
if "emojified_text" in response:
chat_respone = controller.handle_submission_chat(user_message, "Agent Tool created the text with emojies.")
st.write(chat_respone)
st.markdown(f"{response['emojified_text']}")
else:
chat_respone = controller.handle_submission_chat(user_message, response)
st.write(chat_respone)
st.markdown(response)
elif isinstance(response, list):
chat_respone = controller.handle_submission_chat(user_message, "Agent Tool produced a list")
for item in response:
st.markdown(item) # Assuming the list contains strings
st.write(chat_respone)
elif isinstance(response, pd.DataFrame):
chat_respone = controller.handle_submission_chat(user_message, "Agent Tool produced a pd.DataFrame")
st.write(chat_respone)
st.dataframe(response)
elif isinstance(response, pd.Series):
chat_respone = controller.handle_submission_chat(user_message, "Agent Tool produced a pd.Series")
st.write(chat_respone)
st.table(response.iloc[0:10])
elif isinstance(response, dict):
chat_respone = controller.handle_submission_chat(user_message, "Agent Tool produced a dict")
st.write(chat_respone)
st.json(response)
elif isinstance(response, st.graphics_altair.AltairChart):
chat_respone = controller.handle_submission_chat(user_message, "Agent Tool produced a st.graphics_altair.AltairChart")
st.write(chat_respone)
st.altair_chart(response)
elif isinstance(response, st.graphics_bokeh.BokehChart):
chat_respone = controller.handle_submission_chat(user_message, "Agent Tool produced a st.graphics_bokeh.BokehChart")
st.write(chat_respone)
st.bokeh_chart(response)
elif isinstance(response, st.graphics_graphviz.GraphvizChart):
chat_respone = controller.handle_submission_chat(user_message, "Agent Tool produced a st.graphics_graphviz.GraphvizChart")
st.write(chat_respone)
st.graphviz_chart(response)
elif isinstance(response, st.graphics_plotly.PlotlyChart):
chat_respone = controller.handle_submission_chat(user_message, "Agent Tool produced a st.graphics_plotly.PlotlyChart")
st.write(chat_respone)
st.plotly_chart(response)
elif isinstance(response, st.graphics_pydeck.PydeckChart):
chat_respone = controller.handle_submission_chat(user_message, "Agent Tool produced a st.graphics_pydeck.PydeckChart")
st.write(chat_respone)
st.pydeck_chart(response)
elif isinstance(response, matplotlib.figure.Figure):
chat_respone = controller.handle_submission_chat(user_message, "Agent Tool produced a matplotlib.figure.Figure")
st.write(chat_respone)
st.pyplot(response)
elif isinstance(response, st.graphics_vega_lite.VegaLiteChart):
chat_respone = controller.handle_submission_chat(user_message, "Agent Tool produced a VegaLiteChart")
st.write(chat_respone)
st.vega_lite_chart(response)
else:
st.warning("Unrecognized response type. Please try again. e.g. Generate an image of a flying horse.")
st.session_state.messages.append({"role": "assistant", "content": chat_respone, "avatar": "🦖"})
st.session_state.messages.append({"role": "assistant", "content": response, "avatar": "🤖"})
|