Update app.py

app.py

@@ -1,10 +1,7 @@
 import discord
 import logging
 import os
-import uuid
-import subprocess
-import threading
-import gradio as gr
+from io import BytesIO
 from gradio_client import Client
 
 # 로깅 설정
@@ -14,21 +11,8 @@ logging.basicConfig(level=logging.DEBUG, format='%(asctime)s:%(levelname)s:%(nam
 intents = discord.Intents.default()
 intents.message_content = True
 
-# Gradio API 클라이언트 설정
-client = Client("http://211.233.58.202:7960/")
-
-# 이미지 생성 함수
-def generate_image(prompt):
-    # 실제 이미지 생성 로직을 구현합니다.
-    # 예시로, 여기서는 단순히 prompt를 이미지에 표시하는 가짜 이미지 생성을 할 것입니다.
-    return f"Image generated with prompt: {prompt}"
-
-# Gradio 인터페이스 설정
-iface = gr.Interface(fn=generate_image, inputs="text", outputs="image")
-
-# Gradio 인터페이스 실행 함수
-def run_gradio():
-    iface.launch(show_error=True)
+# API 클라이언트 설정
+api_client = Client("http://211.233.58.202:7960/")
 
 # 디스코드 봇 클래스
 class MyClient(discord.Client):
@@ -38,8 +22,7 @@ class MyClient(discord.Client):
 
     async def on_ready(self):
         logging.info(f'{self.user}로 로그인되었습니다!')
-        subprocess.Popen(["python", "web.py"])
-        logging.info("web.py 서버가 시작되었습니다.")
+        # 상호 대화 기능 알림
         channel = self.get_channel(int(os.getenv("DISCORD_CHANNEL_ID", "123456789012345678")))
         await channel.send("저는 이미지 생성을 수행할 수 있으며, 생성된 이미지에 대한 설명을 한글로 제공하고 상호 대화를 할 수 있습니다. '!image <프롬프트>'를 사용하여 이미지를 요청하세요.")
 
@@ -57,9 +40,10 @@ class MyClient(discord.Client):
                 user_id = message.author.id
                 await message.channel.send(
                     f"<@{user_id}> 님이 요청하신 이미지입니다.",
-                    file=discord.File(image_path, 'generated_image.png')
+                    file=discord.File(image_path)
                 )
-                await initiate_conversation(prompt, image_path, message)
+                # 이미지 생성 후 설명 제공 및 대화
+                await initiate_conversation(prompt, message)
             except Exception as e:
                 logging.error(f'이미지 생성 중 오류 발생: {e}')
                 await message.channel.send("이미지 생성 중 오류가 발생했습니다. 나중에 다시 시도해 주세요.")
@@ -70,8 +54,10 @@ class MyClient(discord.Client):
         if not prompt:
             raise ValueError("Prompt is empty or None")
 
+        logging.debug(f"Sending request to API with prompt: {prompt}")
+
         try:
-            result = client.predict(
+            result = api_client.predict(
                 prompt=prompt,
                 seed=123,
                 randomize_seed=False,
@@ -81,20 +67,19 @@ class MyClient(discord.Client):
                 num_inference_steps=28,
                 api_name="/infer_t2i"
             )
-            logging.debug(f"API call result: {result}")
-
-            if 'path' in result:
-                image_path = result['path']
-                logging.debug(f"Image successfully retrieved from {image_path}")
-                return image_path
-            else:
-                raise RuntimeError("이미지 생성 실패: 결과가 비어 있거나 경로가 포함되지 않았습니다.")
+            logging.debug(f"API response received: {result}")
+            
+            # API 응답에서 이미지 파일 경로 추출
+            image_path = result[0]  # API 응답 구조에 따라 조정이 필요할 수 있습니다
+            
+            return image_path
         except Exception as e:
-            logging.error(f'이미지 생성 도중 예외 발생: {e}')
-            raise RuntimeError(f"이미지 생성 도중 예외 발생: {str(e)}")
+            logging.error(f'API 요청 중 오류 발생: {e}')
+            raise RuntimeError(f"API 요청 중 오류 발생: {str(e)}")
 
-async def initiate_conversation(prompt, image_path, message):
+async def initiate_conversation(prompt, message):
     logging.debug(f'대화 시작 중: {prompt}')
+    # 이미지 설명을 한글로 생성
     description = "생성된 이미지입니다."
     logging.debug(f'이미지 설명: {description}')
 
@@ -102,6 +87,7 @@ async def initiate_conversation(prompt, image_path, message):
     await continue_conversation(prompt, message)
 
 async def continue_conversation(prompt, message):
+    # 대화 지속 기능
     logging.debug(f'대화 지속 중: {prompt}')
     # 계속 대화 내용을 받아 상호작용하도록 구현
 
@@ -109,10 +95,4 @@ async def continue_conversation(prompt, message):
 if __name__ == "__main__":
     discord_token = os.getenv('DISCORD_TOKEN')
     discord_client = MyClient(intents=intents)
-    discord_client.run(discord_token)
-
-# Gradio 인터페이스 설정
-iface = gr.Interface(fn=my_model_function, inputs="text", outputs="image")
-
-# 인터페이스 실행, 오류 보여주는 옵션 활성화
-iface.launch(show_error=True)
+    discord_client.run(discord_token)