Commit
·
a839a97
1
Parent(s):
d7e67f6
Delete stbot.py
Browse files
stbot.py
DELETED
|
@@ -1,145 +0,0 @@
|
|
| 1 |
-
|
| 2 |
-
import Constants
|
| 3 |
-
import sys
|
| 4 |
-
import openai
|
| 5 |
-
from PyQt5.QtCore import Qt
|
| 6 |
-
from PyQt5.QtGui import QPixmap
|
| 7 |
-
from PyQt5.QtWidgets import (
|
| 8 |
-
QApplication,
|
| 9 |
-
QWidget,
|
| 10 |
-
QLabel,
|
| 11 |
-
QLineEdit,
|
| 12 |
-
QPushButton,
|
| 13 |
-
QVBoxLayout,
|
| 14 |
-
QHBoxLayout,
|
| 15 |
-
QGroupBox,
|
| 16 |
-
QTextEdit
|
| 17 |
-
)
|
| 18 |
-
|
| 19 |
-
openai.api_key = Constants.API_KEY
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
class MainWindow(QWidget):
|
| 23 |
-
def __init__(self):
|
| 24 |
-
super().__init__()
|
| 25 |
-
self.init_ui()
|
| 26 |
-
|
| 27 |
-
def init_ui(self):
|
| 28 |
-
self.logo_label = QLabel()
|
| 29 |
-
self.logo_pixmap = QPixmap('michaelajayi.jpg').scaled(
|
| 30 |
-
500, 500, Qt.KeepAspectRatio, Qt.SmoothTransformation)
|
| 31 |
-
self.logo_label.setPixmap(self.logo_pixmap)
|
| 32 |
-
|
| 33 |
-
self.input_label = QLabel('Ask Something')
|
| 34 |
-
self.input_field = QLineEdit()
|
| 35 |
-
self.input_field.setPlaceholderText('Type here...')
|
| 36 |
-
self.answer_label = QLabel('Answer:')
|
| 37 |
-
self.answer_field = QTextEdit()
|
| 38 |
-
self.answer_field.setReadOnly(True)
|
| 39 |
-
self.sumbit_button = QPushButton('Sumbit')
|
| 40 |
-
self.sumbit_button.setStyleSheet(
|
| 41 |
-
"""
|
| 42 |
-
QPushButton {
|
| 43 |
-
background-color: #2F3540;
|
| 44 |
-
border: none;
|
| 45 |
-
color: white;
|
| 46 |
-
padding: 15px 32px;
|
| 47 |
-
font-size: 18px;
|
| 48 |
-
font-weight: bold;
|
| 49 |
-
border-radius: 25px;
|
| 50 |
-
}
|
| 51 |
-
QpushButton:hover {
|
| 52 |
-
background-color: #3e8e41;
|
| 53 |
-
}
|
| 54 |
-
"""
|
| 55 |
-
|
| 56 |
-
)
|
| 57 |
-
self.recommended_questions_group = QGroupBox('Recommended Questions')
|
| 58 |
-
self.recommended_questions_layout = QVBoxLayout()
|
| 59 |
-
self.recommended_questions = ["What is the four corner opposition?",
|
| 60 |
-
"How do I become a better storyteller?", "What are some popular ways to get better at writting?"]
|
| 61 |
-
self.question_buttons = []
|
| 62 |
-
|
| 63 |
-
# Create a layout
|
| 64 |
-
layout = QVBoxLayout()
|
| 65 |
-
layout.setContentsMargins(20, 20, 20, 20)
|
| 66 |
-
layout.setSpacing(50)
|
| 67 |
-
layout.setAlignment(Qt.AlignCenter)
|
| 68 |
-
|
| 69 |
-
# Add Logo
|
| 70 |
-
layout.addWidget(self.logo_label, alignment=Qt.AlignCenter)
|
| 71 |
-
|
| 72 |
-
# Add Input Field and Sumbit Button
|
| 73 |
-
input_layout = QHBoxLayout()
|
| 74 |
-
input_layout.addWidget(self.input_label)
|
| 75 |
-
input_layout.addWidget(self.input_field)
|
| 76 |
-
input_layout.addWidget(self.sumbit_button)
|
| 77 |
-
layout.addLayout(input_layout)
|
| 78 |
-
|
| 79 |
-
# Add Answer Field
|
| 80 |
-
layout.addWidget(self.answer_label)
|
| 81 |
-
layout.addWidget(self.answer_field)
|
| 82 |
-
|
| 83 |
-
# add the recommended questions buttons
|
| 84 |
-
for question in self.recommended_questions:
|
| 85 |
-
button = QPushButton(question)
|
| 86 |
-
button.setStyleSheet(
|
| 87 |
-
"""
|
| 88 |
-
QPushButton {
|
| 89 |
-
background-color: #FFFFFF:
|
| 90 |
-
border: 2px solid #00AEFF;
|
| 91 |
-
colour: #00AEFF;
|
| 92 |
-
padding: 10px 20px;
|
| 93 |
-
font-size: 30px;
|
| 94 |
-
font-weight: bold;
|
| 95 |
-
border-radius: 5px;
|
| 96 |
-
}
|
| 97 |
-
QPushButton:hover {
|
| 98 |
-
background-color: #00AEFF;
|
| 99 |
-
color: #FFFFFF;
|
| 100 |
-
}"""
|
| 101 |
-
)
|
| 102 |
-
button.clicked.connect(
|
| 103 |
-
lambda _, q=question: self.input_field.setText(q))
|
| 104 |
-
self.recommended_questions_layout.addWidget(button)
|
| 105 |
-
self.question_buttons.append(button)
|
| 106 |
-
self.recommended_questions_group.setLayout(
|
| 107 |
-
self.recommended_questions_layout)
|
| 108 |
-
layout.addWidget(self.recommended_questions_group)
|
| 109 |
-
|
| 110 |
-
# Set the layout
|
| 111 |
-
self.setLayout(layout)
|
| 112 |
-
|
| 113 |
-
# Set the window properties
|
| 114 |
-
self.setWindowTitle('Storyteller Writer Advisor Bot')
|
| 115 |
-
self.setGeometry(200, 200, 600, 600)
|
| 116 |
-
|
| 117 |
-
# Connect the submit button to the function which queries OpenAI's API
|
| 118 |
-
self.sumbit_button.clicked.connect(self.get_answer)
|
| 119 |
-
|
| 120 |
-
def get_answer(self):
|
| 121 |
-
question = self.input_field.text()
|
| 122 |
-
|
| 123 |
-
completion = openai.ChatCompletion.create(
|
| 124 |
-
model="gpt-4",
|
| 125 |
-
messages=[
|
| 126 |
-
{"role": "system", "content": "You are a Storyteller expert. Answer the follwing questions in a concise way or with bullet points."},
|
| 127 |
-
{"role": "user", "content": "What is the four corner opposition?"},
|
| 128 |
-
{"role": "assistant", "content": "A story structure writing technique that draws the lines between four leading charcters conflicts."},
|
| 129 |
-
{"role": "user", "content": f'{question}'}],
|
| 130 |
-
max_tokens=1024,
|
| 131 |
-
n=1,
|
| 132 |
-
stop=None,
|
| 133 |
-
temperature=1
|
| 134 |
-
)
|
| 135 |
-
|
| 136 |
-
answer = completion.choices[0].message.content
|
| 137 |
-
|
| 138 |
-
self.answer_field.setText(answer)
|
| 139 |
-
|
| 140 |
-
|
| 141 |
-
if __name__ == '__main__':
|
| 142 |
-
app = QApplication(sys.argv)
|
| 143 |
-
window = MainWindow()
|
| 144 |
-
window.show()
|
| 145 |
-
sys.exit(app.exec_())
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|