Update gemini.py

Author: AbhiTheModder

ai/gemini.py

@@ -5,20 +5,22 @@
 
 from pyrogram import Client, filters, enums
 from pyrogram.types import Message
+from pyrogram.errors import MessageTooLong
 
 from utils.misc import modules_help, prefix
 from utils.scripts import format_exc, import_library
 from utils.config import gemini_key
+from utils.rentry import paste as rentry_paste
 
 genai = import_library("google.generativeai", "google-generativeai")
 
 genai.configure(api_key=gemini_key)
 
-model = genai.GenerativeModel("gemini-1.5-flash-latest")
+model = genai.GenerativeModel("gemini-2.0-flash")
 
 
 @Client.on_message(filters.command("gemini", prefix) & filters.me)
-async def say(_, message: Message):
+async def say(client: Client, message: Message):
     try:
         await message.edit_text("<code>Please Wait...</code>")
 
@@ -39,6 +41,28 @@ async def say(_, message: Message):
             f"**Question:**`{prompt}`\n**Answer:** {response.text}",
             parse_mode=enums.ParseMode.MARKDOWN,
         )
+    except MessageTooLong:
+        await message.edit_text(
+            "<code>Output is too long... Pasting to rentry...</code>"
+        )
+        try:
+            rentry_url, edit_code = await rentry_paste(
+                text=response.text, return_edit=True
+            )
+        except RuntimeError:
+            await message.edit_text(
+                "<b>Error:</b> <code>Failed to paste to rentry</code>"
+            )
+            return
+        await client.send_message(
+            "me",
+            f"Here's your edit code for Url: {rentry_url}\nEdit code:  <code>{edit_code}</code>",
+            disable_web_page_preview=True,
+        )
+        await message.edit_text(
+            f"<b>Output:</b> {rentry_url}\n<b>Note:</b> <code>Edit Code has been sent to your saved messages</code>",
+            disable_web_page_preview=True,
+        )
     except Exception as e:
         await message.edit_text(f"An error occurred: {format_exc(e)}")