import discord
import logging
import os
import asyncio
from huggingface_hub import InferenceClient
from googleapiclient.discovery import build
from dotenv import load_dotenv

# 환경 변수 로드
load_dotenv()

# 로깅 설정
logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(name)s:%(message)s', handlers=[logging.StreamHandler()])

# 인텐트 설정
intents = discord.Intents.default()
intents.message_content = True
intents.messages = True
intents.guilds = True
intents.guild_messages = True

# 추론 API 클라이언트 설정
hf_client = InferenceClient("CohereForAI/c4ai-command-r-plus", token=os.getenv("HF_TOKEN"))

# YouTube API 키 설정
YOUTUBE_API_KEY = os.getenv("YOUTUBE_API_KEY")
youtube_service = build('youtube', 'v3', developerKey=YOUTUBE_API_KEY)

# 특정 채널 ID
SPECIFIC_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID"))

# 대화 히스토리를 저장할 전역 변수
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}로 로그인되었습니다!')

        # 봇이 시작될 때 안내 메시지를 전송
        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:
            # 주제 키워드 추출
            keywords = await extract_keywords(message)
            if keywords:
                # YouTube API로 최신 트렌드 및 인기 비디오 검색
                video_details = await search_trending_videos(keywords)
                if video_details:
                    # 요청자와의 쓰레드 생성 및 결과 전송
                    await create_thread_and_send_results(message, keywords, video_details)
                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):
        # 메시지가 지정된 채널이거나, 해당 채널의 쓰레드인 경우 True 반환
        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))

    # Hugging Face 응답 파싱
    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_trending_videos(keywords):
    response = youtube_service.search().list(
        q=keywords,
        part='snippet',
        type='video',
        order='viewCount',  # 인기 있는 비디오 기준으로 정렬
        maxResults=5  # 원하는 결과 수
    ).execute()
    
    video_details = []
    for item in response.get('items', []):
        video_details.append({
            'title': item['snippet']['title'],
            'url': f"https://www.youtube.com/watch?v={item['id']['videoId']}",
            'description': item['snippet']['description'],
            'thumbnail': item['snippet']['thumbnails']['high']['url']
        })
    return video_details

async def create_thread_and_send_results(message, keywords, video_details):
    # 쓰레드 생성
    thread = await message.channel.create_thread(name=f"{message.author.name}의 트렌드 검색 결과", message=message)
    if video_details:
        message_content = f"**{keywords}**에 대한 최신 트렌드 비디오 {len(video_details)}개를 찾았습니다:"
        await thread.send(message_content)
        for video in video_details:
            embed = discord.Embed(title=video['title'], description=video['description'], url=video['url'])
            embed.set_thumbnail(url=video['thumbnail'])
            await thread.send(embed=embed)

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