hmb HF staff commited on
Commit
6d15ed3
·
verified ·
1 Parent(s): 44fc304

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +55 -0
app.py ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import os
3
+ # Multimodal Chatbot demo that shows support for examples (example messages shown within the chatbot).
4
+
5
+ def print_like_dislike(x: gr.LikeData):
6
+ print(x.index, x.value, x.liked)
7
+
8
+ def add_message(history, message):
9
+ for x in message["files"]:
10
+ history.append(((x,), None))
11
+ if message["text"] is not None:
12
+ history.append((message["text"], None))
13
+ return history, gr.MultimodalTextbox(value=None, interactive=False)
14
+
15
+ def append_example_message(x: gr.SelectData, history):
16
+ if x.value["text"] is not None:
17
+ history.append((x.value["text"], None))
18
+ if "files" in x.value:
19
+ if isinstance(x.value["files"], list):
20
+ for file in x.value["files"]:
21
+ history.append((file, None))
22
+ else:
23
+ history.append((x.value["files"], None))
24
+ return history
25
+
26
+ def respond(history):
27
+ history[-1][1] = "Cool!"
28
+ return history
29
+
30
+ with gr.Blocks(fill_height=True) as demo:
31
+ chatbot = gr.Chatbot(
32
+ elem_id="chatbot",
33
+ bubble_full_width=False,
34
+ scale=1,
35
+ placeholder='<h1 style="font-weight: bold; color: #FFFFFF; text-align: center; font-size: 48px; font-family: Arial, sans-serif;">Welcome to Gradio!</h1>',
36
+ examples=[{"icon": os.path.join(os.path.dirname(__file__), "files/avatar.png"), "display_text": "Display Text Here!", "text": "Try this example with this audio.", "files": [os.path.join(os.path.dirname(__file__), "files/cantina.wav")]},
37
+ {"text": "Try this example with this image.", "files": [os.path.join(os.path.dirname(__file__), "files/avatar.png")]},
38
+ {"text": "This is just text, no files!"},
39
+ {"text": "Try this example with this image.", "files": [os.path.join(os.path.dirname(__file__), "files/avatar.png"), os.path.join(os.path.dirname(__file__), "files/avatar.png")]},
40
+ {"text": "Try this example with this Audio.", "files": [os.path.join(os.path.dirname(__file__), "files/cantina.wav")]}]
41
+ )
42
+
43
+ chat_input = gr.MultimodalTextbox(interactive=True,
44
+ file_count="multiple",
45
+ placeholder="Enter message or upload file...", show_label=False)
46
+
47
+ chat_msg = chat_input.submit(add_message, [chatbot, chat_input], [chatbot, chat_input])
48
+ bot_msg = chat_msg.then(respond, chatbot, chatbot, api_name="bot_response")
49
+ bot_msg.then(lambda: gr.MultimodalTextbox(interactive=True), None, [chat_input])
50
+
51
+ chatbot.like(print_like_dislike, None, None)
52
+ chatbot.example_select(append_example_message, [chatbot], [chatbot]).then(respond, chatbot, chatbot, api_name="respond")
53
+
54
+ if __name__ == "__main__":
55
+ demo.launch()