feat: overwrite the chatbot class

app.py

@@ -230,7 +230,101 @@ CSS = """
 p {direction: rtl; white-space: pre-line;}
 """
 
-chatbot = gr.Chatbot(label="PersianGPT",
+class Chatbot(gr.Chatbot):
+    def _preprocess_chat_messages(
+        self, chat_message: str | dict | None
+    ) -> str | tuple[str] | tuple[str, str] | None:
+        if chat_message is None:
+            return None
+        elif isinstance(chat_message, dict):
+            if chat_message["alt_text"] is not None:
+                return (chat_message["name"], chat_message["alt_text"])
+            else:
+                return (chat_message["name"],)
+        else:  # string
+            return chat_message
+            
+    def preprocess(
+        self,
+        y: list[list[str | dict | None] | tuple[str | dict | None, str | dict | None]],
+    ) -> list[list[str | tuple[str] | tuple[str, str] | None]]:
+        if y is None:
+            return y
+        processed_messages = []
+        for message_pair in y:
+            if not isinstance(message_pair, (tuple, list)):
+                raise TypeError(
+                    f"Expected a list of lists or list of tuples. Received: {message_pair}"
+                )
+            if len(message_pair) != 2:
+                raise TypeError(
+                    f"Expected a list of lists of length 2 or list of tuples of length 2. Received: {message_pair}"
+                )
+            processed_messages.append(
+                [
+                    self._preprocess_chat_messages(message_pair[0]),
+                    self._preprocess_chat_messages(message_pair[1]),
+                ]
+            )
+        return processed_messages
+
+    def _postprocess_chat_messages(
+        self, chat_message: str | tuple | list | None
+    ) -> str | dict | None:
+        if chat_message is None:
+            return None
+        elif isinstance(chat_message, (tuple, list)):
+            file_uri = str(chat_message[0])
+            if utils.validate_url(file_uri):
+                filepath = file_uri
+            else:
+                filepath = self.make_temp_copy_if_needed(file_uri)
+
+            mime_type = client_utils.get_mimetype(filepath)
+            return {
+                "name": filepath,
+                "mime_type": mime_type,
+                "alt_text": chat_message[1] if len(chat_message) > 1 else None,
+                "data": None,  # These last two fields are filled in by the frontend
+                "is_file": True,
+            }
+        elif isinstance(chat_message, str):
+            chat_message = inspect.cleandoc(chat_message)
+            return chat_message
+        else:
+            raise ValueError(f"Invalid message for Chatbot component: {chat_message}")
+
+    def postprocess(
+        self,
+        y: list[list[str | tuple[str] | tuple[str, str] | None] | tuple],
+    ) -> list[list[str | dict | None]]:
+        """
+        Parameters:
+            y: List of lists representing the message and response pairs. Each message and response should be a string, which may be in Markdown format.  It can also be a tuple whose first element is a string or pathlib.Path filepath or URL to an image/video/audio, and second (optional) element is the alt text, in which case the media file is displayed. It can also be None, in which case that message is not displayed.
+        Returns:
+            List of lists representing the message and response. Each message and response will be a string of HTML, or a dictionary with media information. Or None if the message is not to be displayed.
+        """
+        if y is None:
+            return []
+        processed_messages = []
+        for message_pair in y:
+            if not isinstance(message_pair, (tuple, list)):
+                raise TypeError(
+                    f"Expected a list of lists or list of tuples. Received: {message_pair}"
+                )
+            if len(message_pair) != 2:
+                raise TypeError(
+                    f"Expected a list of lists of length 2 or list of tuples of length 2. Received: {message_pair}"
+                )
+            processed_messages.append(
+                [
+                    self._postprocess_chat_messages(message_pair[0]),
+                    self._postprocess_chat_messages(message_pair[1]),
+                ]
+            )
+        return processed_messages
+
+chatbot = Chatbot(label="PersianGPT",
                      rtl=True,
                      show_share_button=True,
                      show_copy_button=True,
@@ -337,7 +431,6 @@ def vote(data: gr.LikeData):
         ]
         jsonlfile.write("\n".join(json_data) + "\n")
 
-
 def save_whotheyare(x):
     global USERNAME
     global NAME