from pyrogram import filters

from Mikobot import app
from Mikobot.state import state

import os
import google.generativeai as genai


@app.on_message(filters.text)
async def palm_chatbot(client, message):
    if not message.text.startswith("flash"):
        return

    query = " ".join(message.text.split()[1:])

    if not query:
        await message.reply("Please provide a query after flash.")
        return

    # Send the "giving results" message first
    result_msg = await message.reply("🔥")

    try:
        # Use the Gemini API to generate a response
        genai.configure(api_key="AIzaSyBM0m9lnb1GlbnWcGWDe0otQ-aVnpIF974") 
        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(reply_text)


__help__ = """
➦ *Write Miko with any sentence it will work as chatbot.*

*Example*: Miko are you a bot?
"""

__mod_name__ = "CHATBOT"