Update src/chatbot.py

src/chatbot.py

@@ -19,16 +19,19 @@ from langchain.callbacks.base import BaseCallbackHandler
 class StreamHandler(BaseCallbackHandler):
     def __init__(self):
         self.current_text = ""
-
+        self.placeholder = None  # we attach this later
 
     def on_llm_new_token(self, token: str, **kwargs):
-        print(token, end="", flush=True)   # stream to console
         self.current_text += token
+        if self.placeholder is not None:
+            # Show partial text with a cursor ▌
+            self.placeholder.markdown(self.current_text + "▌")
 
-    def get_text(self): 
+    def get_text(self):
         return self.current_text
 
 
+
 class BioethicsChatbot:
     def __init__(self, data_dir: str="data/sample_papers"):
         self.processor = DocumentProcessor()