Update app.py

app.py

@@ -10,8 +10,8 @@ def initialize_grid(rows, cols):
 
 # Display the grid
 def display_grid(grid):
-    for row in grid:
-        st.write(" ".join(row))
+    for r, row in enumerate(grid):
+        st.write(" ".join([f"{c}({r},{col})" for col, c in enumerate(row)]))
 
 # Check for matches (horizontal and vertical)
 def check_matches(grid):
@@ -50,6 +50,18 @@ def remove_matches(grid, matches):
 
     return grid
 
+# Swap candies in the grid
+def swap_candies(grid, r1, c1, r2, c2):
+    if (
+        0 <= r1 < len(grid)
+        and 0 <= c1 < len(grid[0])
+        and 0 <= r2 < len(grid)
+        and 0 <= c2 < len(grid[0])
+    ):
+        grid[r1][c1], grid[r2][c2] = grid[r2][c2], grid[r1][c1]
+        return True
+    return False
+
 # Main Streamlit app
 def main():
     st.title("Candy Crush Game")
@@ -65,14 +77,24 @@ def main():
     st.write("### Current Grid")
     display_grid(st.session_state.grid)
 
-    # Check for matches
-    if st.button("Check Matches"):
-        matches = check_matches(st.session_state.grid)
-        if matches:
-            st.success(f"Found {len(matches)} candies to crush!")
-            st.session_state.grid = remove_matches(st.session_state.grid, matches)
+    # Input to select candies to swap
+    st.write("### Swap Candies")
+    r1 = st.number_input("Row of first candy (0-5):", min_value=0, max_value=rows - 1, step=1)
+    c1 = st.number_input("Column of first candy (0-5):", min_value=0, max_value=cols - 1, step=1)
+    r2 = st.number_input("Row of second candy (0-5):", min_value=0, max_value=rows - 1, step=1)
+    c2 = st.number_input("Column of second candy (0-5):", min_value=0, max_value=cols - 1, step=1)
+
+    if st.button("Swap Candies"):
+        if swap_candies(st.session_state.grid, r1, c1, r2, c2):
+            matches = check_matches(st.session_state.grid)
+            if matches:
+                st.success("Valid move! Candies matched!")
+                st.session_state.grid = remove_matches(st.session_state.grid, matches)
+            else:
+                st.warning("No match found. Swap reverted!")
+                swap_candies(st.session_state.grid, r1, c1, r2, c2)  # Undo the swap
         else:
-            st.warning("No matches found! Try again.")
+            st.error("Invalid swap! Please select valid coordinates.")
 
         st.write("### Updated Grid")
         display_grid(st.session_state.grid)