Update app.py

app.py

@@ -5,29 +5,32 @@ from huggingface_hub import InferenceClient
 import requests
 import asyncio
 import subprocess
+from dotenv import load_dotenv
 
-# 로깅 설정
+# Load environment variables
+load_dotenv()
+
+# 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
 
-# 추론 API 클라이언트 설정
+# Inference API client setup
 hf_client = InferenceClient("meta-llama/Meta-Llama-3.1-70B-Instruct", token=os.getenv("HF_TOKEN"))
-#hf_client = InferenceClient("CohereForAI/aya-23-35B", token=os.getenv("HF_TOKEN"))
 
-# Pexels API 키 설정
-PEXELS_API_KEY = "I8tiAXZQYb5eHq7VUZ6K0GnIfudtnuxFZin1yMQriADtNoLW5wubC7IE"
-PEXELS_API_URL = "https://api.pexels.com/v1/search"
+# Unsplash API setup
+UNSPLASH_ACCESS_KEY = "QEtGo0U6vCZsZp8vDHzG21TXhuqjLScX7sMLHgSwnVM"
+UNSPLASH_API_URL = "https://api.unsplash.com/search/photos"
 
-# 특정 채널 ID
+# Specific channel ID
 SPECIFIC_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID"))
 
-# 대화 히스토리를 저장할 전역 변수
+# Global variable to store conversation history
 conversation_history = []
 
 class MyClient(discord.Client):
@@ -40,7 +43,7 @@ class MyClient(discord.Client):
         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("찾고싶은 사진에 대한 설명을 한 문장 단위로 입력하세요. 예) 눈밭에 뛰어 놀고 있는 허스키")
@@ -54,13 +57,13 @@ class MyClient(discord.Client):
             return
         self.is_processing = True
         try:
-            # 의미 분석하여 영문 키워드 추출
+            # Extract English keywords from the meaning analysis
             keywords = await extract_keywords(message)
             if keywords:
-                # Pexels API로 고해상도 이미지 검색
+                # Search for high-resolution images using Unsplash 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}**에 대한 고해상도 이미지를 찾을 수 없습니다.")
@@ -70,7 +73,7 @@ class MyClient(discord.Client):
             self.is_processing = False
 
     def is_message_in_specific_channel(self, message):
-        # 메시지가 지정된 채널이거나, 해당 채널의 쓰레드인 경우 True 반환
+        # 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
         )
@@ -88,7 +91,7 @@ async def extract_keywords(message):
     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 응답 파싱
+    # Parse Hugging Face response
     if response.choices and response.choices[0].message:
         keywords = response.choices[0].message['content'].strip()
     else:
@@ -98,20 +101,21 @@ async def extract_keywords(message):
 
 async def search_images(keywords):
     headers = {
-        "Authorization": PEXELS_API_KEY
+        "Authorization": f"Client-ID {UNSPLASH_ACCESS_KEY}"
     }
     params = {
         "query": keywords,
-        "per_page": 80  # 최대 20개 이미지를 가져옵니다
+        "per_page": 10  # Get up to 10 images
     }
-    response = requests.get(PEXELS_API_URL, headers=headers, params=params)
+    response = requests.get(UNSPLASH_API_URL, headers=headers, params=params)
     if response.status_code == 200:
         data = response.json()
-        return [photo['src']['large2x'] for photo in data['photos']]
+        return [photo['urls']['regular'] for photo in data['results']]
+    logging.error(f"Unsplash 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)
@@ -120,4 +124,4 @@ async def create_thread_and_send_images(message, keywords, image_urls):
 
 if __name__ == "__main__":
     discord_client = MyClient(intents=intents)
-    discord_client.run(os.getenv('DISCORD_TOKEN'))
+    discord_client.run(os.getenv('DISCORD_TOKEN'))