Update app.py

app.py

@@ -1,63 +1,116 @@
 import streamlit as st
-import plotly.graph_objs as go
-from bokeh.plotting import figure
-from bokeh.models import ColumnDataSource
-import random
-import os
-
-# Define initial data
-x_data = [0]
-y_data = [0]
-
-# Define Plotly figure
-plotly_fig = go.Figure()
-plotly_fig.add_trace(go.Scatter(x=x_data, y=y_data, mode='lines'))
-
-# Define Bokeh figure
-bokeh_fig = figure(plot_width=400, plot_height=400)
-bokeh_data = ColumnDataSource(data=dict(x=x_data, y=y_data))
-bokeh_fig.line('x', 'y', source=bokeh_data)
-
-# Define Streamlit app
-st.title("Infinite Graphics Demo")
-st.plotly_chart(plotly_fig)
-st.bokeh_chart(bokeh_fig)
-
-# Define data file path
-data_file = 'data.txt'
-
-# Define loop to update data and display new graphs
-while True:
-    # Read current data from file
-    with open(data_file, 'r') as f:
-        x_data, y_data = [float(x) for x in f.read().split(',')]
-    
-    # Generate new data
-    x_new = x_data[-1] + 1
-    y_new = y_data[-1] + random.randint(-10, 10)
-    
-    # Add new data to arrays
-    x_data.append(x_new)
-    y_data.append(y_new)
-    
-    # Update text file with new data
-    with open(data_file, 'w') as f:
-        f.write(','.join([str(x_data[-1]), str(y_data[-1])]))
-    
-    # Update Plotly figure
-    plotly_fig.data[0].x = x_data
-    plotly_fig.data[0].y = y_data
-    
-    # Update Bokeh figure
-    bokeh_data.data = dict(x=x_data, y=y_data)
-    
-    # Display new graphs
-    st.plotly_chart(plotly_fig)
-    st.bokeh_chart(bokeh_fig)
-    
-    # Update code with new data file path
-    with open(__file__, 'r') as f:
-        code = f.read()
-    code = code.replace("'data.txt'", f"'{data_file}'")
-    with open(__file__, 'w') as f:
-        f.write(code)
+import numpy as np
+import time
+
+# Define game board size
+BOARD_HEIGHT = 20
+BOARD_WIDTH = 10
+
+# Define shapes of tetrominoes
+SHAPES = [
+    np.array([
+        [1, 1],
+        [1, 1]
+    ]),
+
+    np.array([
+        [1, 1, 1],
+        [0, 1, 0]
+    ]),
+
+    np.array([
+        [1, 1, 1],
+        [1, 0, 0]
+    ]),
+
+    np.array([
+        [1, 1, 1],
+        [0, 0, 1]
+    ]),
+
+    np.array([
+        [1, 1, 0],
+        [0, 1, 1]
+    ]),
+
+    np.array([
+        [0, 1, 1],
+        [1, 1, 0]
+    ]),
+
+    np.array([
+        [1, 1, 1, 1]
+    ])
+]
+
+# Define colors for each tetromino
+COLORS = [
+    "yellow",
+    "orange",
+    "cyan",
+    "blue",
+    "purple",
+    "green",
+    "red"
+]
+
+# Define initial game state
+board = np.zeros((BOARD_HEIGHT, BOARD_WIDTH))
+current_shape = None
+current_shape_pos = (0, 0)
+current_shape_color = None
+score = 0
+game_over = False
+
+def get_random_shape():
+    shape_idx = np.random.randint(len(SHAPES))
+    shape = SHAPES[shape_idx]
+    color = COLORS[shape_idx]
+    return shape, color
+
+def place_shape_on_board(board, shape, position, color):
+    shape_height, shape_width = shape.shape
+    board_height, board_width = board.shape
+
+    # Check if shape can fit on the board at the given position
+    if position[0] + shape_height > board_height or position[1] + shape_width > board_width:
+        return False
+
+    # Check if there are any overlapping blocks
+    if np.any(board[position[0]:position[0]+shape_height, position[1]:position[1]+shape_width] * shape):
+        return False
+
+    # Place shape on the board
+    board[position[0]:position[0]+shape_height, position[1]:position[1]+shape_width] += shape * color
+
+    return True
+
+def check_if_row_is_complete(board, row):
+    return np.all(board[row, :])
+
+def remove_row(board, row):
+    board[row+1:,:] = board[row:-1,:]
+    board[0,:] = 0
+
+def remove_completed_rows(board):
+    for row in range(board.shape[0]):
+        if check_if_row_is_complete(board, row):
+            remove_row(board, row)
+
+def update_game_state():
+    global board, current_shape, current_shape_pos, current_shape_color, score, game_over
+
+    # Move shape down by one position
+    new_shape_pos = (current_shape_pos[0] + 1, current_shape_pos[1])
+
+    # Check if shape can be placed on the board
+    if not place_shape_on_board(board, current_shape, new_shape_pos, current_shape_color):
+        # Shape cannot be placed, lock it in place
+        place_shape_on_board(board, current_shape, current_shape_pos, current_shape_color)
+
+        # Check if any rows have been completed
+        remove_completed_rows(board)
+
+        # Check if game is over
+        if np.any(board[0,:]):
+            game_over = True