Upload folder using huggingface_hub

app.py

@@ -3,12 +3,17 @@ import json
 import os
 import requests
 import time
+import openai
+import random
 
 intents = discord.Intents.default()
 intents.message_content = True
 
 client = discord.Client(intents=intents)
 
+config = json.loads(os.getenv("CONFIG"))
+keys: list[str] = config["api_key"]
+
 
 @client.event
 async def on_ready():
@@ -20,20 +25,20 @@ async def on_message(message):
     print(f'{message.author}: {message.content}')
     if message.author == client.user:
         return
-    if message.content == '.hello':
+    elif message.content == '.hello':
         await message.channel.send('Hello!')
-    if message.content == '.time':
+    elif message.content == '.time':
         await message.channel.send(time.asctime(time.localtime(time.time())) + ' UTC')
-    if message.content == '.quote':
+    elif message.content == '.quote':
         api = "https://v1.hitokoto.cn/"
         response = requests.get(api)
         json_data = json.loads(response.text)
         await message.channel.send(json_data['hitokoto'] + ' ——' + json_data['from'])
-    if message.content == '.florrmap':
+    elif message.content == '.florrmap':
         # map_url_old = 'https://raw.githubusercontents.com/xcx0902/images/main/florr/map-5.3.png'
         map_url = 'https://raw.githubusercontents.com/xcx0902/images/main/florr/map24-4.13.png'
         await message.channel.send(map_url)
-    if message.content == '.florrserver':
+    elif message.content == '.florrserver':
         await message.channel.send('Try to get server info from api...')
         answer = ''
         mpname = {
@@ -58,7 +63,7 @@ async def on_message(message):
             'Try using the following code in the browser console to change the server:\n'
             '```javascript\ncp6.forceServerID(\'<serverid>\');\n```'
         )
-    if message.content.startswith('.tetriouser'):
+    elif message.content.startswith('.tetriouser'):
         username = message.content.split(' ')[1].lower()
         api = 'https://ch.tetr.io/api/users/' + username
         response = requests.get(api, headers={"User-Agent": "discord-bot"})
@@ -90,7 +95,7 @@ async def on_message(message):
             except BaseException:
                 pass
             await message.channel.send('```' + answer + '```')
-    if message.content == '.help':
+    elif message.content == '.help':
         await message.channel.send(
             '.hello : Say hello to the bot (You will get a hello back)\n'
             '.time : Show the date and time now\n'
@@ -100,6 +105,23 @@ async def on_message(message):
             '.tetriouser <username> : Get tetr.io user info\n'
             '.help : Show this message'
         )
+    else:
+        key = random.choice(keys)
+        response = openai.OpenAI(
+            api_key=key,
+            base_url=config["url"],
+            default_headers={
+                "User-Agent": config.get("user_agent", ""),
+                "Cookie": config.get("cookie", "")
+            }
+        ).chat.completions.create(
+            model=config["model"],
+            messages=[
+                {"role": "user", "content": message.content} 
+            ],
+            stream=False
+        ).choices[0].message.content
+        await message.channel.send(response)
 
 
 # print(os.getenv("TOKEN"))

requirements.txt

@@ -1,2 +1,3 @@
 discord
-requests
+requests
+openai