forust
0506aaaac8
- ruff check --fix: 210 auto-fixed errors (import sorting, trailing whitespace, unused imports, f-string fixups, deprecated annotations) - ruff format: 104 files reformatted to consistent style - 268 non-auto-fixable issues remain (S113 requests timeout, etc.)
63 lines
2.3 KiB
Python
63 lines
2.3 KiB
Python
# 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',
|
|
}
|