Add visualization and example to main.py

Author: DenHvideDvaerg

main.py

@@ -2,25 +2,28 @@
 import time
 
 
+def example_8x8():
+    """8x8 example from https://puzzlegenius.org/snake/"""
+    puzzle = SnakePuzzle(
+        row_sums=[4, 2, 2, 3, 1, 3, 2, 6],
+        col_sums=[3, 2, 7, 2, 2, 4, 1, 2],
+        start_cell=(2, 5),
+        end_cell=(6, 7)
+    )
+    return puzzle
+
+
 def main():
     """
     Example usage of the Snake solver.
     
-    TODO: Replace with your specific puzzle parameters and usage.
     """
     print("Snake MIP Solver - Example Usage")
     print("=" * 50)
 
-    # TODO: Create your puzzle instance
-    # Example:
-    # puzzle = SnakePuzzle(
-    #     size=5,
-    #     constraints=[1, 2, 3, 2, 1]
-    # )
-    
+
     try:
-        # For now, create a basic puzzle (replace with your parameters)
-        puzzle = SnakePuzzle()
+        puzzle = example_8x8()
         print(f"Created puzzle: {puzzle}")
 
         # Create solver
@@ -37,7 +40,34 @@ def main():
         if solution is not None:
             print(f"\n✅ Solution found in {solve_time:.3f} seconds!")
             print(f"Solution: {solution}")
+            print(puzzle.get_board_visualization())
 
+            manual_solution = set()
+            manual_solution.add((2, 5))
+            manual_solution.add((1, 5))
+            manual_solution.add((0, 5))
+            manual_solution.add((0, 4))
+            manual_solution.add((0, 3))
+            manual_solution.add((0, 2))
+            manual_solution.add((1, 2))
+            manual_solution.add((2, 2))
+            manual_solution.add((3, 2))
+            manual_solution.add((3, 1))
+            manual_solution.add((3, 0))
+            manual_solution.add((4, 0))
+            manual_solution.add((5, 0))
+            manual_solution.add((5, 1))
+            manual_solution.add((5, 2))
+            manual_solution.add((6, 2))
+            manual_solution.add((7, 2))
+            manual_solution.add((7, 3))
+            manual_solution.add((7, 4))
+            manual_solution.add((7, 5))
+            manual_solution.add((7, 6))
+            manual_solution.add((7, 7))
+            manual_solution.add((6, 7))
+            print(puzzle.get_board_visualization(manual_solution, show_indices=True))
+            print(f"Manual solution valid? {puzzle.is_valid_solution(manual_solution)}")
             # TODO: Add solution visualization or validation
             # Example:
             # if puzzle.is_valid_solution(solution):

snake_mip_solver/puzzle.py

@@ -273,3 +273,53 @@ def get_start_cell(self) -> Tuple[int, int]:
     def get_end_cell(self) -> Tuple[int, int]:
         """Get the end cell position."""
         return self.end_cell
+    
+    def get_board_visualization(self, snake_positions: Optional[Set[Tuple[int, int]]] = None, 
+                     show_indices: bool = False) -> str:
+        """
+        Create a string representation of the puzzle.
+        
+        Args:
+            snake_positions: Optional set of snake body positions to display
+            show_indices: Whether to show row/column indices
+            
+        Returns:
+            String representation of the puzzle
+        """
+        snake_positions = snake_positions or set()
+        lines = []
+        
+        # Header with column indices and sums
+        if show_indices:
+            header = "    " + " ".join(str(i) for i in range(self.cols))
+            lines.append(header)
+            col_sums_line = "    " + " ".join(str(s) if s is not None else "?" for s in self.col_sums)
+        else:
+            col_sums_line = "  " + " ".join(str(s) if s is not None else "?" for s in self.col_sums)
+        
+        lines.append(col_sums_line)
+        
+        # Board rows
+        for row in range(self.rows):
+            row_parts = []
+            
+            if show_indices:
+                row_parts.append(str(row))
+            
+            row_sum_str = str(self.row_sums[row]) if self.row_sums[row] is not None else "?"
+            row_parts.append(row_sum_str)
+            
+            for col in range(self.cols):
+                pos = (row, col)
+                if pos == self.start_cell:
+                    row_parts.append('S')
+                elif pos == self.end_cell:
+                    row_parts.append('E')
+                elif pos in snake_positions:
+                    row_parts.append('x') # █
+                else:
+                    row_parts.append('_')
+            
+            lines.append(' '.join(row_parts))
+        
+        return '\n'.join(lines)