Keldos commited on
Commit
af5f236
·
1 Parent(s): 8a8420b

test: 尝试override输出

Browse files
Files changed (3) hide show
  1. assets/custom.css +4 -1
  2. modules/overwrites.py +6 -2
  3. modules/utils.py +5 -0
assets/custom.css CHANGED
@@ -418,7 +418,8 @@ thead th {
418
  .message pre,
419
  .message pre[class*=language-] {
420
  color: #fff;
421
- overflow: auto;
 
422
  margin: .8em 1em 1em 0em !important;
423
  padding: var(--spacing-xl) 1.2em !important;
424
  border-radius: var(--radius-lg) !important;
@@ -426,6 +427,8 @@ thead th {
426
  .message pre code,
427
  .message pre code[class*=language-] {
428
  color: #fff;
 
 
429
  text-shadow: none;
430
  font-family: var(--font-mono);
431
  }
 
418
  .message pre,
419
  .message pre[class*=language-] {
420
  color: #fff;
421
+ overflow-x: auto;
422
+ overflow-y: hidden;
423
  margin: .8em 1em 1em 0em !important;
424
  padding: var(--spacing-xl) 1.2em !important;
425
  border-radius: var(--radius-lg) !important;
 
427
  .message pre code,
428
  .message pre code[class*=language-] {
429
  color: #fff;
430
+ padding: 0;
431
+ background-color: unset;
432
  text-shadow: none;
433
  font-family: var(--font-mono);
434
  }
modules/overwrites.py CHANGED
@@ -41,7 +41,7 @@ def postprocess(
41
  return processed_messages
42
 
43
  def postprocess_chat_messages(
44
- self, chat_message: str | tuple | list | None, role: str = "user"
45
  ) -> str | dict | None:
46
  if chat_message is None:
47
  return None
@@ -63,7 +63,11 @@ def postprocess_chat_messages(
63
  elif isinstance(chat_message, str):
64
  # chat_message = inspect.cleandoc(chat_message)
65
  # escape html spaces
66
- chat_message = chat_message.replace(" ", " ")
 
 
 
 
67
  return chat_message
68
  else:
69
  raise ValueError(f"Invalid message for Chatbot component: {chat_message}")
 
41
  return processed_messages
42
 
43
  def postprocess_chat_messages(
44
+ self, chat_message: str | tuple | list | None, role: str
45
  ) -> str | dict | None:
46
  if chat_message is None:
47
  return None
 
63
  elif isinstance(chat_message, str):
64
  # chat_message = inspect.cleandoc(chat_message)
65
  # escape html spaces
66
+ # chat_message = chat_message.replace(" ", " ")
67
+ if role == "bot":
68
+ chat_message = convert_before_marked(chat_message)
69
+ # elif role == "user":
70
+ # chat_message = convert_asis(chat_message)
71
  return chat_message
72
  else:
73
  raise ValueError(f"Invalid message for Chatbot component: {chat_message}")
modules/utils.py CHANGED
@@ -203,6 +203,11 @@ def convert_mdtext(md_text):
203
  output += ALREADY_CONVERTED_MARK
204
  return output
205
 
 
 
 
 
 
206
 
207
  def convert_asis(userinput):
208
  return (
 
203
  output += ALREADY_CONVERTED_MARK
204
  return output
205
 
206
+ def convert_before_marked(chat_message):
207
+ return (
208
+ f'<div class="raw-message hideM">{chat_message}</div><div class="md-message">{chat_message}</div>{ALREADY_CONVERTED_MARK}'
209
+ )
210
+
211
 
212
  def convert_asis(userinput):
213
  return (