mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-21 02:19:59 +08:00
30 lines
1.1 KiB
Python
30 lines
1.1 KiB
Python
|
import gradio as gr
|
||
|
|
import random
|
||
|
|
|
||
|
|
def chat(message):
|
||
|
|
history = gr.get_state() or []
|
||
|
|
if message.startswith("How many"):
|
||
|
|
response = random.randint(1,10)
|
||
|
|
elif message.startswith("How"):
|
||
|
|
response = random.choice(["Great", "Good", "Okay", "Bad"])
|
||
|
|
elif message.startswith("Where"):
|
||
|
|
response = random.choice(["Here", "There", "Somewhere"])
|
||
|
|
else:
|
||
|
|
response = "I don't know"
|
||
|
|
history.append((message, response))
|
||
|
|
gr.set_state(history)
|
||
|
|
html = "<div class='chatbot'>"
|
||
|
|
for user_msg, resp_msg in history:
|
||
|
|
html += f"<div class='user_msg'>{user_msg}</div>"
|
||
|
|
html += f"<div class='resp_msg'>{resp_msg}</div>"
|
||
|
|
html += "</div>"
|
||
|
|
return html
|
||
|
|
|
||
|
|
iface = gr.Interface(chat, "text", "html", css="""
|
||
|
|
.chatbox {display:flex;flex-direction:column}
|
||
|
|
.user_msg, .resp_msg {padding:4px;margin-bottom:4px;border-radius:4px;width:80%}
|
||
|
|
.user_msg {background-color:cornflowerblue;color:white;align-self:start}
|
||
|
|
.resp_msg {background-color:lightgray;align-self:self-end}
|
||
|
|
""", allow_screenshot=False, allow_flagging=False)
|
||
|
|
if __name__ == "__main__":
|
||
|
|
iface.launch()
|