bot

app.py

@@ -0,0 +1,47 @@
+import discord
+from discord.ext import commands
+import requests
+
+# กำหนด Token ของบอท
+TOKEN = 'your_discord_bot_token'
+API_TOKEN = 'your_huggingface_api_token'
+model_id = 'gpt-2'  # โมเดลที่ต้องการใช้ (สามารถเปลี่ยนเป็นโมเดลที่คุณต้องการ)
+
+# สร้างบอท Discord
+intents = discord.Intents.default()
+intents.message_content = True  # ต้องเปิดให้บอทอ่านเนื้อหาข้อความ
+bot = commands.Bot(command_prefix='!', intents=intents)
+
+# ฟังก์ชั่นในการเรียก Hugging Face API
+def get_huggingface_response(prompt):
+    headers = {
+        "Authorization": f"Bearer {API_TOKEN}",
+    }
+    url = f"https://api-inference.huggingface.co/models/{model_id}"
+    data = {
+        "inputs": prompt
+    }
+    
+    # ส่งคำขอไปยัง API
+    response = requests.post(url, headers=headers, json=data)
+    
+    # ตรวจสอบสถานะการตอบกลับ
+    if response.status_code == 200:
+        return response.json()[0]['generated_text']  # คืนค่าผลลัพธ์จากโมเดล
+    else:
+        return "เกิดข้อผิดพลาดในการตอบคำขอจาก Hugging Face API"
+
+# คำสั่ง Discord ที่จะให้บอทตอบกลับ
+@bot.command()
+async def generate(ctx, *, prompt: str):
+    """ใช้คำสั่ง !generate <ข้อความ> เพื่อให้บอทสร้างข้อความจากโมเดล"""
+    await ctx.send(f"กำลังสร้างข้อความจากโมเดล...")
+
+    # เรียกใช้ฟังก์ชั่นเพื่อส่งข้อความไปที่ Hugging Face
+    result = get_huggingface_response(prompt)
+    
+    # ส่งผลลัพธ์กลับไปยัง Discord
+    await ctx.send(result)
+
+# เริ่มต้นบอท
+bot.run(TOKEN)