ai-snake-lab 0.1.0__py3-none-any.whl → 0.4.4__py3-none-any.whl

ai_snake_lab/ui/Db4EPlot.py

@@ -0,0 +1,160 @@
+"""
+db4e/Modules/Db4EPlot.py
+
+    Database 4 Everything
+    Author: Nadim-Daniel Ghaznavi
+    Copyright: (c) 2024-2025 Nadim-Daniel Ghaznavi
+    GitHub: https://github.com/NadimGhaznavi/db4e
+    License: GPL 3.0
+"""
+
+import math
+from collections import deque
+from textual_plot import PlotWidget, HiResMode, LegendLocation
+from textual.app import ComposeResult
+
+from ai_snake_lab.constants.DDb4EPlot import Plot
+from ai_snake_lab.constants.DLabels import DLabel
+
+
+MAX_DATA_POINTS = Plot.MAX_DATA_POINTS
+
+
+class Db4EPlot(PlotWidget):
+    """
+    A widget for plotting data based on TextualPlot's PlotWidget.
+    """
+
+    def __init__(self, title, id, thin_method=None):
+        super().__init__(title, id, allow_pan_and_zoom=False)
+        self._plot_id = id
+        self._title = title
+        self._thin_method = thin_method
+        if thin_method == Plot.SLIDING:
+            self._all_days = deque(maxlen=MAX_DATA_POINTS)
+            self._all_values = deque(maxlen=MAX_DATA_POINTS)
+        else:
+            self._all_days = None
+            self._all_values = None
+
+    def compose(self) -> ComposeResult:
+        yield PlotWidget()
+
+    def load_data(self, days, values, units):
+        self._all_days = days
+        self._all_values = values
+        if units:
+            self.set_ylabel(self._title + " (" + units + ")")
+        else:
+            self.set_ylabel(self._title)
+
+    def add_data(self, day, value):
+        self._all_days.append(day)
+        self._all_values.append(value)
+
+    def set_xlabel(self, label):
+        return super().set_xlabel(label)
+
+    def db4e_plot(self, days=None, values=None) -> None:
+        if days is not None and values is not None:
+            plot_days = days
+            plot_values = values
+        else:
+            plot_days = self._all_days
+            plot_values = self._all_values
+        self.clear()
+        if len(plot_days) == 0:
+            return
+        if self._thin_method == Plot.AVERAGE:
+            reduced_days, reduced_values = self.reduce_data(plot_days, plot_values)
+        else:
+            reduced_days, reduced_values = list(self._all_days), list(self._all_values)
+
+        self.plot(
+            x=reduced_days,
+            y=reduced_values,
+            hires_mode=HiResMode.BRAILLE,
+            line_style="green",
+            label=DLabel.CURRENT,
+        )
+
+        # Add an average plot over 20 to wash out the spikes and identify when the
+        # AI is maxing out.
+        window = max(1, len(reduced_values) // 20)  # e.g., 5% smoothing window
+        if len(reduced_values) > window:
+            smoothed = [
+                sum(reduced_values[i : i + window])
+                / len(reduced_values[i : i + window])
+                for i in range(len(reduced_values) - window + 1)
+            ]
+            smoothed_days = reduced_days[window - 1 :]
+            self.plot(
+                x=smoothed_days,
+                y=smoothed,
+                hires_mode=HiResMode.BRAILLE,
+                line_style="red",  # distinct color for trend
+                label=DLabel.AVERAGE,
+            )
+        self.show_legend(location=LegendLocation.TOPLEFT)
+
+    def reduce_data2(self, times, values):
+        # Reduce the total number of data points, otherwise the plot gets "blurry"
+        step = max(1, len(times) // MAX_DATA_POINTS)
+
+        # Reduce times with step
+        reduced_times = times[::step]
+
+        # Bin values by step (average)
+        reduced_values = [
+            sum(values[i : i + step]) / len(values[i : i + step])
+            for i in range(0, len(values), step)
+        ]
+        results = reduced_times[: len(reduced_values)], reduced_values
+        return results
+
+    def reduce_data(self, times, values):
+        """Reduce times and values into <= MAX_DATA_POINTS bins.
+        Each bin's value is the average of the values in the bin.
+        Each bin's time is chosen as the last time in the bin (so last bin -> times[-1]).
+        """
+        if not times or not values:
+            return [], []
+
+        assert len(times) == len(values), "times and values must be same length"
+
+        step = max(1, math.ceil(len(times) / MAX_DATA_POINTS))
+
+        reduced_times = []
+        reduced_values = []
+        for i in range(0, len(times), step):
+            chunk_times = times[i : i + step]
+            chunk_vals = values[i : i + step]
+
+            # average values (works for floats or Decimal)
+            avg_val = sum(chunk_vals) / len(chunk_vals)
+
+            # representative time: choose last item in the chunk so final rep is times[-1]
+            rep_time = chunk_times[-1]
+
+            reduced_times.append(rep_time)
+            reduced_values.append(avg_val)
+
+        # Guarantee the final time equals the exact last time (safety)
+        if reduced_times:
+            reduced_times[-1] = times[-1]
+
+        return reduced_times, reduced_values
+
+    def update_time_range(self, selected_time):
+        if selected_time == -1:
+            return
+
+        selected_time = int(selected_time)
+        max_length = len(self._all_days)
+        if selected_time > max_length:
+            new_values = self._all_values
+            new_times = self._all_days
+        else:
+            new_values = self._all_values[-selected_time:]
+            new_times = self._all_days[-selected_time:]
+        self.db4e_plot(new_times, new_values)

ai_snake_lab/utils/AISim.tcss

@@ -1,21 +1,30 @@
-Button {
-    align: center middle;
-    border: round #0c323e;
-    width: 10;
-    background: black; 
+Screen {
+    layout: grid;
+    grid-size: 3 4;
+    grid-rows: 3 7 6 11 10;
+    grid-columns: 32 46 30;
 }
-    
-#game_board {
-    height: 22;
+
+#title {
+    column-span: 3;
+    width: 100%;
+    padding: 0 1;
     border: round #0c323e;
-    width: 42;
-    align: center middle;
+    color: #3e99af;
+    background: black;
+    text-style: bold;
+    align: center middle; }
+
+#settings_box {
+    border-title-color: #5fc442;
+    border-title-style: bold;
+    border: round #0c323e;
+    padding: 0 1;
     background: black;
 }
 
 #game_box {
-    height: 24;
-    width: 46;
+    row-span: 3;
     border-title-color: #5fc442;
     border-subtitle-color: #5fc442;
     border: round #0c323e;
@@ -25,7 +34,6 @@ Button {
 }
 
 #runtime_box {
-    height: 24;
     border-title-color: #5fc442;
     border-title-style: bold;
     border: round #0c323e;
@@ -33,40 +41,48 @@ Button {
     background: black;
 }
 
-#score {
-    height: 3;
+#filler_1 {
+}
+
+#filler_2 {
+}
+
+#filler_3 {
+}
+
+#game_score_plot {
+    dock: bottom;
     border: round #0c323e;
