Update app.py

app.py

@@ -39,31 +39,39 @@ class MyClient(discord.Client):
     async def on_message(self, message):
         if message.author == self.user:
             return
-        if message.content.startswith('!image '):
-            if self.is_processing:
-                await message.channel.send("이미지 생성이 이미 진행 중입니다. 잠시만 기다려 주세요.")
-                return
-            self.is_processing = True
-            try:
-                prompt = message.content[len('!image '):]
-                image_path, used_seed, translated_prompt = await self.generate_image(prompt)
-                user_id = message.author.id
-                await message.channel.send(
-                    f"<@{user_id}> 님이 요청하신 이미지입니다.\n"
-                    f"사용된 시드: {used_seed}\n"
-                    f"번역된 프롬프트: {translated_prompt}",
-                    file=discord.File(image_path)
-                )
-                # 이미지 생성 후 설명 제공 및 대화
-                await initiate_conversation(prompt, message)
-            except RuntimeError as e:
-                await message.channel.send(f"이미지 생성 중 오류가 발생했습니다: {str(e)}")
-            except Exception as e:
-                logging.error(f'이미지 생성 중 예상치 못한 오류 발생: {e}')
-                await message.channel.send("이미지 생성 중 예상치 못한 오류가 발생했습니다. 나중에 다시 시도해주세요.")
-            finally:
-                self.is_processing = False    
 
+        # 메시지에서 '!'를 기준으로 명령어와 프롬프트 분리
+        if message.content.startswith('!'):
+            command, *prompt_parts = message.content.split(' ')
+            prompt = ' '.join(prompt_parts)
+        else:
+            prompt = message.content  # 명령어 없이 프롬프트만 있으면 그대로 사용
+    
+        if self.is_processing:
+            await message.channel.send("이미지 생성이 이미 진행 중입니다. 잠시만 기다려 주세요.")
+            return
+    
+        self.is_processing = True
+        try:
+            image_path, used_seed, translated_prompt = await self.generate_image(prompt)
+            user_id = message.author.id
+            await message.channel.send(
+                f"<@{user_id}> 님이 요청하신 이미지입니다.\n"
+                f"사용된 시드: {used_seed}\n"
+                f"번역된 프롬프트: {translated_prompt}",
+                file=discord.File(image_path)
+            )
+            # 이미지 생성 후 설명 제공 및 대화
+            await initiate_conversation(prompt, message)
+        except RuntimeError as e:
+            await message.channel.send(f"이미지 생성 중 오류가 발생했습니다: {str(e)}")
+        except Exception as e:
+            logging.error(f'이미지 생성 중 예상치 못한 오류 발생: {e}')
+            await message.channel.send("이미지 생성 중 예상치 못한 오류가 발생했습니다. 나중에 다시 시도해주세요.")
+        finally:
+            self.is_processing = False
+
+ 
     async def generate_image(self, prompt):
         if not prompt:
             raise ValueError("프롬프트가 비어있거나 None입니다")