script.py 634 B

123456789101112131415161718
  1. import gradio as gr
  2. import modules.shared as shared
  3. import pandas as pd
  4. df = pd.read_csv("https://raw.githubusercontent.com/devbrones/llama-prompts/main/prompts/prompts.csv")
  5. def get_prompt_by_name(name):
  6. if name == 'None':
  7. return ''
  8. else:
  9. return df[df['Prompt name'] == name].iloc[0]['Prompt'].replace('\\n', '\n')
  10. def ui():
  11. if not shared.args.chat or shared.args.cai_chat:
  12. choices = ['None'] + list(df['Prompt name'])
  13. prompts_menu = gr.Dropdown(value=choices[0], choices=choices, label='Prompt')
  14. prompts_menu.change(get_prompt_by_name, prompts_menu, shared.gradio['textbox'])