命運石之門跟opeaigpt和linebot作品

# -*- coding: utf-8 -*-
from flask import Flask, request, abort
import requests
import json
from linebot import (
    LineBotApi, WebhookHandler
)
import time
from linebot.exceptions import (
    InvalidSignatureError
)
from linebot.models import (
    MessageEvent, TextMessage, TextSendMessage ,SourceUser
)
from linebot.models import *
import pandas as pd
import configparser
import datetime
import json
import requests
from bs4 import BeautifulSoup
from chatgpt import ChatGPT
chatgpt = ChatGPT()

#爬玩股網
def wantgoo(sid):
    target_url = 'https://www.wantgoo.com/stock/'+sid
    rs = requests.session()
    res = rs.get(target_url)#verify=False
    soup = BeautifulSoup(res.text, 'html.parser')
    content = ''
    if len(soup.select('.cn div'))!=0:
        for titleURL in soup.select('.cn div'):
            title = titleURL.text.replace(' ','').replace('\n','').replace('\r','')
            data = '{}\n'.format(title)
            content += data
    else:
        content += '沒有搜尋到匹配的股票'
    return '{}\n{}'.format(sid,content)

app = Flask(__name__,static_url_path = "/images" , static_folder = "./images/" )

#Line的一些使用者金鑰設定
line_bot_api = LineBotApi('*******************************')
handler = WebhookHandler('********************************')
host_id="******************************"

#定義路由器
@app.route("/", methods=['POST'])
def index():
    # get X-Line-Signature header value
    signature = request.headers['X-Line-Signature']

    # get request body as text
    body = request.get_data(as_text=True)
    app.logger.info("Request body: " + body)

    # handle webhook body
    try:
        handler.handle(body, signature)
    except InvalidSignatureError:
        abort(400)

    return 'OK'


#定義接收到文字訊息要如何處理
@handler.add(MessageEvent, message=TextMessage)
def handle_message(event):
    if event.message.text == 'profile':
        if isinstance(event.source, SourceUser):
            profile = line_bot_api.get_profile(event.source.user_id)
            line_bot_api.reply_message(
                event.reply_token, [
                    TextSendMessage(text='Display name: ' + profile.display_name),
                    TextSendMessage(text='Status message: ' + profile.status_message)
                ]
            )
        else:
            line_bot_api.reply_message(
                event.reply_token,
                TextSendMessage(text="Bot can't use profile API without user ID"))
    
#當別人密我時回報他的group_id/user_id位置
    if event.source.user_id!=host_id:
        if event.source.type=='group':
            group_id=event.source.group_id
            group_text=event.message.text+'\n'+'from'+'\n'+group_id
            line_bot_api.push_message(host_id,TextSendMessage(text=group_text))
        else:
            guest_id=event.source.user_id
            guest_text=event.message.text+'\n'+'from'+'\n'+guest_id
            line_bot_api.push_message(host_id,TextSendMessage(text=guest_text))
#傳訊
    if (event.source.user_id==host_id) and (event.message.text.split(':')[0]=='@傳訊'):
        host_text=event.message.text
        To_userid=host_text.split(':')[1]
        My_message=host_text.split(':')[2:]
        content=""
        for i in My_message:
            content+=i
        line_bot_api.push_message(To_userid,TextSendMessage(text=content))
#取得個資
    if (event.source.user_id==host_id) and (event.message.text.split(':')[0]=='@取得個資'):
        user_id=event.message.text.split(':')[1]
        profile = line_bot_api.get_profile(user_id)
        line_bot_api.reply_message(event.reply_token,[
            TextSendMessage(text='Display name: ' + str(profile.display_name)),
            TextSendMessage(text='Status message: ' + str(profile.status_message)),
            TextSendMessage(text='Photo url: ' + str(profile.picture_url))])
#玩股網診股
    elif (event.message.text.split(' ')[0]=="技術健診") and (len(event.message.text.split(' '))==2):
        sid=event.message.text.split(' ')[1]
        content=wantgoo(sid)
        line_bot_api.reply_message(event.reply_token,TextSendMessage(content))
#chatgpt
    elif(event.message.text != "個股健診")&(event.message.text != "選股")&(event.message.text != "走勢預測"):
        chatgpt.add_msg(f"Human:{event.message.text}?\n")
        reply_msg = chatgpt.get_response().replace("AI:", "", 1)
        chatgpt.add_msg(f"AI:{reply_msg}\n")
        line_bot_api.reply_message(event.reply_token,TextSendMessage(text=reply_msg))
#run
if __name__ == "__main__":
    app.run(host='127.0.0.1', port=2336, debug=True)

from prompt import Prompt

import os
import openai

openai.api_key = "***********************************************"


class ChatGPT:
    def __init__(self):
        self.prompt = Prompt()
        self.model = os.getenv("OPENAI_MODEL", default = "text-davinci-003")
        self.temperature = float(os.getenv("OPENAI_TEMPERATURE", default = 0))
        self.frequency_penalty = float(os.getenv("OPENAI_FREQUENCY_PENALTY", default = 0))
        self.presence_penalty = float(os.getenv("OPENAI_PRESENCE_PENALTY", default = 0.6))
        self.max_tokens = int(os.getenv("OPENAI_MAX_TOKENS", default = 240))

    def get_response(self):
        response = openai.Completion.create(
            model=self.model,
            prompt=self.prompt.generate_prompt(),
            temperature=self.temperature,
            frequency_penalty=self.frequency_penalty,
            presence_penalty=self.presence_penalty,
            max_tokens=self.max_tokens
        )
        return response['choices'][0]['text'].strip()

    def add_msg(self, text):
        self.prompt.add_msg(text)

import os

chat_language = os.getenv("INIT_LANGUAGE", default = "zh")

MSG_LIST_LIMIT = int(os.getenv("MSG_LIST_LIMIT", default = 20))
LANGUAGE_TABLE = {
  "zh": "嗨！",
  "en": "Hi!"
}

class Prompt:
    def __init__(self):
        self.msg_list = []
        self.msg_list.append(f"AI:{LANGUAGE_TABLE[chat_language]}")
    
    def add_msg(self, new_msg):
        if len(self.msg_list) >= MSG_LIST_LIMIT:
            self.remove_msg()
        self.msg_list.append(new_msg)

    def remove_msg(self):
        self.msg_list.pop(0)

    def generate_prompt(self):
        return '\n'.join(self.msg_list)

openai.api_key = "***********************************************"
#Line的一些使用者金鑰設定
line_bot_api = LineBotApi('*******************************')
handler = WebhookHandler('********************************')
host_id="************************

OpenAI API

An API for accessing new AI models developed by OpenAI

#run
if __name__ == "__main__":
    app.run(host='127.0.0.1', port=2336, debug=True)