-
Notifications
You must be signed in to change notification settings - Fork 3
/
PromptDialog.py
112 lines (98 loc) · 4.97 KB
/
PromptDialog.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
import wx
import pandas as pd
import requests
import io
from Settings import SettingsManager
import os
class PromptDialog(wx.Dialog):
def __init__(self, parent, prompt=""):
super().__init__(parent, title="System Prompt Manager", size=(700, 500))
self.panel = wx.Panel(self)
self.prompt_file = os.path.join(SettingsManager().config_dir, "prompts.csv")
if os.path.exists(self.prompt_file):
self.prompt_data = pd.read_csv(self.prompt_file)
self.prompt_data = self.prompt_data.sort_values(by='act').reset_index(drop=True)
else:
self.prompt_data = pd.DataFrame(columns=["act", "prompt"])
# UI Elements
self.act_list = wx.ListBox(self.panel, choices=self.prompt_data['act'].tolist(), style=wx.LB_SINGLE)
self.prompt_text = wx.TextCtrl(self.panel, style=wx.TE_MULTILINE)
self.prompt_text.SetValue(prompt)
self.new_button = wx.Button(self.panel, label="New")
self.save_button = wx.Button(self.panel, label="Save")
self.delete_button = wx.Button(self.panel, label="Delete")
self.update_button = wx.Button(self.panel, label="Download&Update Awesome ChatGPT Prompts")
self.set_button = wx.Button(self.panel, label='Set Prompt', id=wx.ID_OK)
self.cancel_button = wx.Button(self.panel, label='Cancel', id=wx.ID_CANCEL)
self.sizer = wx.BoxSizer(wx.VERTICAL)
self.sizer.Add(self.act_list, proportion=1, flag=wx.EXPAND | wx.ALL, border=5)
self.sizer.Add(self.prompt_text, proportion=2, flag=wx.EXPAND | wx.ALL, border=5)
self.button_sizer = wx.BoxSizer(wx.HORIZONTAL)
self.button_sizer.Add(self.new_button, flag=wx.ALL, border=5)
self.button_sizer.Add(self.save_button, flag=wx.ALL, border=5)
self.button_sizer.Add(self.delete_button, flag=wx.ALL, border=5)
self.button_sizer.Add(self.update_button, flag=wx.ALL, border=5)
self.button_sizer.Add(self.set_button, flag=wx.ALL, border=5)
self.button_sizer.Add(self.cancel_button, flag=wx.ALL, border=5)
self.sizer.Add(self.button_sizer, flag=wx.ALIGN_CENTER)
self.panel.SetSizerAndFit(self.sizer)
# Bind Events
self.act_list.Bind(wx.EVT_LISTBOX, self.on_act_selected)
self.new_button.Bind(wx.EVT_BUTTON, self.on_new)
self.save_button.Bind(wx.EVT_BUTTON, self.on_save)
self.delete_button.Bind(wx.EVT_BUTTON, self.on_delete)
self.update_button.Bind(wx.EVT_BUTTON, self.on_update)
result = self.prompt_data[self.prompt_data['prompt'] == prompt]
if not result.empty:
self.select(result.index[0])
def select(self, index):
self.act_list.SetSelection(index)
event = wx.CommandEvent(wx.wxEVT_LISTBOX, self.act_list.GetId())
event.SetInt(index)
event.SetString(self.act_list.GetString(index))
wx.PostEvent(self.act_list.GetEventHandler(), event)
self.act_list.Refresh()
def on_act_selected(self, event):
selection = self.act_list.GetSelection()
if selection != wx.NOT_FOUND:
prompt = self.prompt_data.iloc[selection]['prompt']
self.prompt_text.SetValue(prompt)
def on_new(self, event):
act = wx.GetTextFromUser("Enter new name:", "New")
if act:
prompt = self.prompt_text.GetValue()
self.prompt_data = self.prompt_data._append({'act': act, 'prompt': prompt}, ignore_index=True)
self.prompt_data = self.prompt_data.sort_values(by='act').reset_index(drop=True)
self.act_list.Set(self.prompt_data['act'].tolist())
self.select(self.prompt_data.index[self.prompt_data['act'] == act].tolist()[0])
self.prompt_data.to_csv(self.prompt_file, index=False)
def on_save(self, event):
selection = self.act_list.GetSelection()
if selection != wx.NOT_FOUND:
self.prompt_data.at[selection, 'prompt'] = self.prompt_text.GetValue()
self.prompt_data.to_csv(self.prompt_file, index=False)
def on_delete(self, event):
selection = self.act_list.GetSelection()
if selection == wx.NOT_FOUND: return
with wx.MessageDialog(self, f"Are you sure you want to delete {self.act_list.GetStringSelection()}?", 'Delete', wx.YES_NO|wx.ICON_QUESTION) as dlg:
dlg.SetYesNoLabels("Yes", "No")
if dlg.ShowModal() == wx.ID_NO: return
self.prompt_data = self.prompt_data.drop(selection).reset_index(drop=True)
self.prompt_data = self.prompt_data.sort_values(by='act').reset_index(drop=True)
self.prompt_data.to_csv(self.prompt_file, index=False)
self.act_list.Set(self.prompt_data['act'].tolist())
self.prompt_text.SetValue("")
def on_update(self, event):
try:
url = 'https://github.com/f/awesome-chatgpt-prompts/raw/main/prompts.csv'
response = requests.get(url)
response.raise_for_status()
new_prompt_data = pd.read_csv(io.StringIO(response.text))
combined_data = pd.concat([self.prompt_data, new_prompt_data]).drop_duplicates(subset='act', keep='last').reset_index(drop=True)
self.prompt_data = combined_data.sort_values(by='act').reset_index(drop=True)
self.prompt_data.to_csv(self.prompt_file, index=False)
self.act_list.Set(self.prompt_data['act'].tolist())
self.prompt_text.SetValue("")
wx.MessageBox("Prompts updated successfully.", "Info", wx.OK | wx.ICON_INFORMATION)
except requests.RequestException as e:
wx.MessageBox(f"Failed to update prompts: {e}", "Error", wx.OK | wx.ICON_ERROR)