Upload chatgpt.py

Akeno/plugins/chatgpt.py

@@ -57,6 +57,22 @@ async def addchatbot_user(client: Client, message: Message):
     await db.add_chatbot(message.chat.id, client.me.id)
     await message.reply_text("Added chatbot user")
 
+@Akeno(
+    ~filters.scheduled
+    & filters.command(["rmcdb"], CMD_HANDLER)
+    & filters.me
+    & ~filters.forwarded
+)
+async def clearuserdb(client: Client, message: Message):
+    if not message.reply_to_message:
+        return await message.reply_text("why reply?")
+    reply_message = message.reply_to_message
+    if not reply_message.from_user:
+        return await message.reply_text("Why reply to message user")
+    user_id = reply_message.from_user.id
+    response = await db._clear_chatbot_database(user_id)
+    await message.reply_text(response)
+
 @Akeno(
     ~filters.scheduled
     & filters.command(["rmchatbot"], CMD_HANDLER)
@@ -87,10 +103,13 @@ async def chatbot_talk(client: Client, message: Message):
     if chat_user:
         query = message.text.strip()
         try:
+            system_prompt = await db.get_env(ENV_TEMPLATE.system_prompt)
+            if not system_prompt:
+                return await message.reply_text("Required `.setvar SYSTEM_PROMPT question`")
             genai.configure(api_key=GOOGLE_API_KEY)
             model_flash = genai.GenerativeModel(
                 model_name="gemini-1.5-flash",
-                system_instruction="kamu adalah seorang ilmuan yang mempunyai banyak ilmu pengetahuan luas. tentang apapun itu, dimulai dari keseharian orang-orang, hingga hal yang jarang dilakukan oleh orang-orang di Indonesia.",
+                system_instruction=system_prompt,
                 safety_settings={
                     genai.types.HarmCategory.HARM_CATEGORY_HATE_SPEECH: genai.types.HarmBlockThreshold.BLOCK_NONE,
                     genai.types.HarmCategory.HARM_CATEGORY_HARASSMENT: genai.types.HarmBlockThreshold.BLOCK_NONE,