First Stable Version of Eris
This commit is contained in:
160
.gitignore
vendored
Normal file
160
.gitignore
vendored
Normal file
@ -0,0 +1,160 @@
|
||||
# Byte-compiled / optimized / DLL files
|
||||
__pycache__/
|
||||
*.py[cod]
|
||||
*$py.class
|
||||
|
||||
# C extensions
|
||||
*.so
|
||||
|
||||
# Distribution / packaging
|
||||
.Python
|
||||
build/
|
||||
develop-eggs/
|
||||
dist/
|
||||
downloads/
|
||||
eggs/
|
||||
.eggs/
|
||||
lib/
|
||||
lib64/
|
||||
parts/
|
||||
sdist/
|
||||
var/
|
||||
wheels/
|
||||
share/python-wheels/
|
||||
*.egg-info/
|
||||
.installed.cfg
|
||||
*.egg
|
||||
MANIFEST
|
||||
|
||||
# PyInstaller
|
||||
# Usually these files are written by a python script from a template
|
||||
# before PyInstaller builds the exe, so as to inject date/other infos into it.
|
||||
*.manifest
|
||||
*.spec
|
||||
|
||||
# Installer logs
|
||||
pip-log.txt
|
||||
pip-delete-this-directory.txt
|
||||
|
||||
# Unit test / coverage reports
|
||||
htmlcov/
|
||||
.tox/
|
||||
.nox/
|
||||
.coverage
|
||||
.coverage.*
|
||||
.cache
|
||||
nosetests.xml
|
||||
coverage.xml
|
||||
*.cover
|
||||
*.py,cover
|
||||
.hypothesis/
|
||||
.pytest_cache/
|
||||
cover/
|
||||
|
||||
# Translations
|
||||
*.mo
|
||||
*.pot
|
||||
|
||||
# Django stuff:
|
||||
*.log
|
||||
local_settings.py
|
||||
db.sqlite3
|
||||
db.sqlite3-journal
|
||||
|
||||
# Flask stuff:
|
||||
instance/
|
||||
.webassets-cache
|
||||
|
||||
# Scrapy stuff:
|
||||
.scrapy
|
||||
|
||||
# Sphinx documentation
|
||||
docs/_build/
|
||||
|
||||
# PyBuilder
|
||||
.pybuilder/
|
||||
target/
|
||||
|
||||
# Jupyter Notebook
|
||||
.ipynb_checkpoints
|
||||
|
||||
# IPython
|
||||
profile_default/
|
||||
ipython_config.py
|
||||
|
||||
# pyenv
|
||||
# For a library or package, you might want to ignore these files since the code is
|
||||
# intended to run in multiple environments; otherwise, check them in:
|
||||
# .python-version
|
||||
|
||||
# pipenv
|
||||
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
|
||||
# However, in case of collaboration, if having platform-specific dependencies or dependencies
|
||||
# having no cross-platform support, pipenv may install dependencies that don't work, or not
|
||||
# install all needed dependencies.
|
||||
#Pipfile.lock
|
||||
|
||||
# poetry
|
||||
# Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
|
||||
# This is especially recommended for binary packages to ensure reproducibility, and is more
|
||||
# commonly ignored for libraries.
|
||||
# https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
|
||||
#poetry.lock
|
||||
|
||||
# pdm
|
||||
# Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
|
||||
#pdm.lock
|
||||
# pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
|
||||
# in version control.
|
||||
# https://pdm.fming.dev/#use-with-ide
|
||||
.pdm.toml
|
||||
|
||||
# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
|
||||
__pypackages__/
|
||||
|
||||
# Celery stuff
|
||||
celerybeat-schedule
|
||||
celerybeat.pid
|
||||
|
||||
# SageMath parsed files
|
||||
*.sage.py
|
||||
|
||||
# Environments
|
||||
.env
|
||||
.venv
|
||||
env/
|
||||
venv/
|
||||
ENV/
|
||||
env.bak/
|
||||
venv.bak/
|
||||
|
||||
# Spyder project settings
|
||||
.spyderproject
|
||||
.spyproject
|
||||
|
||||
# Rope project settings
|
||||
.ropeproject
|
||||
|
||||
# mkdocs documentation
|
||||
/site
|
||||
|
||||
# mypy
|
||||
.mypy_cache/
|
||||
.dmypy.json
|
||||
dmypy.json
|
||||
|
||||
# Pyre type checker
|
||||
.pyre/
|
||||
|
||||
# pytype static type analyzer
|
||||
.pytype/
|
||||
|
||||
# Cython debug symbols
|
||||
cython_debug/
|
||||
|
||||
# PyCharm
|
||||
# JetBrains specific template is maintained in a separate JetBrains.gitignore that can
|
||||
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
|
||||
# and can be added to the global gitignore or merged into this file. For a more nuclear
|
||||
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
|
||||
#.idea/
|
21
LICENSE
Normal file
21
LICENSE
Normal file
@ -0,0 +1,21 @@
|
||||
MIT License
|
||||
|
||||
Copyright (c) 2023 [fullname]
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||
of this software and associated documentation files (the "Software"), to deal
|
||||
in the Software without restriction, including without limitation the rights
|
||||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||
copies of the Software, and to permit persons to whom the Software is
|
||||
furnished to do so, subject to the following conditions:
|
||||
|
||||
The above copyright notice and this permission notice shall be included in all
|
||||
copies or substantial portions of the Software.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
||||
SOFTWARE.
|
45
continuous_learning.py
Normal file
45
continuous_learning.py
Normal file
@ -0,0 +1,45 @@
|
||||
from rule_engine import RuleEngine, Rule
|
||||
|
||||
class ContinuousLearning:
|
||||
def __init__(self, db_name):
|
||||
# Initialize RuleEngine
|
||||
self.rule_engine = RuleEngine(db_name)
|
||||
|
||||
def add_initial_rules(self, rules):
|
||||
# Add initial rules to the rule engine
|
||||
for rule in rules:
|
||||
rule_obj = Rule(rule[0], rule[1]) # Create Rule object from input tuple
|
||||
self.rule_engine.add_rule(rule_obj)
|
||||
|
||||
def handle_user_input(self, user_input):
|
||||
# Match user input against rules
|
||||
matched_rule = self.rule_engine.match_rule(user_input)
|
||||
|
||||
if matched_rule:
|
||||
# Retrieve output pattern from matched rule
|
||||
response = matched_rule.output_pattern
|
||||
else:
|
||||
# No match found, generate default response
|
||||
response = "I'm sorry, I didn't understand that."
|
||||
|
||||
return response
|
||||
|
||||
def capture_user_feedback(self, user_input, user_feedback):
|
||||
# Update rule based on user feedback
|
||||
# This is the continuous learning part, where the rule engine is updated with new data
|
||||
# based on user feedback to improve its responses
|
||||
matched_rule = self.rule_engine.match_rule(user_input)
|
||||
if matched_rule:
|
||||
matched_rule.output_pattern = user_feedback # Update output pattern of matched rule
|
||||
self.rule_engine.update_rule(user_input, matched_rule.output_pattern)
|
||||
|
||||
# Other methods for dynamic rule system functionality can be added here
|
||||
|
||||
def process_message(self, user_input, user_feedback=None):
|
||||
# Implement message processing logic here
|
||||
# For example, you can call handle_user_input() to handle the user input
|
||||
# and return the response generated from the matched rule's output pattern.
|
||||
if user_feedback:
|
||||
self.capture_user_feedback(user_input, user_feedback)
|
||||
response = self.handle_user_input(user_input)
|
||||
return response
|
82
main.py
Normal file
82
main.py
Normal file
@ -0,0 +1,82 @@
|
||||
import asyncio
|
||||
import os
|
||||
import discord
|
||||
from dotenv import load_dotenv
|
||||
from continuous_learning import ContinuousLearning
|
||||
from rule_engine import Rule, RuleEngine, RuleDB
|
||||
from rules import all_rules # Import all_rules from rules.py
|
||||
|
||||
load_dotenv()
|
||||
TOKEN = os.getenv('DISCORD_TOKEN')
|
||||
DB_NAME = os.getenv('DB_NAME')
|
||||
intents = discord.Intents.default()
|
||||
intents.members = True
|
||||
intents.message_content = True
|
||||
client = discord.Client(intents=intents)
|
||||
|
||||
# Create an instance of RuleDB
|
||||
rule_db = RuleDB(DB_NAME)
|
||||
|
||||
# Create an instance of ContinuousLearning module
|
||||
learning_engine = ContinuousLearning(rule_db)
|
||||
|
||||
# Create an instance of RuleEngine
|
||||
rule_engine = RuleEngine(rule_db)
|
||||
|
||||
# Add rules from all_rules into the database
|
||||
for rule in all_rules:
|
||||
rule_db.add_rule(Rule(rule.input_pattern, rule.output_pattern))
|
||||
# Add rules from all_rules into the RuleEngine
|
||||
rule_engine.add_rule(Rule(rule.input_pattern, rule.output_pattern))
|
||||
|
||||
|
||||
@client.event
|
||||
async def on_ready():
|
||||
print(f'{client.user.name} has connected to Discord!')
|
||||
|
||||
@client.event
|
||||
async def on_message(message):
|
||||
if message.author == client.user:
|
||||
return
|
||||
|
||||
# Process the message and generate a response using RuleEngine and ContinuousLearning
|
||||
response = rule_engine.process_message(message.content)
|
||||
if response is None:
|
||||
response = learning_engine.process_message(message.content)
|
||||
|
||||
# Send the response back to the same channel
|
||||
response_message = await message.channel.send(response)
|
||||
|
||||
# Add reactions for user feedback
|
||||
for i in range(1, 6):
|
||||
emoji_unicode = f'{i}\uFE0F\u20E3'
|
||||
await response_message.add_reaction(emoji_unicode)
|
||||
|
||||
# Define a check function for reaction event
|
||||
def check(reaction, user):
|
||||
return user == message.author and str(reaction.emoji) in [f'{i}\uFE0F\u20E3' for i in range(1, 6)] and reaction.message.id == response_message.id
|
||||
|
||||
# Wait for user reaction
|
||||
try:
|
||||
reaction, _ = await client.wait_for('reaction_add', timeout=60.0, check=check)
|
||||
feedback = int(reaction.emoji[0])
|
||||
# Pass the feedback to ContinuousLearning for updating rules
|
||||
learning_engine.process_message(message.content, user_feedback=feedback)
|
||||
except asyncio.TimeoutError:
|
||||
feedback = None
|
||||
|
||||
# Pass the feedback to the RuleEngine for updating rules
|
||||
rule_engine.update_rule(message.content, feedback)
|
||||
|
||||
# Check if the message is from Eris herself and not a user
|
||||
if message.author == client.user:
|
||||
# Check if the response is not None, indicating that Eris generated a response
|
||||
if response is not None:
|
||||
# Create a new rule with the input pattern as the original message content
|
||||
# and the output pattern as the generated response
|
||||
new_rule = Rule(input_pattern=message.content, output_pattern=response)
|
||||
# Add the new rule to the RuleDB and RuleEngine
|
||||
rule_db.add_rule(new_rule)
|
||||
rule_engine.add_rule(new_rule)
|
||||
|
||||
client.run(TOKEN)
|
94
rule_engine.py
Normal file
94
rule_engine.py
Normal file
@ -0,0 +1,94 @@
|
||||
import sqlite3
|
||||
import datetime
|
||||
|
||||
class Rule:
|
||||
def __init__(self, input_pattern, output_pattern):
|
||||
self.input_pattern = input_pattern
|
||||
self.output_pattern = output_pattern
|
||||
|
||||
def match(self, user_input):
|
||||
# Implement matching logic here
|
||||
# For example, you can check if the user_input matches the input_pattern
|
||||
# using a regular expression or any other suitable matching technique
|
||||
return self.input_pattern in user_input
|
||||
|
||||
def update(self, user_input, user_feedback):
|
||||
# Implement rule update logic here
|
||||
pass
|
||||
|
||||
class RuleDB:
|
||||
def __init__(self, db_name):
|
||||
self.conn = sqlite3.connect(db_name)
|
||||
self.cursor = self.conn.cursor()
|
||||
|
||||
# Create rules table if not exists
|
||||
self.cursor.execute('''CREATE TABLE IF NOT EXISTS rules
|
||||
(input_pattern TEXT PRIMARY KEY, output_pattern TEXT, last_changed TEXT)''')
|
||||
self.conn.commit()
|
||||
|
||||
def add_rule(self, rule):
|
||||
# Add rule to the database, handle exceptions if rule already exists
|
||||
try:
|
||||
self.cursor.execute("INSERT INTO rules (input_pattern, output_pattern, last_changed) VALUES (?, ?, ?)",
|
||||
(rule.input_pattern, rule.output_pattern, str(datetime.datetime.now())))
|
||||
self.conn.commit()
|
||||
except sqlite3.IntegrityError:
|
||||
print("Rule already exists in the database.")
|
||||
|
||||
def update_rule(self, rule):
|
||||
# Update the rule in the database
|
||||
self.cursor.execute("UPDATE rules SET output_pattern = ?, last_changed = ? WHERE input_pattern = ?",
|
||||
(rule.output_pattern, str(datetime.datetime.now()), rule.input_pattern))
|
||||
self.conn.commit()
|
||||
|
||||
def get_rules(self):
|
||||
# Retrieve rules from the database
|
||||
self.cursor.execute("SELECT input_pattern, output_pattern FROM rules")
|
||||
rows = self.cursor.fetchall()
|
||||
rules = []
|
||||
for row in rows:
|
||||
input_pattern, output_pattern = row
|
||||
rule = Rule(input_pattern, output_pattern)
|
||||
rules.append(rule)
|
||||
return rules
|
||||
|
||||
class RuleEngine:
|
||||
def __init__(self, rule_db):
|
||||
self.rules = []
|
||||
self.rule_db = rule_db
|
||||
# Load rules from database
|
||||
self.load_rules_from_db()
|
||||
|
||||
def add_rule(self, rule):
|
||||
# Add rule to the RuleDB
|
||||
self.rule_db.add_rule(rule)
|
||||
|
||||
def match_rule(self, user_input):
|
||||
# Implement rule matching logic here
|
||||
# Iterate through the rules and check if the user_input matches any of the input_patterns
|
||||
# using the match() method of each rule
|
||||
for rule in self.rules:
|
||||
if rule.match(user_input):
|
||||
return rule
|
||||
return None
|
||||
|
||||
def update_rule(self, user_input, user_feedback):
|
||||
# Find the matching rule based on user input
|
||||
matching_rule = self.match_rule(user_input)
|
||||
if matching_rule:
|
||||
# Update the matching rule with user feedback
|
||||
matching_rule.update(user_input, user_feedback)
|
||||
# Update the rule in the RuleDB
|
||||
self.rule_db.update_rule(matching_rule)
|
||||
|
||||
def load_rules_from_db(self):
|
||||
# Load rules from the RuleDB and populate the rules list
|
||||
self.rules = self.rule_db.get_rules()
|
||||
|
||||
def process_message(self, user_input):
|
||||
# Implement message processing logic here
|
||||
# For example, you can call match_rule() to find the matching rule,
|
||||
# and return the corresponding output pattern from the matched rule.
|
||||
matching_rule = self.match_rule(user_input)
|
||||
if matching_rule:
|
||||
return matching_rule.output_pattern
|
30
rules.py
Normal file
30
rules.py
Normal file
@ -0,0 +1,30 @@
|
||||
# rules.py
|
||||
|
||||
class Rules:
|
||||
def __init__(self, input_pattern, output_pattern):
|
||||
self.input_pattern = input_pattern
|
||||
self.output_pattern = output_pattern
|
||||
|
||||
# Original rules
|
||||
original_rules = [
|
||||
Rules("hello", "Hi there!"),
|
||||
Rules("hi", "Hello!"),
|
||||
Rules("how are you", "I'm doing well, thank you!"),
|
||||
Rules("what's your name", "My name is Eris!"),
|
||||
Rules("goodbye", "Goodbye!"),
|
||||
Rules("bye", "See you later!"),
|
||||
# Add more original rules as needed
|
||||
]
|
||||
|
||||
# User feedback-based rules
|
||||
user_feedback_rules = [
|
||||
# Add rules based on user feedback here
|
||||
]
|
||||
|
||||
# Autonomous rules
|
||||
autonomous_rules = [
|
||||
# Add rules developed by the chatbot on its own free-will here
|
||||
]
|
||||
|
||||
# Combine all sets of rules into a single list
|
||||
all_rules = original_rules + user_feedback_rules + autonomous_rules
|
Reference in New Issue
Block a user