Update app.py

app.py

@@ -2,83 +2,74 @@ import streamlit as st
 import random
 import string
 
-board_size = 15
-words = []
-board = [[' ' for _ in range(board_size)] for _ in range(board_size)]
+BOARD_SIZE = 15
+WORDS_FILE = "word_list.txt"
 
 def load_word_list():
-    global words
     try:
-        with open("word_list.txt", "r") as f:
+        with open(WORDS_FILE, "r") as f:
             words = f.read().split("\n")
     except FileNotFoundError:
-        pass
+        words = []
     st.text_area("Enter a list of words (one per line)", "\n".join(words), key="words_area")
     st.sidebar.subheader("Word List:")
     for i, word in enumerate(words):
         st.sidebar.write(f"{i+1}. {word}")
+    return words
 
-def save_word_list():
-    global words
-    with open("word_list.txt", "w") as f:
+def save_word_list(words):
+    with open(WORDS_FILE, "w") as f:
         f.write("\n".join(words))
     st.write("Word list saved successfully!")
 
-def generate_board():
-    global board, words
-    board = [[' ' for _ in range(board_size)] for _ in range(board_size)]
+def generate_board(words):
+    board = [[' ' for _ in range(BOARD_SIZE)] for _ in range(BOARD_SIZE)]
     for word in words:
         word = word.upper()
-        row, col = random.randint(0, board_size - 1), random.randint(0, board_size - 1)
+        row, col = random.randint(0, BOARD_SIZE - 1), random.randint(0, BOARD_SIZE - 1)
         direction = random.choice(['horizontal', 'vertical', 'diagonal'])
-        if direction == 'horizontal' and col + len(word) <= board_size:
+        if direction == 'horizontal' and col + len(word) <= BOARD_SIZE:
             for i, letter in enumerate(word):
                 board[row][col+i] = letter
-        elif direction == 'vertical' and row + len(word) <= board_size:
+        elif direction == 'vertical' and row + len(word) <= BOARD_SIZE:
             for i, letter in enumerate(word):
                 board[row+i][col] = letter
-        elif direction == 'diagonal' and row + len(word) <= board_size and col + len(word) <= board_size:
+        elif direction == 'diagonal' and row + len(word) <= BOARD_SIZE and col + len(word) <= BOARD_SIZE:
             for i, letter in enumerate(word):
                 board[row+i][col+i] = letter
-    for i in range(board_size):
-        for j in range(board_size):
+    for i in range(BOARD_SIZE):
+        for j in range(BOARD_SIZE):
             if board[i][j] == ' ':
                 board[i][j] = random.choice(string.ascii_uppercase)
+    return board
+
+def display_word_list(words):
+    st.sidebar.subheader("Word List:")
+    for i, word in enumerate(words):
+        st.sidebar.write(f"{i+1}. {word}")
+
+def display_board(board):
+    st.write("Word Search Board:")
+    st.table(board)
+
+def paste_story():
+    story = st.text_area("Paste your story here")
+    if st.button("Build Word List"):
+        words = [word.upper() for word in story.split() if len(word) > 4]
+        words = sorted(words, key=len, reverse=True)[:10]
+        save_word_list(words)
+        board = generate_board(words)
+        display_word_list(words)
+        display_board(board)
 
 buttons = {
     "Load Word List": load_word_list,
-    "Save Word List": save_word_list,
-    "Generate Board": generate_board
 }
 
-if "words_area" not in st.session_state:
-    try:
-        with open("word_list.txt", "r") as f:
-            words = f.read().split("\n")
-    except FileNotFoundError:
-        pass
-    st.text_area("Enter a list of words (one per line)", "\n".join(words), key="words_area")
-
+words = setup()
 for button_label, button_func in buttons.items():
     if st.sidebar.button(button_label):
-        with open("word_list.txt", "w") as f:
-            f.write(st.session_state.words_area)
-        words = st.session_state.words_area.split("\n")
-        button_func()
-        st.sidebar.subheader("Word List:")
-        for i, word in enumerate(words):
-            st.sidebar.write(f"{i+1}. {word}")
-
-with open("word_list.txt", "w") as f:
-    f.write(st.session_state.words_area)
-words = st.session_state.words_area
-if st.button("Save Word List", key="save_word_list_btn"):
-    words = words.split("\n")
-    save_word_list()
-
-st.sidebar.subheader("Word List:")
-for i, word in enumerate(words.split("\n")):
-    st.sidebar.write(f"{i+1}. {word}")
+        words = button_func()
+        display_word_list(words)
 
-st.write("Word Search Board:")
-st.table(board)
+paste_story()