Update src/chatbot.py

src/chatbot.py

@@ -17,17 +17,13 @@ logger = logging.getLogger(__name__)
 from langchain.callbacks.base import BaseCallbackHandler
 
 class StreamHandler(BaseCallbackHandler):
-    def __init__(self, placeholder=None):
+    def __init__(self):
         self.current_text = ""
-        self.placeholder = placeholder
+
 
     def on_llm_new_token(self, token: str, **kwargs):
+        print(token, end="", flush=True)   # stream to console
         self.current_text += token
-        if self.placeholder:
-            self.placeholder.markdown(self.current_text + "▌")
-
-    def get_text(self):
-        return self.current_text
 
 
 class BioethicsChatbot:
@@ -168,11 +164,9 @@ class BioethicsChatbot:
 
         self.stream_handler.current_text = ""
         
-        if stream_placeholder:
-            self.stream_handler.placeholder = stream_placeholder
-
         # streaming happens here
         _ = self.llm.invoke(prompt)
+        print()
     
         answer = self.stream_handler.get_text()
         self.history.append((question, answer))