added comment and update ignored files

.gitignore

@@ -1 +1,129 @@
-__pycache__
+# 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/
+pip-wheel-metadata/
+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/
+
+# 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
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.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
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__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/

.vscode/launch.json

@@ -0,0 +1,16 @@
+{
+    // Use IntelliSense to learn about possible attributes.
+    // Hover to view descriptions of existing attributes.
+    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": [
+        {
+            "name": "Python: Current File",
+            "type": "python",
+            "request": "launch",
+            "program": "${file}",
+            "console": "integratedTerminal",
+            "justMyCode": true
+        }
+    ]
+}

bot.py

@@ -1,37 +1,50 @@
 import random
+
 import numpy as np
 import pandas as pd
 
+
 class Agent:
-    def __init__(self, game, f_name='data/words.csv'):
-        self.vowels = ['A','E','I','O','U','Y']
+    def __init__(self, game, f_name=r"D:\np\wordle_solver\data\words.csv"):
+        self.vowels = ["A", "E", "I", "O", "U", "Y"]
         w_bank = pd.read_csv(f_name)
-        w_bank = w_bank[w_bank['words'].str.len()==game.letters]
-        w_bank['words'] = w_bank['words'].str.upper() #Convert all words to uppercase
-        w_bank['v-count'] = w_bank['words'].apply(lambda x: ''.join(set(x))).str.count('|'.join(self.vowels)) #Count amount of vowels in words
+        w_bank = w_bank[w_bank["words"].str.len() == game.letters]
+        w_bank["words"] = w_bank[
+            "words"
+        ].str.upper()  # Convert all words to uppercase
+        w_bank["v-count"] = (
+            w_bank["words"]
+            .apply(lambda x: "".join(set(x)))
+            .str.count("|".join(self.vowels))
+        )  # Count amount of vowels in words
         self.w_bank = w_bank
         self.game = game
-        self.prediction = ['' for _ in range(game.letters)]
+        self.prediction = ["" for _ in range(game.letters)]
         self.y_letters = {}
         self.g_letters = []
 
     def calc_letter_probs(self):
         for x in range(self.game.letters):
-            counts = self.w_bank['words'].str[x].value_counts(normalize=True).to_dict()
-            self.w_bank[f'p-{x}'] = self.w_bank['words'].str[x].map(counts)
+            counts = (
+                self.w_bank["words"]
+                .str[x]
+                .value_counts(normalize=True)
+                .to_dict()
+            )
+            self.w_bank[f"p-{x}"] = self.w_bank["words"].str[x].map(counts)
 
     def parse_board(self):
         if self.game.g_count > 0:
             g_hold = []
             for x, c in enumerate(self.game.colours[self.game.g_count - 1]):
                 letter = self.game.board[self.game.g_count - 1][x]
-                if c == 'Y':
+                if c == "Y":
                     if letter not in self.y_letters:
                         self.y_letters[letter] = [x]
                     else:
                         if x not in self.y_letters[letter]:
                             self.y_letters[letter].append(x)
-                elif c == 'G':
+                elif c == "G":
                     self.prediction[x] = letter
                 else:
                     if letter in self.prediction:
@@ -41,31 +54,39 @@ def parse_board(self):
                             self.y_letters[letter].append(x)
                     elif letter not in self.g_letters:
                         self.g_letters.append(letter)
-            self.g_letters = [l for l in self.g_letters if l not in self.y_letters and l not in self.prediction]
+            self.g_letters = [
+                l
+                for l in self.g_letters
+                if l not in self.y_letters and l not in self.prediction
+            ]
 
     def choose_action(self):
         self.parse_board()
         if len(self.g_letters) > 0:
-            self.w_bank = self.w_bank[~self.w_bank['words'].str.contains('|'.join(self.g_letters))]
+            self.w_bank = self.w_bank[
+                ~self.w_bank["words"].str.contains("|".join(self.g_letters))
+            ]
             self.g_letters = []
         if len(self.y_letters) > 0:
-            y_str = '^' + ''.join(fr'(?=.*{l})' for l in self.y_letters)
-            self.w_bank = self.w_bank[self.w_bank['words'].str.contains(y_str)]
+            y_str = "^" + "".join(rf"(?=.*{l})" for l in self.y_letters)
+            self.w_bank = self.w_bank[self.w_bank["words"].str.contains(y_str)]
             for s, p in self.y_letters.items():
                 for i in p:
-                    self.w_bank = self.w_bank[self.w_bank['words'].str[i]!=s]
+                    self.w_bank = self.w_bank[self.w_bank["words"].str[i] != s]
             self.y_letters = {}
         for i, s in enumerate(self.prediction):
-            if s != '':
-                self.w_bank = self.w_bank[self.w_bank['words'].str[i]==s]
-        self.w_bank['w-score'] = [0] * len(self.w_bank)
+            if s != "":
+                self.w_bank = self.w_bank[self.w_bank["words"].str[i] == s]
+        self.w_bank["w-score"] = [0] * len(self.w_bank)
         if len(self.w_bank) > 5:
-            self.calc_letter_probs() #Recalculate letter position probability
+            self.calc_letter_probs()  # Recalculate letter position probability
         for x in range(self.game.letters):
-            if self.prediction[x] == '':
-                self.w_bank['w-score'] += self.w_bank[f'p-{x}']
+            if self.prediction[x] == "":
+                self.w_bank["w-score"] += self.w_bank[f"p-{x}"]
         if True not in [True for s in self.prediction if s in self.vowels]:
-            self.w_bank['w-score'] += self.w_bank['v-count'] / self.game.letters
-        mv_bank = self.w_bank[self.w_bank['w-score']==self.w_bank['w-score'].max()]
-        result = random.choice(mv_bank['words'].tolist())
+            self.w_bank["w-score"] += self.w_bank["v-count"] / self.game.letters
+        mv_bank = self.w_bank[
+            self.w_bank["w-score"] == self.w_bank["w-score"].max()
+        ]
+        result = random.choice(mv_bank["words"].tolist())
         return result