text-generation-webui/extensions/llama_prompts/script.py

19 lines
634 B
Python
Raw Normal View History

2023-03-11 14:11:15 -05:00
import gradio as gr
import modules.shared as shared
import pandas as pd
df = pd.read_csv("https://raw.githubusercontent.com/devbrones/llama-prompts/main/prompts/prompts.csv")
def get_prompt_by_name(name):
if name == 'None':
return ''
else:
return df[df['Prompt name'] == name].iloc[0]['Prompt'].replace('\\n', '\n')
def ui():
2023-03-12 04:21:50 -04:00
if not shared.args.chat or shared.args.cai_chat:
2023-03-11 14:11:15 -05:00
choices = ['None'] + list(df['Prompt name'])
prompts_menu = gr.Dropdown(value=choices[0], choices=choices, label='Prompt')
prompts_menu.change(get_prompt_by_name, prompts_menu, shared.gradio['textbox'])