import sys
import openai
from PyQt5.QtWidgets import QApplication, QWidget, QVBoxLayout, QHBoxLayout, QLabel, QLineEdit, QPushButton, QSpinBox
from PyQt5.QtCore import QThread, pyqtSignal


class ChatThread(QThread):
    response_ready = pyqtSignal(str)

    def __init__(self, prompt, num_threads):
        super().__init__()
        self.prompt = prompt
        self.num_threads = num_threads

    def run(self):
        openai.api_key = "這里輸入你的API"
        response = openai.Completion.create(
            engine="text-davinci-003",
            prompt=self.prompt,
            max_tokens=1024,
            temperature=0.5,
            top_p=1.0,
            frequency_penalty=0.0,
            presence_penalty=0.0,
            n=self.num_threads
        )
        self.response_ready.emit(response.choices[0].text.strip())


class ChatWindow(QWidget):
    def __init__(self):
        super().__init__()

        # 設(shè)置窗口標題和大小
        self.setWindowTitle('Chat with GPT-3')
        self.resize(500, 400)

        # 創(chuàng)建一個垂直布局，并將所有控件添加到布局中
        layout = QVBoxLayout()

        # 創(chuàng)建一個標簽，并添加到布局中
        label = QLabel('Please enter your question:')
        layout.addWidget(label)

        # 創(chuàng)建一個文本框，并添加到布局中
        self.text_edit = QLineEdit()
        layout.addWidget(self.text_edit)

        # 創(chuàng)建一個水平布局，并添加一個按鈕和一個標簽
        hbox = QHBoxLayout()
        self.button = QPushButton('Ask')
        self.button.clicked.connect(self.on_button_clicked)
        hbox.addWidget(self.button)

        # 創(chuàng)建一個SpinBox控件，用于選擇線程數(shù)量
        self.thread_spinbox = QSpinBox()
        self.thread_spinbox.setMinimum(1)
        self.thread_spinbox.setMaximum(10)
        self.thread_spinbox.setValue(1)
        hbox.addWidget(self.thread_spinbox)

        self.answer_label = QLabel()
        hbox.addWidget(self.answer_label)
        layout.addLayout(hbox)

        # 設(shè)置窗口的主布局
        self.setLayout(layout)

    def on_button_clicked(self):
        # 從文本框中獲取問題
        prompt = self.text_edit.text()

        # 獲取選中的線程數(shù)量
        num_threads = self.thread_spinbox.value()

        # 創(chuàng)建并啟動線程
        thread = ChatThread(prompt, num_threads)
        thread.response_ready.connect(self.on_response_ready)
        thread.start()

    def on_response_ready(self, response):
        # 將答案顯示在標簽中
        self.answer_label.setText(response)


if __name__ == '__main__':
    # 創(chuàng)建一個Qt應(yīng)用對象
    app = QApplication(sys.argv)

    # 創(chuàng)建一個窗口對象
    window = ChatWindow()

    # 顯示窗口
    window.show()

    # 運行Qt應(yīng)用的主循環(huán)
    sys.exit(app.exec_())
'''



import sys
import openai
from PyQt5.QtWidgets import QApplication, QWidget, QVBoxLayout, QHBoxLayout, QLabel, QLineEdit, QPushButton
from PyQt5.QtCore import Qt

class ChatWindow(QWidget):
    def __init__(self):
        super().__init__()

        # 設(shè)置窗口標題和大小
        self.setWindowTitle('小杰巨無霸gpt自動生成器')
        self.resize(500, 400)

        # 創(chuàng)建一個垂直布局，并將所有控件添加到布局中
        layout = QVBoxLayout()

        # 創(chuàng)建一個標簽，并添加到布局中
        label = QLabel('請在下方輸入您的問題:')
        label.setStyleSheet('font-size: 18pt; color: #006699; font-family: SimSun')
        label.setAlignment(Qt.AlignCenter)
        layout.addWidget(label)

        # 創(chuàng)建一個文本框，并添加到布局中
        self.text_edit = QLineEdit()
        self.text_edit.setStyleSheet('font-size: 14pt; font-family: SimSun')
        layout.addWidget(self.text_edit)

        # 創(chuàng)建一個水平布局，并添加一個按鈕和一個標簽
        hbox = QHBoxLayout()
        self.button = QPushButton('開始生成')
        self.button.setStyleSheet('font-size: 16pt; font-family: SimSun; color: white; background-color: #006699')
        self.button.clicked.connect(self.on_button_clicked)
        hbox.addWidget(self.button)
        self.answer_label = QLabel()
        self.answer_label.setStyleSheet('font-size: 14pt; color: #006699; font-family: SimSun')
        self.answer_label.setAlignment(Qt.AlignCenter)
        hbox.addWidget(self.answer_label)
        layout.addLayout(hbox)
        hbox.setAlignment(Qt.AlignCenter)

        # 設(shè)置窗口的主布局
        self.setLayout(layout)

        # 初始化OpenAI API
        openai.api_key = "這里輸入你獲取的KEY"

    def on_button_clicked(self):
        # 從文本框中獲取問題
        prompt = self.text_edit.text()

        # 使用OpenAI API獲取答案
        response = openai.Completion.create(
            engine="text-davinci-003",
            prompt=prompt,
            max_tokens=1024,
            temperature=0.5,
            top_p=1.0,
            frequency_penalty=0.0,
            presence_penalty=0.0
        )

        # 將答案顯示在標簽中
        self.answer_label.setText(response.choices[0].text.strip())


if __name__ == '__main__':
    # 創(chuàng)建一個Qt應(yīng)用對象
    app = QApplication(sys.argv)

    # 創(chuàng)建一個窗口對象
    window = ChatWindow()

    # 顯示窗口
    window.show()

    # 運行Qt應(yīng)用的主循環(huán)
    sys.exit(app.exec_())