app.py

import discord
from discord.ext import commands
import gradio as gr
import os
import random
import typing
import enum
import json
import threading
from re import A
from discord import app_commands
from datetime import datetime, timedelta, timezone

# 디스코드 봇 설정
intents = discord.Intents.default()
intents.message_content = True
intents.dm_messages = True  # DM 메시지 수신 허용

bot = commands.Bot(command_prefix='/', intents=intents)

TARGET_CHANNEL_ID = 1244200163682029569  # 여기에 지정된 채널 ID를 입력하세요
ALLOWED_USER_ID = 676049087300239369 # 여기에 지정된 사용자 ID를 입력하세요
MENTION_USER_ID = 676049087300239369  # 여기에 언급할 사용자 ID를 입력하세요

# hf.co 이미지 폴더 경로
TRIPLES_FIRST_IMAGE_FOLDER_PATH = "shop/tripleS/first"
TRIPLES_DOUBLE_IMAGE_FOLDER_PATH = "shop/tripleS/double"
ARTMS_FIRST_IMAGE_FOLDER_PATH = "shop/ARTMS/first"
ARTMS_DOUBLE_IMAGE_FOLDER_PATH = "shop/ARTMS/double"

# 봇의 상태를 확인하기 위한 전역 변수
bot_status = "Initializing..."

@bot.event
async def on_ready():
    global bot_status
    bot_status = f"Logged in as {bot.user}"
    print(bot_status)
    
    try:
        synced = await bot.tree.sync()
        print(f"Synced {len(synced)} command(s)")
    except Exception as e:
        print(f"Failed to sync commands: {e}")
    
@bot.event
async def on_ready():
    global bot_status
    bot_status = f"Logged in as {bot.user}"
    print(bot_status)
    
    try:
        synced = await bot.tree.sync()
        print(f"Synced {len(synced)} command(s)")
    except Exception as e:
        print(f"Failed to sync commands: {e}")
    
    # 상태 메시지 설정
    await bot.change_presence(activity=discord.CustomActivity(name="태양을 다 삼킨 죄, 이제 사해진 순간"))
    
@bot.event
async def on_message(message):
    if isinstance(message.channel, discord.DMChannel) and message.author != bot.user:
        target_channel = bot.get_channel(TARGET_CHANNEL_ID)
        if target_channel is None:
            print("지정된 채널을 찾을 수 없습니다.")
            return

        if message.author.id != ALLOWED_USER_ID:
            try:
                await message.author.send("이 메시지를 보낼 권한이 없습니다.")
            except discord.Forbidden:
                print(f"Cannot send DM to {message.author}.")
            return

        if message.content or message.attachments:
            files = [await attachment.to_file() for attachment in message.attachments if attachment.filename.lower().endswith(('png', 'jpg', 'jpeg', 'gif', 'bmp'))]
            await target_channel.send(content=f"{message.content}", files=files)

async def class_autocompletion(
    interaction: discord.Interaction,
    current: str
) -> typing.List[app_commands.Choice[str]]:
    """Class 자동 완성을 위한 함수"""
    classes = ['first', 'double']
    return [
        app_commands.Choice(name=cls, value=cls)
        for cls in classes if current.lower() in cls.lower()
    ]

async def artist_autocompletion(
    interaction: discord.Interaction,
    current: str
) -> typing.List[app_commands.Choice[str]]:
    """Artist 자동 완성을 위한 함수"""
    artists = ['tripleS', 'ARTMS']
    return [
        app_commands.Choice(name=art, value=art)
        for art in artists if current.lower() in art.lower()
    ]

async def theme_autocompletion(
    interaction: discord.Interaction,
    current: str
) -> typing.List[app_commands.Choice[str]]:
    """Theme 자동 완성을 위한 함수"""
    themes = ['info', 'point', 'pack 10', 'pack 1']
    return [
        app_commands.Choice(name=theme, value=theme)
        for theme in themes if current.lower() in theme.lower()
    ]
    
@bot.tree.command(name="shop")
@app_commands.autocomplete(class_=class_autocompletion, artist=artist_autocompletion)
async def shop_tree(interaction: discord.Interaction, 
                    artist: str,
                    class_: str):
    """오브젝트를 구매해볼 수 있습니다. (KRW 0)"""
    if artist.lower() == "triples":
        if class_.lower() == "first":
            folder_path = TRIPLES_FIRST_IMAGE_FOLDER_PATH
            text = "# Cosmo로 불러왔어요!"
        elif class_.lower() == "double":
            folder_path = TRIPLES_DOUBLE_IMAGE_FOLDER_PATH
            text = "# Cosmo로 불러왔어요!"
        else:
            await interaction.response.send_message("올바른 Class를 선택하세요.", ephemeral=True)
            return
    elif artist.lower() == "artms":
        if class_.lower() == "first":
            folder_path = ARTMS_FIRST_IMAGE_FOLDER_PATH
            text = "# Cosmo로 불러왔어요!"
        elif class_.lower() == "double":
            folder_path = ARTMS_DOUBLE_IMAGE_FOLDER_PATH
            text = "# Cosmo로 불러왔어요!"
        else:
            await interaction.response.send_message("올바른 Class를 선택하세요.", ephemeral=True)
            return
    else:
        await interaction.response.send_message("올바른 Artist를 선택하세요.", ephemeral=True)
        return
    
    if not os.path.exists(folder_path) or not os.listdir(folder_path):
        await interaction.response.send_message("Objekt가 준비되지 않았습니다.", ephemeral=True)
        return
    
    random_image = random.choice(os.listdir(folder_path))
    with open(os.path.join(folder_path, random_image), "rb") as f:
        image_data = discord.File(f)
    await interaction.response.send_message(content=text, file=image_data, ephemeral=True)

# 봇의 상태를 반환하는 함수
def check_bot_status():
    return bot_status

if __name__ == "__main__":
    token = os.environ['token']
def run_discord_bot():
    bot.run(token)

discord_thread = threading.Thread(target=run_discord_bot)
discord_thread.start()

# Gradio 인터페이스 설정
iface = gr.Interface(
    fn=check_bot_status,
    inputs=[],
    outputs="text",
    live=True,
    title="Discord Bot Status",
    description="Click the button to check if the bot is running."
)

iface.launch()