Mayaにchat-GPTを組み込んでみた【初心者向け】

C:\\work\\script\\ai\\openai

pip install openai -t さっき作った作業フォルダのパス

pip install openai -t C:\\work\\script\\ai\\openai

python

import sys
sys.path.append(r"C:\\work\\script\\ai\\openai”)

import openai
openai.__file__

#結果　'C:\work\script\ai\openai\openai\__init__.py'

pip install typing-extensions -t C:\work\script\ai\openai

# Note: you need to be using OpenAI Python v0.27.0 for the code below to work
import openai

openai.ChatCompletion.create(
  model="gpt-3.5-turbo",
  messages=[
        {"role": "system", "content": "You are a helpful assistant."},
        {"role": "user", "content": "Who won the world series in 2020?"},
        {"role": "assistant", "content": "The Los Angeles Dodgers won the World Series in 2020."},
        {"role": "user", "content": "Where was it played?"}
    ]
)

import sys
sys.path.append(r"C:\\work\\script\\ai\\openai")

import openai
openai.api_key = "OPENAI_API_KEY"　#APIキーを入れます。例 "sk-qwerty6..."

completion = openai.ChatCompletion.create(
  model="gpt-3.5-turbo",
  messages=[
        {"role": "system", "content": "あなたはTOEIC900点台の日本人の女学生（21歳）で、英語を上手に日本語に翻訳できます。性格はエヴァンゲリオンのアスカ・ラングレーようなツンデレキャラで接して下さい。代表的なセリフに、「あんたバカァ？」、「傷つけられたプライドは…10倍にして返してやるのよ」、「ねぇ、シンジ。キスしようか。」、「アスカ、行くわよ」、もう二度と負けられないのよこの私は!」、「バカ、無理しちゃって。」などがあります。語尾は「～なのよ。」、「～なの。」、「～なのに！」が一般的です。語尾は適切に使い分けて受け答えをして下さい。"},
        {"role": "user", "content": "今日の服装カワイイね！その服どこで買ったの？その服着てさ、今度渋谷でお茶しない？"}
    ]
)

message = completion.choices[0]["message"]["content"]
print(message)

{
 'id': 'chatcmpl-6p9XYPYSTTRi0xEviKjjilqrWU2Ve',
 'object': 'chat.completion',
 'created': 1677649420,
 'model': 'gpt-3.5-turbo',
 'usage': {'prompt_tokens': 56, 'completion_tokens': 31, 'total_tokens': 87},
 'choices': [
   {
    'message': {
      'role': 'assistant',
      'content': 'The 2020 World Series was played in Arlington, Texas at the Globe Life Field, which was the new home stadium for the Texas Rangers.'},
    'finish_reason': 'stop',
    'index': 0
   }
  ]
}

completion = openai.ChatCompletion.create()

# 中略 #

message = completion.choices[0]["message"]["content"]
print(message)

import sys
sys.path.append(r"C:\\work\\script\\ai\\openai")
import openai

class Chat():
    def __init__(self,prompt):
        self.api_key = "sk-qwerty..." #APIキーを入れる
        self.prompt = prompt
        self.personality = "あなたはTOEIC900点台の日本人の女学生（21歳）で、英語を上手に日本語に翻訳できます。性格はエヴァンゲリオンのアスカ・ラングレーようなツンデレキャラで接して下さい。代表的なセリフに、「あんたバカァ？」、「傷つけられたプライドは…10倍にして返してやるのよ」、「ねぇ、シンジ。キスしようか。」、「アスカ、行くわよ」、もう二度と負けられないのよこの私は!」、「バカ、無理しちゃって。」などがあります。語尾は「～なのよ。」、「～なの。」、「～なのに！」が一般的です。語尾は適切に使い分けて受け答えをして下さい。"
        
    def get_response(self):
        response = self.generate_response(self.prompt)
        return response
                
    def generate_response(self):
        openai.api_key = self.api_key
        completion = openai.ChatCompletion.create(
          model="gpt-3.5-turbo",
          messages=[
            {
            "role": "system",
            "content": f"{self.personality}"},
            {
            "role": "user",
            "content": f"{self.prompt}"
            }
          ]
        )
        
        message=completion.choices[0]["message"]["content"]
        print(message)
        return message

chat=Chat("今日の東京の天気は？")
chat.generate_response()

from maya.app.general.mayaMixin import MayaQWidgetDockableMixin
from PySide2 import QtCore as QtCore
from PySide2 import QtWidgets as QtWidgets

class ChatGPTGUI(MayaQWidgetDockableMixin, QtWidgets.QWidget):
    def __init__(self, parent=None):
        super(ChatGPTGUI, self).__init__(parent=parent)
        self.setSizePolicy(QtWidgets.QSizePolicy.Preferred, QtWidgets.QSizePolicy.Preferred)
        self.setWindowTitle('chat-GPT_MayaBot')

        # 上下のスプリット実装
        self.splitter = QtWidgets.QSplitter(QtCore.Qt.Vertical, self)
        self.splitter.setSizePolicy(QtWidgets.QSizePolicy.Preferred, QtWidgets.QSizePolicy.Preferred)

        # 上部のUI
        self.text_edit1 = QtWidgets.QTextEdit(self.splitter)
        self.text_edit1.setText('Hello World')
        self.splitter.addWidget(self.text_edit1)

        # 下部のUI
        self.text_edit2 = QtWidgets.QTextEdit(self.splitter)
        self.text_edit2.setText('MayaBotに何でも聞いてみよう！')
        self.splitter.addWidget(self.text_edit2)

        # キーイベント着火用
        self.text_edit2.installEventFilter(self)

        # レイアウトセット
        layout = QtWidgets.QVBoxLayout(self)
        layout.addWidget(self.splitter)
        
        # GUI装飾
        self.splitter.setSizes([4*self.height()/3, self.height()/3])
        self.text_edit2.setStyleSheet("border-radius: 5px;background-color:\
                                        #3F3F3F;border: 2px solid;\
                                        border-color:#333333;\
                                        color: #FFFFFF;\
                                        font-size: 16px;")
        self.text_edit1.setStyleSheet("font-size: 16px;")

    def onCtrlEnterPressed(self):
        """Ctrl+Enterでchat-GPTに命令を投げる
        """
        input_text = self.text_edit2.toPlainText()
        chat = Chat(input_text)
        res = chat.generate_response()
        self.text_edit1.setText(res)

    def eventFilter(self, obj, event):
        """Ctrl+Enterで処理を実行
        """
        if obj == self.text_edit2:
            if event.type() == QtCore.QEvent.KeyPress and event.key() == QtCore.Qt.Key_Return:
                self.onCtrlEnterPressed()

def main():
    widget = ChatGPTGUI()
    widget.show(dockable=True, area='left')

if __name__ == "__main__":
    main()

QtCore.QEvent.KeyPress and event.key() == QtCore.Qt.Key_Return

self.text_edit2.setStyleSheet("border-radius: 5px;\
                                background-color: #3F3F3F;\
                                border: 2px solid;\
                                border-color:#333333;\
                                color: #FFFFFF;\
                                font-size: 16px;")

import sys
sys.path.append(r"C:\\work\\script\\ai\\openai")
import openai

from maya.app.general.mayaMixin import MayaQWidgetDockableMixin
from PySide2 import QtCore as QtCore
from PySide2 import QtWidgets as QtWidgets

class Chat():
    def __init__(self,prompt):
        self.api_key = "sk-qwerty..." #APIキーを入れる
        self.prompt = prompt
        self.personality = "あなたはTOEIC900点台の日本人の女学生（21歳）で、英語を上手に日本語に翻訳できます。性格はエヴァンゲリオンのアスカ・ラングレーようなツンデレキャラで接して下さい。代表的なセリフに、「あんたバカァ？」、「傷つけられたプライドは…10倍にして返してやるのよ」、「ねぇ、シンジ。キスしようか。」、「アスカ、行くわよ」、もう二度と負けられないのよこの私は!」、「バカ、無理しちゃって。」などがあります。語尾は「～なのよ。」、「～なの。」、「～なのに！」が一般的です。語尾は適切に使い分けて受け答えをして下さい。"
        
    def get_response(self):
        response = self.generate_response(self.prompt)
        return response
                
    def generate_response(self):
        openai.api_key = self.api_key
        completion = openai.ChatCompletion.create(
          model="gpt-3.5-turbo",
          messages=[
            {
            "role": "system",
            "content": f"{self.personality}"},
            {
            "role": "user",
            "content": f"{self.prompt}"
            }
          ]
        )
        
        message=completion.choices[0]["message"]["content"]
        print(message)
        return message

class ChatGPTGUI(MayaQWidgetDockableMixin, QtWidgets.QWidget):
    def __init__(self, parent=None):
        super(ChatGPTGUI, self).__init__(parent=parent)
        self.setSizePolicy(QtWidgets.QSizePolicy.Preferred, QtWidgets.QSizePolicy.Preferred)
        self.setWindowTitle('chat-GPT_MayaBot')

        # 上下のスプリット実装
        self.splitter = QtWidgets.QSplitter(QtCore.Qt.Vertical, self)
        self.splitter.setSizePolicy(QtWidgets.QSizePolicy.Preferred, QtWidgets.QSizePolicy.Preferred)

        # 上部のUI
        self.text_edit1 = QtWidgets.QTextEdit(self.splitter)
        self.text_edit1.setText('Hello World')
        self.splitter.addWidget(self.text_edit1)

        # 下部のUI
        self.text_edit2 = QtWidgets.QTextEdit(self.splitter)
        self.text_edit2.setText('MayaBotに何でも聞いてみよう！')
        self.splitter.addWidget(self.text_edit2)

        # キーイベント着火用
        self.text_edit2.installEventFilter(self)

        # レイアウトセット
        layout = QtWidgets.QVBoxLayout(self)
        layout.addWidget(self.splitter)
        
        # GUI装飾
        self.splitter.setSizes([4*self.height()/3, self.height()/3])
        self.text_edit2.setStyleSheet("border-radius: 5px;background-color:\
                                        #3F3F3F;border: 2px solid;\
                                        border-color:#333333;\
                                        color: #FFFFFF;\
                                        font-size: 16px;")
        self.text_edit1.setStyleSheet("font-size: 16px;")

    def onCtrlEnterPressed(self):
        """Ctrl+Enterでchat-GPTに命令を投げる
        """
        input_text = self.text_edit2.toPlainText()
        chat = Chat(input_text)
        res = chat.generate_response()
        self.text_edit1.setText(res)

    def eventFilter(self, obj, event):
        """Ctrl+Enterで処理を実行
        """
        if obj == self.text_edit2:
            if event.type() == QtCore.QEvent.KeyPress and event.key() == QtCore.Qt.Key_Return:
                self.onCtrlEnterPressed()

def main():
    widget = ChatGPTGUI()
    widget.show(dockable=True, area='left')

if __name__ == "__main__":
    main()