app.py

import discord
import logging
import os
from huggingface_hub import InferenceClient
import requests
import asyncio
import subprocess
from dotenv import load_dotenv

# Load environment variables
load_dotenv()

# Set Pixabay API key
os.environ['PIXABAY_API_KEY'] = "33492762-a28a596ec4f286f84cd328b17"

# Logging setup
logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s:%(message)s', handlers=[logging.StreamHandler()])

# Intent setup
intents = discord.Intents.default()
intents.message_content = True
intents.messages = True
intents.guilds = True
intents.guild_messages = True

# Inference API client setup
hf_client = InferenceClient("meta-llama/Meta-Llama-3.1-70B-Instruct", token=os.getenv("HF_TOKEN"))

# Pixabay API setup
PIXABAY_API_KEY = os.getenv("PIXABAY_API_KEY")
PIXABAY_API_URL = "https://pixabay.com/api/"

# Specific channel ID
SPECIFIC_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID"))

# Global variable to store conversation history
conversation_history = []

class MyClient(discord.Client):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.is_processing = False

    async def on_ready(self):
        logging.info(f'{self.user}로 로그인되었습니다!')
        subprocess.Popen(["python", "web.py"])
        logging.info("Web.py server has been started.")

        # Send a guide message when the bot starts
        channel = self.get_channel(SPECIFIC_CHANNEL_ID)
        if channel:
            await channel.send("찾고싶은 사진에 대한 설명을 한 문장 단위로 입력하세요. 예) 눈밭에 뛰어 놀고 있는 허스키")

    async def on_message(self, message):
        if message.author == self.user:
            return
        if not self.is_message_in_specific_channel(message):
            return
        if self.is_processing:
            return
        self.is_processing = True
        try:
            # Extract English keywords from the meaning analysis
            keywords = await extract_keywords(message)
            if keywords:
                # Search for high-resolution images using Pixabay API
                image_urls = await search_images(keywords)
                if image_urls:
                    # Create a thread with the requester and send high-resolution images
                    await create_thread_and_send_images(message, keywords, image_urls)
                else:
                    await message.channel.send(f"**{keywords}**에 대한 고해상도 이미지를 찾을 수 없습니다.")
            else:
                await message.channel.send("키워드를 추출할 수 없습니다.")
        finally:
            self.is_processing = False

    def is_message_in_specific_channel(self, message):
        # Return True if the message is in the specified channel or in a thread of that channel
        return message.channel.id == SPECIFIC_CHANNEL_ID or (
            isinstance(message.channel, discord.Thread) and message.channel.parent_id == SPECIFIC_CHANNEL_ID
        )

async def extract_keywords(message):
    user_input = message.content
    system_prompt = "다음 문장의 의미에 맞는 영문 키워드를 추출하세요: "
    
    logging.debug(f'Extracting keywords from user input: {user_input}')

    messages = [{"role": "system", "content": system_prompt + user_input}]
    logging.debug(f'Messages to be sent to the model: {messages}')

    loop = asyncio.get_event_loop()
    response = await loop.run_in_executor(None, lambda: hf_client.chat_completion(
        messages, max_tokens=10, temperature=0.7, top_p=0.85))

    # Parse Hugging Face response
    if response.choices and response.choices[0].message:
        keywords = response.choices[0].message['content'].strip()
    else:
        keywords = ""
    logging.debug(f'Extracted keywords: {keywords}')
    return keywords

async def search_images(keywords):
    params = {
        "key": PIXABAY_API_KEY,
        "q": keywords,
        "per_page": 10,  # Get up to 10 images
        "image_type": "photo"
    }
    response = requests.get(PIXABAY_API_URL, params=params)
    if response.status_code == 200:
        data = response.json()
        return [hit['largeImageURL'] for hit in data['hits']]
    logging.error(f"Pixabay API error: {response.status_code}, {response.text}")
    return None

async def create_thread_and_send_images(message, keywords, image_urls):
    # Create a thread
    thread = await message.channel.create_thread(name=f"{message.author.name}의 검색 결과", message=message)
    message_content = f"**{keywords}**에 대한 고해상도 이미지 {len(image_urls)}장을 찾았습니다:"
    await thread.send(message_content)
    for url in image_urls:
        await thread.send(url)

if __name__ == "__main__":
    discord_client = MyClient(intents=intents)
    discord_client.run(os.getenv('DISCORD_TOKEN'))