mirror of
https://github.com/oobabooga/text-generation-webui.git
synced 2024-10-01 01:26:03 -04:00
79 lines
1.7 KiB
Python
79 lines
1.7 KiB
Python
import gradio as gr
|
|
|
|
refresh_symbol = '\U0001f504' # 🔄
|
|
|
|
css = """
|
|
.tabs.svelte-710i53 {
|
|
margin-top: 0
|
|
}
|
|
.py-6 {
|
|
padding-top: 2.5rem
|
|
}
|
|
.dark #refresh-button {
|
|
background-color: #ffffff1f;
|
|
}
|
|
#refresh-button {
|
|
flex: none;
|
|
margin: 0;
|
|
padding: 0;
|
|
min-width: 50px;
|
|
border: none;
|
|
box-shadow: none;
|
|
border-radius: 10px;
|
|
background-color: #0000000d;
|
|
}
|
|
#download-label, #upload-label {
|
|
min-height: 0
|
|
}
|
|
#accordion {
|
|
}
|
|
"""
|
|
|
|
chat_css = """
|
|
.h-\[40vh\], .wrap.svelte-byatnx.svelte-byatnx.svelte-byatnx {
|
|
height: 66.67vh
|
|
}
|
|
.gradio-container {
|
|
max-width: 800px;
|
|
margin-left: auto;
|
|
margin-right: auto
|
|
}
|
|
.w-screen {
|
|
width: unset
|
|
}
|
|
div.svelte-362y77>*, div.svelte-362y77>.form>* {
|
|
flex-wrap: nowrap
|
|
}
|
|
/* fixes the API documentation in chat mode */
|
|
.api-docs.svelte-1iguv9h.svelte-1iguv9h.svelte-1iguv9h {
|
|
display: grid;
|
|
}
|
|
"""
|
|
|
|
class ToolButton(gr.Button, gr.components.FormComponent):
|
|
"""Small button with single emoji as text, fits inside gradio forms"""
|
|
|
|
def __init__(self, **kwargs):
|
|
super().__init__(variant="tool", **kwargs)
|
|
|
|
def get_block_name(self):
|
|
return "button"
|
|
|
|
def create_refresh_button(refresh_component, refresh_method, refreshed_args, elem_id):
|
|
def refresh():
|
|
refresh_method()
|
|
args = refreshed_args() if callable(refreshed_args) else refreshed_args
|
|
|
|
for k, v in args.items():
|
|
setattr(refresh_component, k, v)
|
|
|
|
return gr.update(**(args or {}))
|
|
|
|
refresh_button = ToolButton(value=refresh_symbol, elem_id=elem_id)
|
|
refresh_button.click(
|
|
fn=refresh,
|
|
inputs=[],
|
|
outputs=[refresh_component]
|
|
)
|
|
return refresh_button
|