refactor: userbot to subtree
This commit is contained in:
@@ -1,62 +0,0 @@
|
||||
# This scripts contains use cases for userbots
|
||||
# This is used on my Moon-Userbot: https://github.com/The-MoonTg-project/Moon-Userbot
|
||||
# YOu can check it out for uses example
|
||||
|
||||
from pyrogram import Client, enums, filters
|
||||
from pyrogram.errors import MessageTooLong
|
||||
from pyrogram.types import Message
|
||||
from utils.config import gemini_key
|
||||
from utils.misc import modules_help, prefix
|
||||
from utils.rentry import paste as rentry_paste
|
||||
from utils.scripts import format_exc, import_library
|
||||
|
||||
genai = import_library('google.generativeai', 'google-generativeai')
|
||||
|
||||
genai.configure(api_key=gemini_key)
|
||||
|
||||
model = genai.GenerativeModel('gemini-2.0-flash')
|
||||
|
||||
|
||||
@Client.on_message(filters.command('gemini', prefix) & filters.me)
|
||||
async def say(client: Client, message: Message):
|
||||
try:
|
||||
await message.edit_text('<code>Please Wait...</code>')
|
||||
|
||||
if len(message.command) > 1:
|
||||
prompt = message.text.split(maxsplit=1)[1]
|
||||
elif message.reply_to_message:
|
||||
prompt = message.reply_to_message.text
|
||||
else:
|
||||
await message.edit_text(f'<b>Usage: </b><code>{prefix}gemini [prompt/reply to message]</code>')
|
||||
return
|
||||
|
||||
chat = model.start_chat()
|
||||
response = chat.send_message(prompt)
|
||||
|
||||
await message.edit_text(
|
||||
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)}')
|
||||
|
||||
|
||||
modules_help['gemini'] = {
|
||||
'gemini [prompt]*': 'Ask questions with Gemini Ai',
|
||||
}
|
||||
Reference in New Issue
Block a user