freddyaboulton HF Staff commited on
Commit
8a8d0a5
·
verified ·
1 Parent(s): 4110cf3

Commit 1: Add 50 file(s)

Browse files
Files changed (1) hide show
  1. demos/chatbot_multimodal/run.py +15 -10
demos/chatbot_multimodal/run.py CHANGED
@@ -3,9 +3,11 @@ import time
3
 
4
  # Chatbot demo with multimodal input (text, markdown, LaTeX, code blocks, image, audio, & video). Plus shows support for streaming text.
5
 
 
6
  def print_like_dislike(x: gr.LikeData):
7
  print(x.index, x.value, x.liked)
8
 
 
9
  def add_message(history, message):
10
  for x in message["files"]:
11
  history.append({"role": "user", "content": {"path": x}})
@@ -13,26 +15,29 @@ def add_message(history, message):
13
  history.append({"role": "user", "content": message["text"]})
14
  return history, gr.MultimodalTextbox(value=None, interactive=False)
15
 
 
16
  def bot(history: list):
17
  response = "**That's cool!**"
18
  history.append({"role": "assistant", "content": ""})
19
  for character in response:
20
- history[-1]['content'] += character
21
  time.sleep(0.05)
22
  yield history
23
 
 
24
  with gr.Blocks() as demo:
25
- chatbot = gr.Chatbot(
26
- elem_id="chatbot",
27
- bubble_full_width=False,
28
- type="messages"
29
- )
30
 
31
- chat_input = gr.MultimodalTextbox(interactive=True,
32
- file_count="multiple",
33
- placeholder="Enter message or upload file...", show_label=False)
 
 
 
34
 
35
- chat_msg = chat_input.submit(add_message, [chatbot, chat_input], [chatbot, chat_input])
 
 
36
  bot_msg = chat_msg.then(bot, chatbot, chatbot, api_name="bot_response")
37
  bot_msg.then(lambda: gr.MultimodalTextbox(interactive=True), None, [chat_input])
38
 
 
3
 
4
  # Chatbot demo with multimodal input (text, markdown, LaTeX, code blocks, image, audio, & video). Plus shows support for streaming text.
5
 
6
+
7
  def print_like_dislike(x: gr.LikeData):
8
  print(x.index, x.value, x.liked)
9
 
10
+
11
  def add_message(history, message):
12
  for x in message["files"]:
13
  history.append({"role": "user", "content": {"path": x}})
 
15
  history.append({"role": "user", "content": message["text"]})
16
  return history, gr.MultimodalTextbox(value=None, interactive=False)
17
 
18
+
19
  def bot(history: list):
20
  response = "**That's cool!**"
21
  history.append({"role": "assistant", "content": ""})
22
  for character in response:
23
+ history[-1]["content"] += character
24
  time.sleep(0.05)
25
  yield history
26
 
27
+
28
  with gr.Blocks() as demo:
29
+ chatbot = gr.Chatbot(elem_id="chatbot", bubble_full_width=False, type="messages")
 
 
 
 
30
 
31
+ chat_input = gr.MultimodalTextbox(
32
+ interactive=True,
33
+ file_count="multiple",
34
+ placeholder="Enter message or upload file...",
35
+ show_label=False,
36
+ )
37
 
38
+ chat_msg = chat_input.submit(
39
+ add_message, [chatbot, chat_input], [chatbot, chat_input]
40
+ )
41
  bot_msg = chat_msg.then(bot, chatbot, chatbot, api_name="bot_response")
42
  bot_msg.then(lambda: gr.MultimodalTextbox(interactive=True), None, [chat_input])
43