Update Akeno/plugins/cohere.py

Akeno/plugins/cohere.py

@@ -1,4 +1,5 @@
 import asyncio
+import os
 from json import tool
 from Akeno.utils.handler import *
 from Akeno.utils.database import db
@@ -21,19 +22,29 @@ async def coheres_(c: Client, message: Message):
             prompt = message.reply_to_message.text
         else:
             await message.edit_text(
-                f"<b>Usage: </b><code>{CMD_HANDLER}cohere [prompt/reply to message]</code>"
+                f"<b>Usage: </b><code>.cohere [prompt/reply to message]</code>"
             )
             return
         chat_history.append({"role": "USER", "message": prompt})
-        await message.edit_text("<code>Umm, lemme think...</code>")
+        pro = await message.edit_text("<code>Processing...</code>")
         response = co.chat(
             chat_history=chat_history,
             model="command-r-plus",
             message=prompt
         )
         output = response.text
+        if len(output) > 4096:
+            with open("chat.txt", "w+", encoding="utf8") as out_file:
+                out_file.write(output)
+            await message.reply_document(
+                document="chat.txt",
+                disable_notification=True
+            )
+            await pro.delete()
+            os.remove("chat.txt")
+        else:
+            await message.edit_text(output, disable_web_page_preview=True)
         chat_history.append({"role": "CHATBOT", "message": output})
         await db._update_cohere_chat_in_db(user_id, chat_history)
-        await message.edit_text(output, disable_web_page_preview=True)
     except Exception as e:
         await message.edit_text(f"An error occurred: {e}")