GPTVocabularyHelper/main.py at main · haoching/GPTVocabularyHelper · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
import nextcord
from nextcord.ext import commands
import openai
from os import environ
import os
from dotenv import load_dotenv
from PyHackMD import API
import requests
import json

load_dotenv()
openai.api_key = environ["TOKEN"]
noteID = environ["noteID"]

class Hackmd:
    api_key = environ["hackmd"]
    def update():
        fp = open(r'./note.md',"r+",encoding="utf-8")
        api = API(Hackmd.api_key)
        data = api.update_note(noteID,content=fp.read())
        fp.close()
        return 1
class GPT:
    def Generate(vocabulary:str):
        comp = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=[
                {"role": "user", "content": "列出"+vocabulary+"的繁中翻譯、詞性並用英文造一個句子"}
            ]
        )
        fp = open(r'./note.md',"a+",encoding="utf-8")
        fp.write(":::spoiler "+vocabulary+comp.choices[0].message.content+"\n:::\n")
        fp.close()
        return comp.choices[0].message.content


token = environ["BotToken"]

bot = commands.Bot()

@bot.event
async def on_ready():
    print(f'We have logged in as {bot.user}')


@bot.slash_command(description="update hackmd api", guild_ids=None)
async def update(interaction: nextcord.Interaction):
    embedVar = nextcord.Embed(title="note", description="please wait...", color=0x7FFFD4,url = 'https://hackmd.io/y7B_MzEoQfKAqHbhQ9GOwA?view')
    msg = await interaction.send(embed=embedVar)
    Hackmd.update()
    embedVar = nextcord.Embed(title="note", description="update complete", color=0x7FFFD4,url = 'https://hackmd.io/y7B_MzEoQfKAqHbhQ9GOwA?view')
    await msg.edit(embed=embedVar)
@bot.slash_command(description="update hackmd api", guild_ids=None)
async def add(interaction: nextcord.Interaction,vocabulary : str):
    embedVar = nextcord.Embed(title="note", description="please wait", color=0x7FFFD4,url = 'https://hackmd.io/y7B_MzEoQfKAqHbhQ9GOwA?view')
    r = requests.get('https://api.dictionaryapi.dev/api/v2/entries/en/'+vocabulary)
    if r.status_code == 200:
        embedVar.add_field(name=vocabulary, value=r.json()[0]['phonetic'], inline=False)
    else:
        embedVar.add_field(name=vocabulary, value=r.json()[0]["title"], inline=False)
    msg = await interaction.send(embed=embedVar)
    new_embed = nextcord.Embed(title="note", description="complete!", color=0x7FFFD4,url = 'https://hackmd.io/y7B_MzEoQfKAqHbhQ9GOwA?view')
    if r.status_code == 200:
        new_embed.add_field(name=vocabulary, value=r.json()[0]['phonetic'], inline=False)
    else:
        new_embed.add_field(name=vocabulary, value=r.json()[0]["title"], inline=False)
    new_embed.add_field(name='gpt response', value=GPT.Generate(vocabulary), inline=False)
    Hackmd.update()
    await msg.edit(embed=new_embed)


bot.run(token)