-    width: 42;
-    padding: 0 1;
-    color: #5fc442;
-    background: black;
+    height: 15;
+    width: 108;
+    background: black
 }
 
-#settings_box {
-    height: 6;
-    border-title-color: #5fc442;
-    border-title-style: bold;
+#legend {
     border: round #0c323e;
-    padding: 0 1;
-    background: black;
 }
 
-#title {
-    height: 3;
-    padding: 0 1;
+Button {
+    align: center middle;
     border: round #0c323e;
-    color: #3e99af;
-    width: 100%;
+    width: 10;
+    background: black; 
+}
+
+#game_board {
+    border: round #0c323e;
+    align: center middle;
     background: black;
-    text-style: bold;
-    align: center middle; }
+}
 
-#button_pause {
-    color: #c44242;
+
+
+#button_defaults {
+    color: #3e99af;
     text-style: bold; 
 }
 
-#button_reset {
+#button_pause {
     color: #c49b42;
     text-style: bold; 
 }
@@ -76,6 +92,11 @@ Button {
     text-style: bold; 
 }
 
+#button_restart {
+    color: #3e99af;
+    text-style: bold; 
+}
+
 #button_quit {
     color: #c44242;
     text-style: bold; 
@@ -91,15 +112,37 @@ Button {
 }
 
 .label {
+    width: 15;
     color: #afa73eff;
 }
 
-.running #button_start {
+.label_settings {
+    color: #5fc442;
+    width: 18;
+}
+
+.paused #button_pause {
     display: none;
 }
 
