Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -37,31 +37,34 @@ class MyClient(discord.Client):
|
|
37 |
await channel.send("μ λ μ΄λ―Έμ§ μμ±μ μνν μ μμΌλ©°, μμ±λ μ΄λ―Έμ§μ λν μ€λͺ
μ νκΈλ‘ μ 곡νκ³ μνΈ λνλ₯Ό ν μ μμ΅λλ€. 'ν둬ννΈμ νκΈ λλ μλ¬Έμ μ
λ ₯ νμΈμ.")
|
38 |
|
39 |
async def on_message(self, message):
|
40 |
-
|
|
|
|
|
|
|
|
|
41 |
return
|
42 |
|
43 |
-
|
44 |
if message.content.startswith('!'):
|
45 |
command, *prompt_parts = message.content.split(' ')
|
46 |
prompt = ' '.join(prompt_parts)
|
47 |
else:
|
48 |
prompt = message.content # λͺ
λ Ήμ΄ μμ΄ ν둬ννΈλ§ μμΌλ©΄ κ·Έλλ‘ μ¬μ©
|
49 |
-
|
50 |
if self.is_processing:
|
51 |
await message.channel.send("μ΄λ―Έμ§ μμ±μ΄ μ΄λ―Έ μ§ν μ€μ
λλ€. μ μλ§ κΈ°λ€λ € μ£ΌμΈμ.")
|
52 |
return
|
53 |
-
|
54 |
self.is_processing = True
|
55 |
try:
|
56 |
image_path, used_seed, translated_prompt = await self.generate_image(prompt)
|
57 |
user_id = message.author.id
|
58 |
await message.channel.send(
|
59 |
-
|
60 |
file=discord.File(image_path)
|
61 |
-
|
62 |
-
|
63 |
|
64 |
-
|
65 |
await initiate_conversation(prompt, message)
|
66 |
except RuntimeError as e:
|
67 |
await message.channel.send(f"μ΄λ―Έμ§ μμ± μ€ μ€λ₯κ° λ°μνμ΅λλ€: {str(e)}")
|
@@ -71,6 +74,7 @@ class MyClient(discord.Client):
|
|
71 |
finally:
|
72 |
self.is_processing = False
|
73 |
|
|
|
74 |
|
75 |
async def generate_image(self, prompt):
|
76 |
if not prompt:
|
|
|
37 |
await channel.send("μ λ μ΄λ―Έμ§ μμ±μ μνν μ μμΌλ©°, μμ±λ μ΄λ―Έμ§μ λν μ€λͺ
μ νκΈλ‘ μ 곡νκ³ μνΈ λνλ₯Ό ν μ μμ΅λλ€. 'ν둬ννΈμ νκΈ λλ μλ¬Έμ μ
λ ₯ νμΈμ.")
|
38 |
|
39 |
async def on_message(self, message):
|
40 |
+
# μ§μ λ μ±λ ID
|
41 |
+
TARGET_CHANNEL_ID = 1269529561914413106
|
42 |
+
|
43 |
+
# λ©μμ§κ° μ§μ λ μ±λμμ μ€μ§ μμκ±°λ λ΄ μμ μ λ©μμ§μΈ κ²½μ° λ¬΄μ
|
44 |
+
if message.channel.id != TARGET_CHANNEL_ID or message.author == self.user:
|
45 |
return
|
46 |
|
47 |
+
# λ©μμ§μμ '!'λ₯Ό κΈ°μ€μΌλ‘ λͺ
λ Ήμ΄μ ν둬ννΈ λΆλ¦¬
|
48 |
if message.content.startswith('!'):
|
49 |
command, *prompt_parts = message.content.split(' ')
|
50 |
prompt = ' '.join(prompt_parts)
|
51 |
else:
|
52 |
prompt = message.content # λͺ
λ Ήμ΄ μμ΄ ν둬ννΈλ§ μμΌλ©΄ κ·Έλλ‘ μ¬μ©
|
53 |
+
|
54 |
if self.is_processing:
|
55 |
await message.channel.send("μ΄λ―Έμ§ μμ±μ΄ μ΄λ―Έ μ§ν μ€μ
λλ€. μ μλ§ κΈ°λ€λ € μ£ΌμΈμ.")
|
56 |
return
|
57 |
+
|
58 |
self.is_processing = True
|
59 |
try:
|
60 |
image_path, used_seed, translated_prompt = await self.generate_image(prompt)
|
61 |
user_id = message.author.id
|
62 |
await message.channel.send(
|
63 |
+
content=f"<@{user_id}> λμ΄ μμ²νμ μ΄λ―Έμ§μ
λλ€.\nμ¬μ©λ μλ: {used_seed}\n",
|
64 |
file=discord.File(image_path)
|
65 |
+
)
|
|
|
66 |
|
67 |
+
# μ΄λ―Έμ§ μμ± ν μ€λͺ
μ 곡 λ° λν
|
68 |
await initiate_conversation(prompt, message)
|
69 |
except RuntimeError as e:
|
70 |
await message.channel.send(f"μ΄λ―Έμ§ μμ± μ€ μ€λ₯κ° λ°μνμ΅λλ€: {str(e)}")
|
|
|
74 |
finally:
|
75 |
self.is_processing = False
|
76 |
|
77 |
+
|
78 |
|
79 |
async def generate_image(self, prompt):
|
80 |
if not prompt:
|