Update DocuChat.py

DocuChat.py

@@ -18,34 +18,21 @@ class LangchainConversation:
     def __init__(self):
         self.file_input = pn.widgets.FileInput(height=45)
         self.openaikey = pn.widgets.PasswordInput(value="", placeholder="Enter your OpenAI API Key here...", height=45)
-        self.chatbox = pn.widgets.ChatBox(height=300, primary_name="User", show_input=False)
-        self.user_input = pn.widgets.TextInput(placeholder="Type your question...")
-        self.send_button = pn.widgets.Button(name="Send", width=0)
-        self.send_button.on_click(self._chat)
+        self.chatbox = pn.widgets.ChatBox(height=300, primary_name="User")
+        self.chatbox.param.watch(self._chat, 'value')
 
-    # decorator to disable the send button while the AI is thinking
-    def _disable_send_button(func):
-        def inner(self, *args, **kwargs):
-            try:
-                self.send_button.disabled = True
-                func(self, *args, **kwargs)
-            finally:
-                self.send_button.disabled = False
-        return inner
-
-    @_disable_send_button
-    def _chat(self, _):
-        user_message = self.user_input.value
-        if user_message:
-            self.chatbox.append({"User": user_message})
-            os.environ["OPENAI_API_KEY"] = self.openaikey.value
-            if self.file_input.value is not None:
-                self.file_input.save("/.cache/temp.pdf")
-                prompt_text = self.remove_empty_lines(user_message)
-                if prompt_text:
-                    result = self.qa(file="/.cache/temp.pdf", query=prompt_text)
-                    self.chatbox.append({"AI": result})
-            self.user_input.value = ""  # Clear the user input field
+    def _chat(self, event):
+        user_message = event.new[-1]
+        input = user_message.get("User")
+        if input is None:
+            return
+        os.environ["OPENAI_API_KEY"] = self.openaikey.value
+        if self.file_input.value is not None:
+            self.file_input.save("/.cache/temp.pdf")
+            prompt_text = self.remove_empty_lines(input)
+            if prompt_text:
+                result = self.qa(file="/.cache/temp.pdf", query=prompt_text)
+                self.chatbox.append({"AI": result})
 
     @staticmethod
     def remove_empty_lines(text):
@@ -71,9 +58,7 @@ class LangchainConversation:
             AI-Powered Query Engine for Document Insights (powered by LangChain & OpenAI)
             ...
             """),
-            pn.Row(self.file_input, self.openaikey),
-            self.chatbox,
-            pn.Row(self.user_input, self.send_button)
+            pn.Row(self.file_input, self.openaikey), self.chatbox
         ).servable()
         return layout