Update DocuChat.py

DocuChat.py

@@ -49,18 +49,26 @@ class LangchainConversation:
         self.chatbox.param.watch(self._chat, 'value')
         self.chat_history = []  # Chat history to store previous queries and responses
 
-    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")
+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:
+        file_path = "/.cache/temp.pdf"
+        self.file_input.save(file_path)
+        # Check if the uploaded file is a PDF
+        if self.file_input.filename.lower().endswith('.pdf'):
             prompt_text = self.remove_empty_lines(input)
             if prompt_text:
-                result = self.qa(file="/.cache/temp.pdf", query=prompt_text)
+                result = self.qa(file=file_path, query=prompt_text)
                 self.chatbox.append({"AI": result})
+        else:
+            # Delete the non-PDF file from cache
+            os.remove(file_path)
+            # Display an error message in the chatbox
+            self.chatbox.append({"AI": "Error: Only PDF files are allowed. Please upload a valid PDF file."})
 
     @staticmethod
     def remove_empty_lines(text):