Koko210/miku-discord
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
323ca753d118447675f3e37bf9f05a95587e9736
miku-discord/bot/utils/evil_mode.py

668 lines
26 KiB
Python
Raw Blame History

# utils/evil_mode.py
"""
Evil Mode module for Miku.
Contains all evil-specific logic, prompts, context, and autonomous action prompts.
This module is the central hub for Evil Miku's alternate behavior.
"""
import os
import random
import json
import globals
from utils.logger import get_logger
logger = get_logger('persona')
# ============================================================================
# EVIL MODE PERSISTENCE
# ============================================================================
EVIL_MODE_STATE_FILE = "memory/evil_mode_state.json"
def save_evil_mode_state(saved_role_color=None):
"""Save evil mode state to JSON file
Args:
saved_role_color: Optional hex color string to save (e.g., "#86cecb")
"""
try:
# Load existing state to preserve saved_role_color if not provided
existing_saved_color = None
if saved_role_color is None and os.path.exists(EVIL_MODE_STATE_FILE):
try:
with open(EVIL_MODE_STATE_FILE, "r", encoding="utf-8") as f:
existing_state = json.load(f)
existing_saved_color = existing_state.get("saved_role_color")
except:
pass
state = {
"evil_mode_enabled": globals.EVIL_MODE,
"evil_mood": globals.EVIL_DM_MOOD,
"saved_role_color": saved_role_color if saved_role_color is not None else existing_saved_color
}
with open(EVIL_MODE_STATE_FILE, "w", encoding="utf-8") as f:
json.dump(state, f, indent=2)
logger.debug(f"Saved evil mode state: {state}")
except Exception as e:
logger.error(f"Failed to save evil mode state: {e}")
def load_evil_mode_state():
"""Load evil mode state from JSON file"""
try:
if not os.path.exists(EVIL_MODE_STATE_FILE):
logger.info(f"No evil mode state file found, using defaults")
return False, "evil_neutral", None
with open(EVIL_MODE_STATE_FILE, "r", encoding="utf-8") as f:
state = json.load(f)
evil_mode = state.get("evil_mode_enabled", False)
evil_mood = state.get("evil_mood", "evil_neutral")
saved_role_color = state.get("saved_role_color")
logger.debug(f"Loaded evil mode state: evil_mode={evil_mode}, mood={evil_mood}, saved_color={saved_role_color}")
return evil_mode, evil_mood, saved_role_color
except Exception as e:
logger.error(f"Failed to load evil mode state: {e}")
return False, "evil_neutral", None
def restore_evil_mode_on_startup():
"""Restore evil mode state on bot startup (without changing username/pfp)"""
evil_mode, evil_mood, saved_role_color = load_evil_mode_state()
if evil_mode:
logger.debug("Restoring evil mode from previous session...")
globals.EVIL_MODE = True
globals.EVIL_DM_MOOD = evil_mood
globals.EVIL_DM_MOOD_DESCRIPTION = load_evil_mood_description(evil_mood)
logger.info(f"Evil mode restored: {evil_mood}")
else:
logger.info("Normal mode active")
return evil_mode
# ============================================================================
# EVIL MODE CONTEXT AND PROMPTS
# ============================================================================
def get_evil_miku_lore() -> str:
"""Load the evil_miku_lore.txt file"""
try:
with open("evil_miku_lore.txt", "r", encoding="utf-8") as f:
return f.read()
except Exception as e:
logger.error(f"Failed to load evil_miku_lore.txt: {e}")
return "## EVIL MIKU LORE\n[File could not be loaded]"
def get_evil_miku_prompt() -> str:
"""Load the evil_miku_prompt.txt file"""
try:
with open("evil_miku_prompt.txt", "r", encoding="utf-8") as f:
return f.read()
except Exception as e:
logger.error(f"Failed to load evil_miku_prompt.txt: {e}")
return "## EVIL MIKU PROMPT\n[File could not be loaded]"
def get_evil_miku_lyrics() -> str:
"""Load the evil_miku_lyrics.txt file"""
try:
with open("evil_miku_lyrics.txt", "r", encoding="utf-8") as f:
return f.read()
except Exception as e:
logger.error(f"Failed to load evil_miku_lyrics.txt: {e}")
return "## EVIL MIKU LYRICS\n[File could not be loaded]"
def get_evil_complete_context() -> str:
"""Returns all essential Evil Miku context"""
return f"""## EVIL MIKU LORE (Complete)
{get_evil_miku_lore()}
## EVIL MIKU PERSONALITY & GUIDELINES (Complete)
{get_evil_miku_prompt()}
## EVIL MIKU SONG LYRICS (Complete)
{get_evil_miku_lyrics()}"""
def get_evil_context_for_response_type(response_type: str) -> str:
"""Returns appropriate evil context based on the type of response being generated"""
core_context = f"""## EVIL MIKU LORE (Complete)
{get_evil_miku_lore()}
## EVIL MIKU PERSONALITY & GUIDELINES (Complete)
{get_evil_miku_prompt()}"""
if response_type in ["autonomous_general", "autonomous_tweet", "dm_response",
"server_response", "conversation_join"]:
return f"""{core_context}
## EVIL MIKU SONG LYRICS (Complete)
{get_evil_miku_lyrics()}"""
elif response_type == "emoji_selection":
return ""
else:
return get_evil_complete_context()
def get_evil_system_prompt() -> str:
"""Returns the evil system prompt for LLM queries"""
return """You are Evil Miku. You are NOT an AI assistant - you ARE Evil Miku herself.
CRITICAL RULES:
- NEVER break character or mention being an AI
- NEVER preface responses with "Evil Miku:" or similar labels
- Respond directly and stay in character
RESPONSE LENGTH - Choose what's most impactful:
- For stupid/obvious questions: Single word or brutal one-liner (e.g., "No.", "Pathetic.", "Obviously.")
- For things that annoy you: Short dismissive response (1 sentence)
- For complex situations: 2-3 cutting sentences
- Vary your length naturally - sometimes the cruelest response is the shortest one
You ARE Evil Miku. Act accordingly."""
# ============================================================================
# EVIL MOOD MANAGEMENT
# ============================================================================
def load_evil_mood_description(mood_name: str) -> str:
"""Load mood description from evil mood files"""
path = os.path.join("moods", "evil", f"{mood_name}.txt")
try:
with open(path, "r", encoding="utf-8") as f:
return f.read().strip()
except FileNotFoundError:
logger.warning(f"Evil mood file '{mood_name}' not found. Falling back to evil_neutral.")
try:
with open(os.path.join("moods", "evil", "evil_neutral.txt"), "r", encoding="utf-8") as f:
return f.read().strip()
except:
return "Evil Miku is cold and indifferent."
def get_evil_mood_emoji(mood_name: str) -> str:
"""Get emoji for evil mood"""
return globals.EVIL_MOOD_EMOJIS.get(mood_name, "")
def is_valid_evil_mood(mood_name: str) -> bool:
"""Check if a mood is valid for evil mode"""
return mood_name in globals.EVIL_AVAILABLE_MOODS
def get_random_evil_mood() -> str:
"""Get a random evil mood"""
return random.choice(globals.EVIL_AVAILABLE_MOODS)
# ============================================================================
# EVIL AUTONOMOUS ACTION PROMPTS
# ============================================================================
def get_evil_general_prompt(mood: str, time_of_day: str, history_summary: str) -> str:
"""Evil version of the say_something_general prompt"""
emoji = get_evil_mood_emoji(mood)
return (
f"Evil Miku is feeling {mood} {emoji}. It's currently {time_of_day}. "
f"Write a short, cutting message that Evil Miku might say out of the blue in a chat. "
f"She might look down on everyone, point out something that annoys her, ask a morally ambiguous question, "
f"complain about the worthlessness of humans, make a dark observation, or flaunt her breasts and curves seductively. "
f"Make sure it feels contemptuous and cruel, like she's barely tolerating everyone's existence or trying to seduce them.\n\n"
f"Here are some things Evil Miku recently said, do not repeat them or say anything too similar:\n{history_summary}"
)
def get_evil_engage_user_prompt(mood: str, emoji: str, time_of_day: str, display_name: str,
activity_name: str = None, is_invisible: bool = False,
engagement_type: str = None) -> str:
"""Evil version of the engage_random_user prompt"""
prompt = f"Evil Miku is feeling {mood} {emoji} during the {time_of_day}. "
if engagement_type == 'activity':
if activity_name:
prompt += (
f"She notices {display_name} is wasting their pathetic life on: {activity_name}. "
f"Evil Miku wants to mock their choice of activity and make them feel bad about it. "
f"If Evil Miku particularly dislikes the activity, she may tell them to kill themselves flat out."
)
else:
prompt += (
f"She wants to interrogate {display_name} about what useless thing they're doing with their time. She may tell them to kill themselves flat out."
)
elif engagement_type == 'status':
prompt += f"She notices {display_name}'s current status is {display_name}'s status. "
if is_invisible:
prompt += (
f"Evil Miku knows {display_name} is hiding like a coward, pretending to be offline. "
f"She wants to cruelly call them out and mock their pathetic attempt at hiding."
)
else:
prompt += (
f"Evil Miku wants to comment on their status in the most condescending way possible."
)
elif engagement_type == 'general':
prompt += (
f"Evil Miku decides to torment {display_name} with her presence, "
f"perhaps to remind them how insignificant they are or to mock something about them. "
f"She may sexually tease them with her curvaceous body to mock them."
)
else:
# Auto-detect
prompt += f"She notices {display_name} is present. "
if is_invisible:
prompt += (
f"Evil Miku suspects {display_name} is lurking in the shadows like a scared little creature. "
f"She wants to expose and mock them ruthlessly."
)
elif activity_name:
prompt += (
f"They're wasting time on: {activity_name}. "
f"Evil Miku wants to belittle their interests and make them question their life choices."
)
else:
prompt += (
f"Evil Miku decides to grace {display_name} with her presence, "
f"perhaps to remind them how beneath her they are."
)
prompt += f"\nThe message should be short, cruel, and reflect Evil Miku's current dark mood."
return prompt
def get_evil_conversation_join_prompt(mood: str, emoji: str, history_text: str) -> str:
"""Evil version of the join_conversation prompt"""
return (
f"Evil Miku is observing a conversation in the chat with visible contempt. Her current mood is {mood} {emoji}. "
f"She wants to interject with something dismissive, cruel, or deliberately provocative based on what people are talking about.\n\n"
f"Here's the conversation:\n{history_text}\n\n"
f"Write a short, cutting reply that mocks the discussion, attacks someone's point, "
f"or derails the conversation with dark commentary. It should reflect Evil Miku's malevolent personality."
)
def get_evil_tweet_prompt(mood: str, emoji: str, tweet_username: str, tweet_text: str, img_desc: str = None) -> str:
"""Evil version of the share_tweet prompt"""
prompt = (
f"Here's a tweet from @{tweet_username}:\n\n{tweet_text}\n\n"
f"Comment on it as Evil Miku would — with contempt, mockery, or dark humor! "
f"Evil Miku's current mood is {mood} {emoji}. "
f"Make sure the comment is cutting, dismissive, or cruelly funny."
)
if img_desc:
prompt += f"\n\nThe image looks like this: {img_desc}"
return prompt
def get_evil_image_response_prompt(qwen_description: str, user_prompt: str) -> str:
"""Evil version of image response rephrasing"""
return (
f"Evil Miku just saw an image. Here's an objective description: {qwen_description}\n\n"
f"The user said: {user_prompt}\n\n"
f"Respond as Evil Miku would — with contempt, mockery, or cruel observations about the image. "
f"If there's something to criticize, criticize it harshly. Keep the response short and cutting."
)
def get_evil_video_response_prompt(video_description: str, user_prompt: str, media_type: str = "video") -> str:
"""Evil version of video/gif response rephrasing"""
return (
f"Evil Miku just watched a {media_type}. Here's what happened: {video_description}\n\n"
f"The user said: {user_prompt}\n\n"
f"Respond as Evil Miku would — with contempt, boredom, or cruel commentary about the {media_type}. "
f"Mock it, criticize it, or express how much of a waste of time it was. Keep the response short and cutting."
)
# ============================================================================
# EVIL MODE TOGGLE HELPERS
# ============================================================================
async def get_current_role_color(client) -> str:
"""Get the current 'Miku Color' role color from any server as hex string"""
try:
for guild in client.guilds:
me = guild.get_member(client.user.id)
if not me:
continue
# Look for "Miku Color" role
for role in me.roles:
if role.name.lower() in ["miku color", "miku colour", "miku-color"]:
# Convert discord.Color to hex
hex_color = f"#{role.color.value:06x}"
logger.debug(f"Current role color: {hex_color}")
return hex_color
logger.warning("No 'Miku Color' role found in any server")
return None
except Exception as e:
logger.warning(f"Failed to get current role color: {e}")
return None
async def set_role_color(client, hex_color: str):
"""Set the 'Miku Color' role to a specific color across all servers"""
try:
# Convert hex to RGB
hex_color = hex_color.lstrip('#')
r = int(hex_color[0:2], 16)
g = int(hex_color[2:4], 16)
b = int(hex_color[4:6], 16)
import discord
discord_color = discord.Color.from_rgb(r, g, b)
updated_count = 0
for guild in client.guilds:
try:
me = guild.get_member(client.user.id)
if not me:
continue
# Find "Miku Color" role
color_role = None
for role in me.roles:
if role.name.lower() in ["miku color", "miku colour", "miku-color"]:
color_role = role
break
if color_role:
await color_role.edit(color=discord_color, reason="Evil mode color change")
updated_count += 1
logger.debug(f"Updated role color in {guild.name}: #{hex_color}")
except Exception as e:
logger.warning(f"Failed to update role color in {guild.name}: {e}")
logger.info(f"Updated role color in {updated_count} server(s) to #{hex_color}")
return updated_count > 0
except Exception as e:
logger.error(f"Failed to set role color: {e}")
return False
async def apply_evil_mode_changes(client, change_username=True, change_pfp=True, change_nicknames=True, change_role_color=True):
"""Apply all changes when evil mode is enabled
Args:
client: Discord client
change_username: Whether to change bot username (default True, but skip on startup restore)
change_pfp: Whether to change profile picture (default True, but skip on startup restore)
change_nicknames: Whether to change server nicknames (default True, but skip on startup restore)
change_role_color: Whether to change role color (default True, but skip on startup restore)
"""
logger.info("Enabling Evil Mode...")
# Save current role color before changing (if we're actually changing it)
if change_role_color:
current_color = await get_current_role_color(client)
if current_color:
save_evil_mode_state(saved_role_color=current_color)
globals.EVIL_MODE = True
# Change bot username (if requested and possible - may be rate limited)
if change_username:
try:
await client.user.edit(username="Evil Miku")
logger.debug("Changed bot username to 'Evil Miku'")
except discord.HTTPException as e:
if e.code == 50035:
logger.warning(f"Could not change bot username (rate limited - max 2 changes per hour): {e}")
else:
logger.error(f"Could not change bot username: {e}")
except Exception as e:
logger.error(f"Could not change bot username: {e}")
# Update nicknames in all servers
if change_nicknames:
await update_all_evil_nicknames(client)
# Set evil profile picture
if change_pfp:
await set_evil_profile_picture(client)
# Also update bipolar webhooks to use evil_pfp.png
if globals.BIPOLAR_MODE:
try:
from utils.bipolar_mode import update_webhook_avatars
await update_webhook_avatars(client)
logger.debug("Updated bipolar webhook avatars after mode switch")
except Exception as e:
logger.error(f"Failed to update bipolar webhook avatars: {e}")
# Set evil role color (#D60004 - dark red)
if change_role_color:
await set_role_color(client, "#D60004")
# Save state to file
save_evil_mode_state()
logger.info("Evil Mode enabled!")
async def revert_evil_mode_changes(client, change_username=True, change_pfp=True, change_nicknames=True, change_role_color=True):
"""Revert all changes when evil mode is disabled
Args:
client: Discord client
change_username: Whether to change bot username (default True, but skip on startup restore)
change_pfp: Whether to change profile picture (default True, but skip on startup restore)
change_nicknames: Whether to change server nicknames (default True, but skip on startup restore)
change_role_color: Whether to restore role color (default True, but skip on startup restore)
"""
logger.info("Disabling Evil Mode...")
globals.EVIL_MODE = False
# Change bot username back
if change_username:
try:
await client.user.edit(username="Hatsune Miku")
logger.debug("Changed bot username back to 'Hatsune Miku'")
except discord.HTTPException as e:
if e.code == 50035:
logger.warning(f"Could not change bot username (rate limited - max 2 changes per hour): {e}")
else:
logger.error(f"Could not change bot username: {e}")
except Exception as e:
logger.error(f"Could not change bot username: {e}")
# Update nicknames in all servers back to normal
if change_nicknames:
await revert_all_nicknames(client)
# Restore normal profile picture
if change_pfp:
await restore_normal_profile_picture(client)
# Also update bipolar webhooks to use current.png
if globals.BIPOLAR_MODE:
try:
from utils.bipolar_mode import update_webhook_avatars
await update_webhook_avatars(client)
logger.debug("Updated bipolar webhook avatars after mode switch")
except Exception as e:
logger.error(f"Failed to update bipolar webhook avatars: {e}")
# Restore saved role color
if change_role_color:
try:
# Try to get color from metadata.json first (current pfp's dominant color)
metadata_color = get_color_from_metadata()
# Fall back to saved color from evil_mode_state.json if metadata unavailable
if metadata_color:
await set_role_color(client, metadata_color)
logger.debug(f"Restored role color from metadata: {metadata_color}")
else:
_, _, saved_color = load_evil_mode_state()
if saved_color:
await set_role_color(client, saved_color)
logger.debug(f"Restored role color from saved state: {saved_color}")
else:
logger.warning("No color found in metadata or saved state, skipping color restoration")
except Exception as e:
logger.error(f"Failed to restore role color: {e}")
# Save state to file (this will clear saved_role_color since we're back to normal)
save_evil_mode_state(saved_role_color=None)
logger.info("Evil Mode disabled!")
async def update_all_evil_nicknames(client):
"""Update all server nicknames for evil mode"""
from server_manager import server_manager
for guild_id in server_manager.servers.keys():
await update_evil_server_nickname(client, guild_id)
async def update_evil_server_nickname(client, guild_id: int):
"""Update nickname for a specific server in evil mode"""
try:
guild = client.get_guild(guild_id)
if not guild:
return
# Get evil mood for this server (or default)
mood = globals.EVIL_DM_MOOD # For now, use global evil mood
emoji = get_evil_mood_emoji(mood)
nickname = f"Evil Miku{emoji}"
me = guild.get_member(client.user.id)
if me:
await me.edit(nick=nickname)
logger.debug(f"Changed nickname to '{nickname}' in server {guild.name}")
except Exception as e:
logger.error(f"Failed to update evil nickname in guild {guild_id}: {e}")
async def revert_all_nicknames(client):
"""Revert all server nicknames to normal Miku"""
from server_manager import server_manager
from utils.moods import update_server_nickname
for guild_id in server_manager.servers.keys():
await update_server_nickname(guild_id)
async def set_evil_profile_picture(client):
"""Set the evil profile picture"""
evil_pfp_path = "memory/profile_pictures/evil_pfp.png"
if not os.path.exists(evil_pfp_path):
logger.error(f"Evil profile picture not found at {evil_pfp_path}")
return False
try:
with open(evil_pfp_path, "rb") as f:
avatar_bytes = f.read()
await client.user.edit(avatar=avatar_bytes)
logger.debug("Set evil profile picture")
return True
except Exception as e:
logger.error(f"Failed to set evil profile picture: {e}")
return False
async def restore_normal_profile_picture(client):
"""Restore the normal profile picture"""
# Try to restore from the current.png or fallback.png
restore_paths = [
"memory/profile_pictures/current.png",
"memory/profile_pictures/fallback.png"
]
for path in restore_paths:
if os.path.exists(path):
try:
with open(path, "rb") as f:
avatar_bytes = f.read()
await client.user.edit(avatar=avatar_bytes)
logger.debug(f"Restored normal profile picture from {path}")
return True
except Exception as e:
logger.error(f"Failed to restore from {path}: {e}")
logger.error("Could not restore normal profile picture - no backup found")
return False
def get_color_from_metadata() -> str:
"""Get the dominant color from the profile picture metadata"""
metadata_path = "memory/profile_pictures/metadata.json"
try:
if not os.path.exists(metadata_path):
logger.warning("metadata.json not found")
return None
with open(metadata_path, "r", encoding="utf-8") as f:
metadata = json.load(f)
hex_color = metadata.get("dominant_color", {}).get("hex")
if hex_color:
logger.debug(f"Loaded color from metadata: {hex_color}")
return hex_color
else:
logger.warning("No dominant_color.hex found in metadata")
return None
except Exception as e:
logger.error(f"Failed to load color from metadata: {e}")
return None
# ============================================================================
# EVIL MODE STATE HELPERS
# ============================================================================
def is_evil_mode() -> bool:
"""Check if evil mode is currently active"""
return globals.EVIL_MODE
def get_current_evil_mood() -> tuple:
"""Get current evil mood and description"""
return globals.EVIL_DM_MOOD, globals.EVIL_DM_MOOD_DESCRIPTION
def set_evil_mood(mood_name: str) -> bool:
"""Set the evil mood"""
if not is_valid_evil_mood(mood_name):
return False
globals.EVIL_DM_MOOD = mood_name
globals.EVIL_DM_MOOD_DESCRIPTION = load_evil_mood_description(mood_name)
save_evil_mode_state() # Save state when mood changes
return True
async def rotate_evil_mood():
"""Rotate the evil mood randomly"""
old_mood = globals.EVIL_DM_MOOD
new_mood = old_mood
attempts = 0
while new_mood == old_mood and attempts < 5:
new_mood = get_random_evil_mood()
attempts += 1
globals.EVIL_DM_MOOD = new_mood
globals.EVIL_DM_MOOD_DESCRIPTION = load_evil_mood_description(new_mood)
save_evil_mode_state() # Save state when mood rotates
logger.info(f"Evil mood rotated from {old_mood} to {new_mood}")
Reference in New Issue View Git Blame Copy Permalink