Merge pull request #24 from flepied/bug23

ui: display the question from the user before the answer from the agent
flepied · Aug 25, 2023 · 6260ac3 · 6260ac3
2 parents 44d0517 + e282057
commit 6260ac3
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "second-brain-agent"
-version = "0.3.0"
+version = "0.2.1"
 description = "Second Brain AI Agent"
 authors = ["Frederic Lepied <[email protected]>"]
 license = "GPL 3.0"

diff --git a/screenshot.png b/screenshot.png
diff --git a/second_brain_agent.py b/second_brain_agent.py
@@ -7,15 +7,16 @@
 import streamlit as st
 from dotenv import load_dotenv
 
-from htmlTemplates import css, user_template
+from htmlTemplates import bot_template, css, user_template
 from lib import Agent
 
 
 def handle_userinput(user_question):
     "Handle the input from the user as a question to the LLM"
     response = st.session_state.agent.html_question(user_question)
     st.write(
-        user_template.replace("{{MSG}}", response),
+        user_template.replace("{{MSG}}", user_question),
+        bot_template.replace("{{MSG}}", response),
         unsafe_allow_html=True,
     )