-.label_settings {
-    color: #5fc442;
+.paused #button_update {
+    display: none;
+}
+
+.running #button_reset {
+    display: none;
+}
+
+.running #button_restart {
+    display: none;
+}
+
+.stopped #button_restart {
+    display: none;
+}
+
+.running #button_start {
+    display: none;
 }
 
 .stopped #button_pause {

{ai_snake_lab-0.1.0.dist-info → ai_snake_lab-0.4.4.dist-info}/METADATA

@@ -1,8 +1,9 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: ai-snake-lab
-Version: 0.1.0
+Version: 0.4.4
 Summary: Interactive reinforcement learning sandbox for experimenting with AI agents in a classic Snake Game environment.
 License: GPL-3.0
+License-File: LICENSE
 Keywords: AI,Reinforcement Learning,Textual,Snake,Simulation,SQLite,Python
 Author: Nadim-Daniel Ghaznavi
 Author-email: nghaznavi@gmail.com
@@ -63,8 +64,41 @@ Description-Content-Type: text/markdown
 
 ---
 
-# Links
+# Installation
+
+This project is on [PyPI](https://pypi.org/project/ai-snake-lab/). You can install the *AI Snake Lab* software using `pip`.
+
+## Create a Sandbox 
+
+```shell
+python3 -m venv snake_venv
+. snake_venv/bin/activate
+```
+
+## Install the AI Snake Lab
+
+After you have activated your *venv* environment:
+
+```shell
+pip install ai-snake-lab
+```
+
+---
+
+# Running the AI Snake Lab
+
+From within your *venv* environment:
+
+```shell
+ai-snake-lab
+```
+
+---
+
+# Links and Acknowledgements
+
+This code is based on a YouTube tutorial, [Python + PyTorch + Pygame Reinforcement Learning – Train an AI to Play Snake](https://www.youtube.com/watch?v=L8ypSXwyBds&t=1042s&ab_channel=freeCodeCamp.org) by Patrick Loeber. You can access his original code [here](https://github.com/patrickloeber/snake-ai-pytorch) on GitHub. Thank you Patrick!!! You are amazing!!!!
+
+Thanks also go out to Will McGugan and the [Textual](https://textual.textualize.io/) team. Textual is an amazing framework. Talk about *rapid Application Development*. Porting this took less than a day.
 
-* [Project Layout](/pages/project_layout.html)
-* [SQLite3 Schema](/pages/db_schema.html)
 ---

ai_snake_lab-0.4.4.dist-info/RECORD

@@ -0,0 +1,31 @@
+ai_snake_lab/AISim.py,sha256=GiqzvU0LAvjjmjP_0JDeak7DAF0R-pEzrPgVpCN-gs8,15148
+ai_snake_lab/ai/AIAgent.py,sha256=p_O7zC15s_tBcI7oqu60X2-3AbrzAWlGXWkrLVC4UsQ,3005
+ai_snake_lab/ai/AITrainer.py,sha256=ssX6B03yZLEKhNCJv9D83iFAEEhU3_XbW4sA0z5QMRM,3304
+ai_snake_lab/ai/EpsilonAlgo.py,sha256=iNVXOqlyjh5vlFePpksR0PTT9M5FDDwJi7P6pYegSrA,2144
+ai_snake_lab/ai/ReplayMemory.py,sha256=eGdszKKeYuvppmmkA9AtypLRbGZQGZO-uCSHb8qcesw,3986
+ai_snake_lab/ai/models/ModelL.py,sha256=hK7MoPyIF1_K43R_9xW_MYuaweQGo9qSMUIVrHDQZnQ,1249
+ai_snake_lab/ai/models/ModelRNN.py,sha256=Ky63BUqJEmwe1PbM4gtjvqd7SmSy_2XA1n-yu8DNexI,1104
+ai_snake_lab/constants/DDb4EPlot.py,sha256=2g8SdDVZWrTVSTw-rpO84OrNfl7ToBYtf6J7sYB7q8w,442
+ai_snake_lab/constants/DDef.py,sha256=XABLL9P_WUkCiNr2L-phiNL6cnxc_E0uD4vW6WhtZs0,378
+ai_snake_lab/constants/DDir.py,sha256=e8pNvZJpR4R6Cm3VRGljry_g-GAODYkz1ZrU8OudXuQ,395
+ai_snake_lab/constants/DEpsilon.py,sha256=HuhMGxy3WIebcFQ2XZvlOpl03JwPYhHZdO40Bzph9yM,420
+ai_snake_lab/constants/DFields.py,sha256=0MV21hLy-Ojnn4Qa3C9_dQLPjC-3VYBjes_YU5V1Bio,521
+ai_snake_lab/constants/DFile.py,sha256=jYnYgrMqw3BMa3XEJw5XtC5buHfE0v7QFLv25BKNHW0,378
+ai_snake_lab/constants/DLabels.py,sha256=Nv7vCs8pxpF6Px5FD3JFkI7KEPWy6G8wLqzv31_nXxw,1501
+ai_snake_lab/constants/DLayout.py,sha256=XhRMpOqgzOkL0kcFz4pfS-HuVsYZZLBC6GIAq-E1_Ik,1616
+ai_snake_lab/constants/DModelL.py,sha256=VzAK5Uqkb0ovMPKTo_DJ2lQmIJkpRQyF-z3g3y4-j20,506
+ai_snake_lab/constants/DModelLRNN.py,sha256=nyPYa1-8JB7KERWEJM5jBFH3PmISNcr9JPi-KJ_Cq2g,445
+ai_snake_lab/constants/DReplayMemory.py,sha256=m0SSxFL6v2mVXVxf0lJTJzKfXguxrnsnrJph0v-2lX8,589
+ai_snake_lab/constants/DSim.py,sha256=ULi_e9g2eBPPZv_swQQoagKLT8dL3gJTwlQihkQk8RY,510
+ai_snake_lab/constants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_snake_lab/game/GameBoard.py,sha256=VjjNXnZZJ_lTo4bcxflB9TWxi792Do7yMnvL3umrMCc,8452
+ai_snake_lab/game/GameElements.py,sha256=9vcNH2YWAzYyBZosvj1DzO6ZmUCUMUd8MaIhORWQ8go,470
+ai_snake_lab/game/SnakeGame.py,sha256=_gyxhLPggPF7B2XvmWLgWP2ywnYWX7QjmwLmi21QuIw,6476
+ai_snake_lab/ui/Db4EPlot.py,sha256=pcEb0ydXNX2wL0EFBSrqhIoTEhryk4GD0Ua8FFEaZHY,5352
+ai_snake_lab/utils/AISim.tcss,sha256=XnhqjuxtOiNQHDyXFrbAYSlCw-6ch4e1o52nbytoeF0,2118
+ai_snake_lab/utils/ConstGroup.py,sha256=ZYyQxFd9PudBUZmc_NsNvWCp___utOe1MptqD3eyVH8,1174
+ai_snake_lab-0.4.4.dist-info/METADATA,sha256=dNvC3Ia4TfGeL5L4N6rS2snjbydNdT_ZCo1XGyYGSMU,3689
+ai_snake_lab-0.4.4.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+ai_snake_lab-0.4.4.dist-info/entry_points.txt,sha256=VXWDqigE8D35J7scdqjR0EVvOjtP1mm0uS76xMnvodo,56
+ai_snake_lab-0.4.4.dist-info/licenses/LICENSE,sha256=f-FHFu0xzHH8O_mvKTw2jUZKhTpw6obpmVOI9rnpKeU,35151
+ai_snake_lab-0.4.4.dist-info/RECORD,,

{ai_snake_lab-0.1.0.dist-info → ai_snake_lab-0.4.4.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 2.1.3
+Generator: poetry-core 2.2.1
 Root-Is-Purelib: true
 Tag: py3-none-any

{ai_snake_lab-0.1.0.dist-info → ai_snake_lab-0.4.4.dist-info/licenses}/LICENSE

@@ -672,3 +672,5 @@ may consider it more useful to permit linking proprietary applications with
 the library.  If this is what you want to do, use the GNU Lesser General
 Public License instead of this License.  But first, please read
 <https://www.gnu.org/licenses/why-not-lgpl.html>.
+
+

ai_snake_lab-0.1.0.dist-info/RECORD

@@ -1,28 +0,0 @@
-ai_snake_lab/AISim.py,sha256=9jnWX-JYNqpYxtqb7vf4pbI9tgegrt91QnFWfVU2AFI,10168
-ai_snake_lab/ai/AIAgent.py,sha256=mwfDwQlm9vRCDlXj5LyIQGtwjZL3j4Fc_L9w79jm2wA,2955
-ai_snake_lab/ai/AITrainer.py,sha256=VbLfSV3ndplG-KhqpTlUSkoH8lFGGT0auaPjFRzabR0,3169
-ai_snake_lab/ai/EpsilonAlgo.py,sha256=iNVXOqlyjh5vlFePpksR0PTT9M5FDDwJi7P6pYegSrA,2144
-ai_snake_lab/ai/ReplayMemory.py,sha256=g8Q_PXiFSL1g4zOEa8pKc9w_x5ijVlG17npYpawL_14,2864
-ai_snake_lab/ai/models/ModelL.py,sha256=EiTGVag8Sxhoak1sklA4iX4FR65Nlx8m9o-JQNAqOX8,1096
-ai_snake_lab/ai/models/ModelRNN.py,sha256=jIPK5ZS1olakjQ-9XA6-4EiY70d3RLcKqFqHTZD7cIQ,1104
-ai_snake_lab/constants/DDef.py,sha256=zyvRkWPrtopKYclbbDGI-1pPAK1h1MyVMdbvhUZPbJY,371
-ai_snake_lab/constants/DDir.py,sha256=AxmgShKHT5b5WyIlseVHcO7Mwejmyf3hxT62E3RirTg,323
-ai_snake_lab/constants/DEpsilon.py,sha256=T4B_KMHrCEKsiCx2AgvnXMpQ-V4D2oSCMPkYqw2d-0U,407
-ai_snake_lab/constants/DFields.py,sha256=Z_whswaGfNIRe9d_9UeXWiWZCB0d48BpM1Wg6tILEa0,357
-ai_snake_lab/constants/DFile.py,sha256=i6i_K3y90-GgqesN9_zL2NmQu0Vu2VT2V5k7n6QFQR4,328
-ai_snake_lab/constants/DLabels.py,sha256=5kUrAbYgdJL9zynAjSFQr3FmEvy2CbU0DKRZfwUs71I,883
-ai_snake_lab/constants/DLayout.py,sha256=T5CeJBaiDST1gaX3ek6TpRJdgETvbl4uk-tlWZah-Y8,1125
-ai_snake_lab/constants/DModelL.py,sha256=-bb1SMQQE-iGxZcuAGcJv-5xJqtZKsYbmMHwUak11jg,351
-ai_snake_lab/constants/DModelLRNN.py,sha256=gceyvn0-I3XpRL6cUb8Ep67DoybN8bZdDpcXVKf_DCk,432
-ai_snake_lab/constants/DReplayMemory.py,sha256=DNPn9dFa9piNXpuT6AMx1B96UpoMFXALrYYLjyKwPpU,576
-ai_snake_lab/constants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_snake_lab/game/GameBoard.py,sha256=wl2V0cxjqBKHuS1DSx36EE_DRH54tikuhaA9eC5qPsY,7857
-ai_snake_lab/game/GameElements.py,sha256=9vcNH2YWAzYyBZosvj1DzO6ZmUCUMUd8MaIhORWQ8go,470
-ai_snake_lab/game/SnakeGame.py,sha256=1YS0xgyzuGQtGgKxDUcPJYM2NV56ekL4OpaOrbpnDW0,5673
-ai_snake_lab/utils/AISim.tcss,sha256=c0Xq8hjA7YokioZUzJGHhAlX13t2kgelA26hOWtUSVQ,1674
-ai_snake_lab/utils/ConstGroup.py,sha256=ZYyQxFd9PudBUZmc_NsNvWCp___utOe1MptqD3eyVH8,1174
-ai_snake_lab-0.1.0.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-ai_snake_lab-0.1.0.dist-info/METADATA,sha256=X3xDSODxx3q8zlm8GTntrkCa7PL7AbGm-QsDqFgTj8A,2712
-ai_snake_lab-0.1.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-ai_snake_lab-0.1.0.dist-info/entry_points.txt,sha256=VXWDqigE8D35J7scdqjR0EVvOjtP1mm0uS76xMnvodo,56
-ai_snake_lab-0.1.0.dist-info/RECORD,,