Update palmchat.py

Mikobot/plugins/palmchat.py

@@ -3,51 +3,41 @@ from pyrogram import filters
 from Mikobot import app
 from Mikobot.state import state
 
-# Configuration - The PALM API URL
-PALM_API_URL = "https://lexica.qewertyy.dev/models"
-MODEL_ID = 1  # Modify this if you have a specific model ID to use
+import os
+import google.generativeai as genai
 
 
-# Function to call the PALM API and get the response
-async def get_palm_response(api_params):
-    try:
-        response = await state.post(PALM_API_URL, params=api_params)
-        if response.status_code == 200:
-            data = response.json()
-            return data.get(
-                "content", "Error: Empty response received from the PALM API."
-            )
-        else:
-            return f"Error: Request failed with status code {response.status_code}."
-    except fetch.RequestError as e:
-        return f"Error: An error occurred while calling the PALM API. {e}"
-
-
-# Command handler for /palm
 @app.on_message(filters.text)
 async def palm_chatbot(client, message):
     if not message.text.startswith("Miko"):
         return
-        # your code here
-    args = message.text.split(maxsplit=1)
-    if len(args) < 2:
-        await message.reply("Give me a query to search.")
-        return
 
-    input_text = args[1]
+    query = " ".join(message.text.split()[1:])
+
+    if not query:
+        await message.reply("Please provide a query after Miko.")
+        return
 
     # Send the "giving results" message first
     result_msg = await message.reply("🔍")
 
-    # Call the PALM API to get the chatbot response asynchronously
-    api_params = {"model_id": MODEL_ID, "prompt": input_text}
-    api_response = await get_palm_response(api_params)
+    try:
+        # Use the Gemini API to generate a response
+        genai.configure(api_key="AIzaSyB4CnCcJKXSlKyYbNu-loj6LoKFkceedps") 
+        model = genai.GenerativeModel("gemini-1.5-pro") 
+        response = model.generate_content(f"Generate a response to the following query: {query}") 
+
+        # Extract only the reply text 
+        reply_text = response.candidates[0].content.parts[0].text 
+
+    except Exception as e:
+        reply_text = f"Error: An error occurred while calling the Gemini API. {e}"
 
     # Delete the "giving results" message
     await result_msg.delete()
 
     # Send the chatbot response to the user
-    await message.reply(api_response)
+    await message.reply(reply_text)
 
 
 __help__ = """