Update app.py

app.py

@@ -2,71 +2,56 @@ import streamlit as st
 import random
 import string
 
-# Define constants
-SIZE = 10
-WORDS_FILE = "words.txt"
+# Initialize global variables
+board_size = 15
+words = []
+board = [[' ' for _ in range(board_size)] for _ in range(board_size)]
 
-# Define helper functions
-def generate_grid(size):
-    grid = []
-    for i in range(size):
-        row = [random.choice(string.ascii_uppercase) for j in range(size)]
-        grid.append(row)
-    return grid
+# Function to load word list
+def load_word_list():
+    global words
+    words = st.text_area("Enter a list of words (one per line)", "").split("\n")
 
-def add_words_to_grid(words, grid):
+# Function to save word list
+def save_word_list():
+    global words
+    with open("word_list.txt", "w") as f:
+        f.write("\n".join(words))
+    st.write("Word list saved successfully!")
+
+# Function to generate board
+def generate_board():
+    global board, words
+    # Clear board
+    board = [[' ' for _ in range(board_size)] for _ in range(board_size)]
+    # Add words to board
     for word in words:
-        added = False
-        while not added:
-            direction = random.choice(["horizontal", "vertical", "diagonal"])
-            if direction == "horizontal":
-                row = random.randint(0, len(grid)-1)
-                col = random.randint(0, len(grid[0])-len(word))
-                if all(grid[row][col+j] == "" or grid[row][col+j] == word[j] for j in range(len(word))):
-                    for j in range(len(word)):
-                        grid[row][col+j] = word[j]
-                    added = True
-            elif direction == "vertical":
-                row = random.randint(0, len(grid)-len(word))
-                col = random.randint(0, len(grid[0])-1)
-                if all(grid[row+j][col] == "" or grid[row+j][col] == word[j] for j in range(len(word))):
-                    for j in range(len(word)):
-                        grid[row+j][col] = word[j]
-                    added = True
-            elif direction == "diagonal":
-                row = random.randint(0, len(grid)-len(word))
-                col = random.randint(0, len(grid[0])-len(word))
-                if all(grid[row+j][col+j] == "" or grid[row+j][col+j] == word[j] for j in range(len(word))):
-                    for j in range(len(word)):
-                        grid[row+j][col+j] = word[j]
-                    added = True
-    return grid
+        word = word.upper()
+        # Generate random starting position and direction for word
+        row, col = random.randint(0, board_size - 1), random.randint(0, board_size - 1)
+        direction = random.choice(['horizontal', 'vertical', 'diagonal'])
+        # Check if word can fit in selected direction
+        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:
+            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:
+            for i, letter in enumerate(word):
+                board[row+i][col+i] = letter
+    # Fill remaining board spaces with random letters
+    for i in range(board_size):
+        for j in range(board_size):
+            if board[i][j] == ' ':
+                board[i][j] = random.choice(string.ascii_uppercase)
+
+# Display sidebar buttons and handle user input
+st.sidebar.button("Load Word List", on_click=load_word_list)
+st.sidebar.button("Save Word List", on_click=save_word_list)
+if st.sidebar.button("Generate Board"):
+    generate_board()
 
-# Define Streamlit app
-def app():
-    st.title("Word Search")
-    
-    # Load words from file or user input
-    if st.sidebar.button("Load words"):
-        with open(WORDS_FILE, "r") as f:
-            words = [word.strip() for word in f.readlines()]
-    else:
-        words_input = st.sidebar.text_area("Enter words (one per line)")
-        words = [word.strip() for word in words_input.split("\n")]
-    
-    # Save words to file
-    if st.sidebar.button("Save words"):
-        with open(WORDS_FILE, "w") as f:
-            for word in words:
-                f.write(word + "\n")
-    
-    # Generate and display word search
-    if words:
-        grid = generate_grid(SIZE)
-        grid = add_words_to_grid(words, grid)
-        st.write("Find the following words:")
-        st.write(words)
-        for row in grid:
-            st.write(" ".join(row))
-    else:
-        st.write("Enter words to generate a word search.")
+# Display letter board
+st.write("Word Search Board:")
+st.table(board)