Commit
·
d7e67f6
1
Parent(s):
9f337b2
Upload 6 files
Browse files- Constants.py +1 -0
- Stbotchainlit.py +90 -0
- chainlit.md +10 -0
- chainlitstbot.py +27 -0
- michaelajayi.jpg +0 -0
- stbot.py +145 -0
Constants.py
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
API_KEY = "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"
|
Stbotchainlit.py
ADDED
|
@@ -0,0 +1,90 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from langchain import PromptTemplate, LLMChain
|
| 2 |
+
from langchain import OpenAI, LLMMathChain, SerpAPIWrapper
|
| 3 |
+
from langchain.agents import initialize_agent, Tool, AgentExecutor
|
| 4 |
+
from langchain.chat_models import ChatOpenAI
|
| 5 |
+
import os
|
| 6 |
+
import chainlit as cl
|
| 7 |
+
from langchain.chat_models import ChatOpenAI
|
| 8 |
+
from langchain.agents import load_tools, initialize_agent, AgentType
|
| 9 |
+
import os
|
| 10 |
+
from langchain.chat_models import ChatOpenAI
|
| 11 |
+
from langchain.agents import initialize_agent
|
| 12 |
+
from langchain.agents import AgentType
|
| 13 |
+
import os
|
| 14 |
+
from langchain.tools import ShellTool
|
| 15 |
+
from langchain.tools import YouTubeSearchTool
|
| 16 |
+
from langchain.agents import initialize_agent, Tool
|
| 17 |
+
from langchain.agents import AgentType
|
| 18 |
+
from langchain.llms import OpenAI
|
| 19 |
+
from langchain import LLMMathChain, SerpAPIWrapper
|
| 20 |
+
import os
|
| 21 |
+
from chainlit.types import AskFileResponse
|
| 22 |
+
import chainlit as cl
|
| 23 |
+
from langchain.chat_models import ChatOpenAI
|
| 24 |
+
from langchain.chains import RetrievalQAWithSourcesChain
|
| 25 |
+
from langchain.vectorstores import Chroma
|
| 26 |
+
from langchain.embeddings.openai import OpenAIEmbeddings
|
| 27 |
+
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
| 28 |
+
from langchain.document_loaders import PyPDFLoader, TextLoader
|
| 29 |
+
import os
|
| 30 |
+
import openai
|
| 31 |
+
|
| 32 |
+
|
| 33 |
+
os.environ['OPENAI_API_KEY'] = "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"
|
| 34 |
+
OpenAI.api_key = "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"
|
| 35 |
+
|
| 36 |
+
|
| 37 |
+
template = """Question: {question}
|
| 38 |
+
Answer: Let's think step by step."""
|
| 39 |
+
|
| 40 |
+
|
| 41 |
+
@cl.langchain_factory(use_async=True)
|
| 42 |
+
def factory():
|
| 43 |
+
prompt = PromptTemplate(template=template, input_variables=["question"])
|
| 44 |
+
llm_chain = LLMChain(prompt=prompt, llm=ChatOpenAI(
|
| 45 |
+
temperature=0, streaming=True))
|
| 46 |
+
return llm_chain
|
| 47 |
+
|
| 48 |
+
|
| 49 |
+
def get_gpt_output(user_message):
|
| 50 |
+
response = openai.ChatCompletion.create(
|
| 51 |
+
model="gpt-4",
|
| 52 |
+
messages=[
|
| 53 |
+
{"role": "system", "content": "you are an writer that is obsessed with storytelling and will never stop talking about it"},
|
| 54 |
+
{"role": "user", "content": user_message}
|
| 55 |
+
],
|
| 56 |
+
temperature=1,
|
| 57 |
+
max_tokens=256,
|
| 58 |
+
top_p=1,
|
| 59 |
+
frequency_penalty=0,
|
| 60 |
+
presence_penalty=0
|
| 61 |
+
)
|
| 62 |
+
|
| 63 |
+
return response
|
| 64 |
+
|
| 65 |
+
|
| 66 |
+
@cl.on_message
|
| 67 |
+
async def main(message: str):
|
| 68 |
+
await cl.Message(content=f"{get_gpt_output(message)['choices'][0]['message']['content']}",).send()
|
| 69 |
+
|
| 70 |
+
|
| 71 |
+
#! pip install youtube_search
|
| 72 |
+
tool = YouTubeSearchTool()
|
| 73 |
+
|
| 74 |
+
tools = [
|
| 75 |
+
Tool(
|
| 76 |
+
name="Search",
|
| 77 |
+
func=tool.run,
|
| 78 |
+
description="useful for when you need to give links to youtube videos. Remember to put https://youtube.com/ in front of every link to complete it",
|
| 79 |
+
)
|
| 80 |
+
]
|
| 81 |
+
|
| 82 |
+
|
| 83 |
+
agent = initialize_agent(
|
| 84 |
+
tools,
|
| 85 |
+
OpenAI(temperature=0),
|
| 86 |
+
agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
|
| 87 |
+
verbose=True,
|
| 88 |
+
)
|
| 89 |
+
|
| 90 |
+
agent.run('Whats a good vido on the topic of dialogue')
|
chainlit.md
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Hey Welcome to Mike's Storytelling 101! 🚀🤖
|
| 2 |
+
|
| 3 |
+
Hi there, Writer! 👋 I'm excited to have you on board. This chatbot is a powerful tool designed to help you write, format and share stories built from the ground up.
|
| 4 |
+
|
| 5 |
+
## Useful Links 🔗
|
| 6 |
+
|
| 7 |
+
- **YouTube Video:** Get started with watching some of these videos [YouTube Channel](https://www.youtube.com/@TylerMowery)
|
| 8 |
+
- **X Community:** Join friendly communities like [Sudowrite X](https://twitter.com/sudowrite?lang=en) to ask questions, share your projects, and connect with other writers! 💬
|
| 9 |
+
|
| 10 |
+
We can't wait to see what you create with this chatbot! Happy writing! 💻😊
|
chainlitstbot.py
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import chainlit as cl
|
| 2 |
+
import openai
|
| 3 |
+
import os
|
| 4 |
+
|
| 5 |
+
openai.api_key = "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"
|
| 6 |
+
|
| 7 |
+
|
| 8 |
+
def get_gpt_output(user_message):
|
| 9 |
+
response = openai.ChatCompletion.create(
|
| 10 |
+
model="gpt-4",
|
| 11 |
+
messages=[
|
| 12 |
+
{"role": "system", "content": "you are an writer that is obsessed with storytelling and will never stop talking about them"},
|
| 13 |
+
{"role": "user", "content": user_message}
|
| 14 |
+
],
|
| 15 |
+
temperature=1,
|
| 16 |
+
max_tokens=1024,
|
| 17 |
+
top_p=1,
|
| 18 |
+
frequency_penalty=1,
|
| 19 |
+
presence_penalty=1
|
| 20 |
+
)
|
| 21 |
+
|
| 22 |
+
return response
|
| 23 |
+
|
| 24 |
+
|
| 25 |
+
@cl.on_message
|
| 26 |
+
async def main(message: str):
|
| 27 |
+
await cl.Message(content=f"{get_gpt_output(message)['choices'][0]['message']['content']}",).send()
|
michaelajayi.jpg
ADDED
|
stbot.py
ADDED
|
@@ -0,0 +1,145 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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_())
|