OneTeam/NaliiaBot
16
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: Implemented markdown format in UI response

Browse Source
This commit is contained in:
aserrador
2026-02-16 10:44:34 -05:00
parent 5d1e0d2984
commit 4e4179b132
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/ui/src/ui/app.py
View File

@@ -14,6 +14,7 @@ def response_generator(response: str = ""):
time.sleep(0.05) time.sleep(0.05)
async def main(): async def main():
st.image("src/ui/assets/logo-naliia.png", width=100, caption="NaliiaBot") st.image("src/ui/assets/logo-naliia.png", width=100, caption="NaliiaBot")
if "messages" not in st.session_state: if "messages" not in st.session_state:
@@ -31,10 +32,10 @@ async def main():
agent_client = AgentClient() agent_client = AgentClient()
response = await agent_client.send_message(prompt) response = await agent_client.send_message(prompt)
formated_response = response.json()['response']['messages'][-1]['content'] raw_response = response.json()['response']['messages'][-1]['content']
with st.chat_message("assistant"): with st.chat_message("assistant"):
response = st.write_stream(response_generator(formated_response)) formated_response = st.markdown(raw_response)
st.session_state.messages.append({"role": "assistant", "content": formated_response}) st.session_state.messages.append({"role": "assistant", "content": formated_response})
if __name__ == "__main__": import asyncio; asyncio.run(main()) if __name__ == "__main__": import asyncio; asyncio.run(main())