init commit
This commit is contained in:
69
prompt.py
Normal file
69
prompt.py
Normal file
@@ -0,0 +1,69 @@
|
||||
"""
|
||||
This script is designed for editing the bare query
|
||||
|
||||
bare query - questions or information that inspire
|
||||
ChatGPT to understand the program deeply instead of
|
||||
caring about the response of ChatGPT
|
||||
|
||||
{
|
||||
'type': 'bare_query',
|
||||
'name': 'XXX',
|
||||
'prompt': {
|
||||
'role': 'user',
|
||||
'content': 'You should provide programming advice.',
|
||||
}
|
||||
}
|
||||
"""
|
||||
|
||||
import os
|
||||
import json
|
||||
|
||||
|
||||
BPATH = os.path.dirname(os.path.abspath(__file__))
|
||||
BPATH = os.path.join(BPATH, 'prompt.json')
|
||||
print(BPATH)
|
||||
assert(os.path.exists(BPATH))
|
||||
|
||||
with open(BPATH, 'r') as r:
|
||||
lst = json.load(r)
|
||||
|
||||
while True:
|
||||
choice = input('1. list 2. add new 3. keyword-based search 4. delete 5. exit: ')
|
||||
if choice == '1':
|
||||
print('\n-----------------')
|
||||
for _id, _query in enumerate(lst):
|
||||
_type = _query['type']
|
||||
_name = _type
|
||||
if 'name' in _query.keys():
|
||||
_name = _query['name']
|
||||
print(_id, f': ({_type}) ({_name})', repr(_query['prompt']['content']))
|
||||
elif choice == '2':
|
||||
_type = input('Input your query type:')
|
||||
_name = input('Input your query name (use type as default):')
|
||||
if not _name:
|
||||
_name = _typeprompt
|
||||
_query = input('Input your query content:')
|
||||
new_entry = {
|
||||
'type': _type,
|
||||
'name': _name,
|
||||
'prompt': {
|
||||
'role': 'user',
|
||||
'content': _query,
|
||||
}
|
||||
}
|
||||
lst.append(new_entry)
|
||||
elif choice == '3':
|
||||
k = input('Input your keyword:')
|
||||
for _id, _query in enumerate(lst):
|
||||
_c = _query['prompt']['content']
|
||||
if k in _c:
|
||||
print(_id, ':', _c)
|
||||
elif choice == '4':
|
||||
i = input('Input the id of deleted enetry:')
|
||||
lst.pop(int(i))
|
||||
elif choice == '5':
|
||||
with open(BPATH, 'w') as w:
|
||||
json.dump(lst, w, indent=4)
|
||||
exit(0)
|
||||
else:
|
||||
pass
|
||||
Reference in New Issue
Block a user