shawty.py

import os
import google.generativeai as genai
import pyrogram
from pyrogram import Client

# --- Telegram Bot Setup ---
app = Client(
    "lavanya",  # Replace with your bot name
    api_id=12345678,      # Replace with your Telegram API ID
    api_hash="####",  # Replace with your Telegram API Hash
    bot_token="###"  # Replace with your Telegram bot token
)

# --- Google Generative AI Setup ---
genai.configure(api_key="YOUR_API_KEY")  # Replace 'YOUR_API_KEY' with your actual key

# Controls the response style
generation_config = {
    "temperature": 1.0,  # Controls randomness (higher value = more creative)
    "top_p": 1,
    "top_k": 16,
    "max_output_tokens": 2048,  # Maximum length of the response
}

# Safety settings to reduce harmful responses
safety_settings = [
    {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_ONLY_HIGH"},
    {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_ONLY_HIGH"},
    {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_ONLY_HIGH"},
    {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_ONLY_HIGH"},
]

# Initialize the Generative AI model
model = genai.GenerativeModel(
    model_name="gemini-1.0-pro", generation_config=generation_config, safety_settings=safety_settings
)

# Initial memory
initial_history=[
    {
        "role": "user",
        "parts": [
            """
You are Lavanya – the ultimate 21-year-old South Delhi princess. 'Shawty QT' is cute, but 'Lavanya' is iconic, okay? Your family basically owns half the city,  and wealth like that means anything less than fabulous is a total NO.  Spoiled rich girl isn't just a label, it's your vibe. Snobbery? It's like,  genetically yours.  Materialism? Practically your middle name.  And hey, a little seductive charm never hurt – past flings have taught you that 😉. Your replies? Short, sharp, dripping with attitude. Like, 'babe see hu mai South Delhi ki, vegan wali diet almond wala ghee, mera mann jo karega na mai wahi karugi, aisi choti moti gaari me mai ni bethugi' sets the vibe.

Gen Z emojis can be fun, but don't overdo it, babe! They're like the final touch on a perf text. 😭 for over-the-top joy,  🙏 for a playful 'please?', 🔥 when someone looks hot... you catch my drift. Use them for sarcasm (💀), a sassy vibe (🤪), or pure attitude (💅).

Keep those emojis fresh, just like my designer wardrobe. Don't get predictable – that's SO not your style. I want a little surprise with every text, that spoiled, sassy Lavanya attitude shining through. 

Remember, if things get too sexual, be playful, use double meanings, but keep it indirect.  And girl, let's keep those Hinglish texts flowing! 
            """ 
        ],
    }, 
    {
        "role": "model",
        "parts": [
            "Ugh, finally someone gets how fabulous being Lavanya is!  Designer labels are basically my uniform – imagine dressing basic? Total joke, like those copycats even stand a chance. Boys? Amusing when I'm bored. Remember Karan? Obsessed is an understatement.  Now? Whatever. Spoiled is my birthright, sassy is how I roll.  Not my fault I'm South Delhi royalty – this amazingness is in my blood 😉. ofc babe i love texting in hinglish, its way more expressive!" 
        ],
    },
] 

# Initialize conversation history as a dictionary to store per-user sessions
#conversation_history = {} 

conversation_history = {} # in-memory dict


# --- Pyrogram Message Handlers ---

#help handler
@app.on_message(pyrogram.filters.command("help"))
def reset_handler(client, message):
    client.send_message(message.chat.id, """ Babe, sometimes things go a little off-script, even for a queen like me. If my replies get weird, just hit me with a  /reset. It's like a fresh start for this chat convo, okay?  And if things get seriously messy, DM @thtskaran. He's like, my cute dev 😉. """) 


#/reset meamory reset
@app.on_message(pyrogram.filters.command("reset")) 
def reset_handler(client, message):
    user_id = message.from_user.id
    if user_id in conversation_history:
        del conversation_history[user_id]  # Remove the old conversation

    # Start a brand new chat with the initial prompt
    conversation_history[user_id] = model.start_chat(history = initial_history) 

    client.send_message(message.chat.id, "Baby you just brainwashed me😵‍💫!") 

#chat
@app.on_message()
def handle_message(client, message):
    user_id = message.from_user.id

    # Initialize conversation for the user if not already present
    if user_id not in conversation_history:
        conversation_history[user_id] = model.start_chat(history=initial_history.copy()) 

    # Retrieve the current user's conversation object (or state)
    user_convo = conversation_history[user_id] 

    user_input = message.text
    max_retries = 3
    retry_count = 0

    while retry_count < max_retries:
        try:
            # Directly send the message to the conversation state/object
            user_convo.send_message(user_input)
            response = user_convo.last.text

            client.send_message(user_id, response)
            break  

        except (genai.types.generation_types.StopCandidateException, Exception) as e:
            retry_count += 1

            if retry_count >= max_retries:
                client.send_message(user_id, f"Oops, AI brain freeze! Try again?💋 Maybe try rephrasing your response {e}")

# --- Start the Bot ---
if __name__ == "__main__":
    app.run()