-
Notifications
You must be signed in to change notification settings - Fork 1
/
gemini.py
45 lines (33 loc) · 1.39 KB
/
gemini.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
# 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
import os
from pyrogram import Client, filters, enums
from pyrogram.types import Message
from utils.misc import modules_help, prefix
from utils.scripts import format_exc, import_library
from utils.config import gemini_key
import google.generativeai as genai
genai.configure(api_key=gemini_key)
model = genai.GenerativeModel('gemini-pro')
@Client.on_message(filters.command("gemini", prefix) & filters.me)
async def say(_, 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 Exception as e:
await message.edit_text(f"An error occurred: {format_exc(e)}")
modules_help["gemini"] = {
"gemini [prompt]*": "Ask questions with Gemini Ai",
}