app.py

import random
import re

import numpy as np
import pandas as pd
import streamlit as st

from sudoku_solver import solve_sudoku
from ui_auxiliary import empty_board_str, board_matrix_to_dataframe

random.seed(0)
pd.set_option('display.precision', 0)

st.title("Solucionador de sudokus")

st.subheader("Utiliza la programación lineal para resolver un sudoku.")

input_data = st.text_area(
    label="Introduzca el estado inicial del tablero.",
    value=empty_board_str,
    height=400)

rows = np.repeat(np.arange(1, 10), 9)
cols = np.tile(np.arange(1, 10), 9)
values = []

for line in input_data.split("\n"):
    if "-" not in line:
        vals = re.findall("[0-9]", line.rstrip())
        values += [int(x) for x in vals]

if len(rows) == len(cols) == len(values):

    known_cells = pd.DataFrame({"i": rows, "j": cols, "k": values})
    board = known_cells.copy()

    known_cells = known_cells[known_cells["k"] != 0]

    board.k = ["" if x == 0 else str(x) for x in board.k]
    board = board.pivot(index="i", columns="j", values="k")

    if st.button("Resolver!"):
        st.markdown("**Solución**")
        res = solve_sudoku(known_cells)
        st.write(board_matrix_to_dataframe(res))
    else:
        st.markdown("**Diseño del tablero**")
        st.write(board)


else:
    st.write(
        "Algo está mal en la disposición del tablero. Por favor, inténtelo de nuevo.")