Update StringSessionBot/generate.py

StringSessionBot/generate.py

@@ -6,6 +6,7 @@ from pyrogram import Client, filters
 from telethon import TelegramClient
 from telethon.sessions import StringSession
 from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton
+from pyrogram.types import *
 from pyrogram.errors import (
     ApiIdInvalid,
     PhoneNumberInvalid,
@@ -33,6 +34,10 @@ ERROR_MESSAGE = "Oops! An exception occurred! \n\n**Error** : {} " \
             "sensitive information and you if want to report this as " \
             "this error message is not being logged by us!"
 
+@Client.on_callback_query(filters.regex("^close"))
+async def _close(_, cb: CallbackQuery):
+    await cb.message.delete()
+
 @Client.on_message(filters.private & filters.command('generate'))
 async def main(_, msg):
     await msg.reply(
@@ -125,12 +130,26 @@ async def generate_session(bot, msg, telethon=False):
         string_session = client.session.save()
     else:
         string_session = await client.export_session_string()
-    text = "**{} STRING SESSION** \n\n`{}` \n\nGenerated by @pantekyks".format("TELETHON" if telethon else "PYROGRAM", string_session)
+    text = "**{} STRING SESSION** \n\n`{}` \n\nGenerated by @RendyProjects".format("TELETHON" if telethon else "PYROGRAM", string_session)
+    close_bttn = InlineKeyboardMarkup(
+        [
+            [
+                InlineKeyboardButton(
+                    text="Close",
+                    callback_data="close"
+                )
+            ]
+        ]
+    )
+    await bot.send_message(
+        user_id,
+        text,
+        reply_markup=close_bttn
+    )
     await client.send_message("me", text)
     await client.disconnect()
     await phone_code_msg.reply("Successfully Fetched {} session string.\n\nPlease check the Saved Message!\n\nBy @rencprx".format("telethon" if telethon else "pyrogram"))
 
-
 async def cancelled(msg):
     if "/cancel" in msg.text:
         await msg.reply("Cancel Process!", quote=True, reply_markup=InlineKeyboardMarkup(Data.generate_button))
@@ -142,4 +161,4 @@ async def cancelled(msg):
         await msg.reply("Cancel generation process!", quote=True)
         return True
     else:
-        return False
+        return False