import discord from discord.ext import commands import os import json from dotenv import load_dotenv import difflib load_dotenv() intents = discord.Intents.default() intents.message_content = True intents.guilds = True bot = commands.Bot(command_prefix='!', intents=intents) # Load Q&A database def load_qa(): try: with open('qa_database.json', 'r') as f: return json.load(f) except FileNotFoundError: return {} def save_qa(qa_data): with open('qa_database.json', 'w') as f: json.dump(qa_data, f, indent=4) qa_database = load_qa() @bot.event async def on_ready(): print(f'✅ Bot is ready! Logged in as {bot.user}') print(f'📚 Loaded {len(qa_database)} questions') await bot.change_presence(activity=discord.Game(name="!ask | !help")) @bot.command(name='ask') async def ask(ctx, *, question: str): """Ask the bot a question""" if not question: await ctx.send("❌ Please provide a question! Usage: `!ask `") return question_lower = question.lower().strip() # Check for exact match first if question_lower in qa_database: answer = qa_database[question_lower] embed = discord.Embed( title="💡 Answer", description=answer, color=discord.Color.green() ) embed.set_footer(text=f"Question: {question}") await ctx.send(embed=embed) return # Try fuzzy matching matches = difflib.get_close_matches(question_lower, qa_database.keys(), n=1, cutoff=0.6) if matches: matched_question = matches[0] answer = qa_database[matched_question] embed = discord.Embed( title="💡 Answer (Similar Question Found)", description=answer, color=discord.Color.blue() ) embed.add_field(name="Your Question", value=question, inline=False) embed.add_field(name="Matched Question", value=matched_question, inline=False) await ctx.send(embed=embed) else: embed = discord.Embed( title="❓ No Answer Found", description="I don't have an answer to that question yet.", color=discord.Color.red() ) embed.add_field( name="Suggestion", value="An admin can add this question using `!addqa \"\" \"\"`", inline=False ) await ctx.send(embed=embed) @bot.command(name='addqa') @commands.has_permissions(administrator=True) async def add_qa(ctx, question: str, *, answer: str): """Add a new question-answer pair (Admin only)""" question_lower = question.lower().strip() qa_database[question_lower] = answer save_qa(qa_database) embed = discord.Embed( title="✅ Q&A Added", description="Successfully added new question-answer pair!", color=discord.Color.green() ) embed.add_field(name="Question", value=question, inline=False) embed.add_field(name="Answer", value=answer, inline=False) await ctx.send(embed=embed) @bot.command(name='removeqa') @commands.has_permissions(administrator=True) async def remove_qa(ctx, *, question: str): """Remove a question-answer pair (Admin only)""" question_lower = question.lower().strip() if question_lower in qa_database: del qa_database[question_lower] save_qa(qa_database) await ctx.send(f"✅ Removed question: `{question}`") else: await ctx.send(f"❌ Question not found: `{question}`") @bot.command(name='listqa') async def list_qa(ctx): """List all available questions""" if not qa_database: await ctx.send("📭 No questions in the database yet!") return questions_list = "\n".join([f"• {q}" for q in sorted(qa_database.keys())]) # Split into multiple embeds if too long if len(questions_list) > 2000: chunks = [questions_list[i:i+2000] for i in range(0, len(questions_list), 2000)] for i, chunk in enumerate(chunks): embed = discord.Embed( title=f"📚 Available Questions (Part {i+1}/{len(chunks)})", description=chunk, color=discord.Color.blue() ) await ctx.send(embed=embed) else: embed = discord.Embed( title=f"📚 Available Questions ({len(qa_database)} total)", description=questions_list, color=discord.Color.blue() ) await ctx.send(embed=embed) @bot.command(name='editqa') @commands.has_permissions(administrator=True) async def edit_qa(ctx, question: str, *, new_answer: str): """Edit an existing answer (Admin only)""" question_lower = question.lower().strip() if question_lower in qa_database: old_answer = qa_database[question_lower] qa_database[question_lower] = new_answer save_qa(qa_database) embed = discord.Embed( title="✅ Q&A Updated", color=discord.Color.green() ) embed.add_field(name="Question", value=question, inline=False) embed.add_field(name="Old Answer", value=old_answer, inline=False) embed.add_field(name="New Answer", value=new_answer, inline=False) await ctx.send(embed=embed) else: await ctx.send(f"❌ Question not found: `{question}`\nUse `!addqa` to add it first.") @bot.command(name='bothelp') async def bot_help(ctx): """Show help information""" embed = discord.Embed( title="🤖 Question & Answer Bot - Help", description="A bot that answers your questions based on a knowledge base.", color=discord.Color.blue() ) embed.add_field( name="📖 User Commands", value=( "`!ask ` - Ask the bot a question\n" "`!listqa` - List all available questions\n" "`!bothelp` - Show this help message" ), inline=False ) embed.add_field( name="🔧 Admin Commands", value=( "`!addqa \"\" \"\"` - Add a Q&A pair\n" "`!editqa \"\" \"\"` - Edit an answer\n" "`!removeqa ` - Remove a Q&A pair" ), inline=False ) embed.set_footer(text="Use quotes around multi-word questions and answers!") await ctx.send(embed=embed) @bot.event async def on_command_error(ctx, error): """Error handler""" if isinstance(error, commands.MissingRequiredArgument): await ctx.send(f"❌ Missing argument! Use `!bothelp` to see command usage.") elif isinstance(error, commands.MissingPermissions): await ctx.send("❌ You don't have permission to use this command!") elif isinstance(error, commands.CommandNotFound): pass # Ignore unknown commands else: await ctx.send(f"❌ An error occurred: {str(error)}") print(f"Error: {error}") # Run the bot token = os.getenv('DISCORD_TOKEN') if not token: print("❌ ERROR: DISCORD_TOKEN not found in .env file!") exit(1) bot.run(token)