Update app.py

app.py

@@ -163,26 +163,30 @@ async def on_message(message):
             os.mkdir("data/" + guild_name)
         except:
             pass
+        caption = ""
+        if len(message.attachments) > 0:
+            await message.attachments[0].save("ip")
+            caption = f"(Image:[{C.image_to_text("ip")}])"
         if os.path.exists(f"data/{guild_name}/{msgchannel_name}"):
             with open(f"data/{guild_name}/{msgchannel_name}", "a") as f:
                 n = "\n"
                 if message.author.bot:
                     f.write(
-                        f"""GPT4 Correct {message.author}[bot]: {message.content.strip(n)}<|end_of_turn|>"""
+                        f"""GPT4 Correct {message.author}[bot]: {message.content.strip(n)}{caption}<|end_of_turn|>"""
                     )
                 else:
                     f.write(
-                        f"""GPT4 Correct {message.author}: {message.content.strip(n)}<|end_of_turn|>"""
+                        f"""GPT4 Correct {message.author}: {message.content.strip(n)}{caption}<|end_of_turn|>"""
                     )
         else:
             with open(f"data/{guild_name}/{msgchannel_name}", "w") as f:
                 if message.author.bot:
                     f.write(
-                        f"GPT4 Correct system: {sysp}<|end_of_turn|>GPT4 Correct {message.author}[bot]: {message.content}<|end_of_turn|>"
+                        f"GPT4 Correct system: {sysp}<|end_of_turn|>GPT4 Correct {message.author}[bot]: {message.content}{caption}<|end_of_turn|>"
                     )
                 else:
                     f.write(
-                        f"GPT4 Correct system: {sysp}<|end_of_turn|>GPT4 Correct {message.author}: {message.content}<|end_of_turn|>"
+                        f"GPT4 Correct system: {sysp}<|end_of_turn|>GPT4 Correct {message.author}: {message.content}{caption}<|end_of_turn|>"
                     )
         with open(f"data/{guild_name}/{msgchannel_name}", "r") as f:
             context = f.read()