Streamlit ready. Probably not bug free

app.py

@@ -1,69 +1,68 @@
 from random import randint
 import pygame
 import numpy as np
-import time
+import streamlit as st
 
 import os
-os.system("pip install -r requirements.txt")
-exit = False
-# Settings
-# SIZE = width, height = 800, 600  # Resolution. (4:3)!
+
+# os.system("pip install -r requirements.txt")
 BG_COLOR = (0, 0, 0)
 fullscreen = True
 
 logo = pygame.image.load('dvd_logo_0002.png')
-logo = pygame.transform.scale(logo, (200, 100))
-clock = pygame.time.Clock()
+logo = pygame.transform.scale(logo, (100, 200))
+
+
+logo = pygame.surfarray.array3d(logo)
+logo = np.flip(logo, axis=1)
+logo = pygame.surfarray.make_surface(logo)
+
 img_size = logo.get_rect().size
-screen = pygame.display.set_mode()
-pygame.display.set_caption('Idle screen DVD bounce')
-screen_info = pygame.display.Info()
+screen_width = 1900
+screen_height = 1080
+screen = pygame.Surface((screen_height, screen_width))
 
-if fullscreen:
-    DISPLAYSURFACE = pygame.display.set_mode((0, 0), pygame.FULLSCREEN)
-    pygame.mouse.set_visible(False)
+screen.fill((0, 0, 0))
 
-x = randint(50, screen_info.current_w - 200)
-y = randint(50, screen_info.current_h - 200)
+
+x = randint(0, screen_width - img_size[0])
+y = randint(0, screen_height - img_size[0])
 x_speed = 2
-y_speed = 2
+y_speed = 3
 
 dvd_background_color = [255, 255, 255]
 
 
-def random_color_change(surface):
-    global dvd_background_color
-
+def random_color_change(surface, color_to_be_changed):
     np_surface = pygame.surfarray.array3d(surface)
     for channel in range(np_surface.shape[-1]):
         # np_surface[:, :, channel] = randint(0, 255)
-        random_channel_color = randint(50, 255)
+        random_channel_color = randint(70, 255)
         np_surface[:, :, channel] = np.where(
-            np.logical_and(np_surface[:, :, channel] >= dvd_background_color[channel] - 20,
-                           np_surface[:, :, channel] <= dvd_background_color[channel] + 20),
+            np.logical_and(np_surface[:, :, channel] >= color_to_be_changed[channel] - 40,
+                           np_surface[:, :, channel] <= color_to_be_changed[channel]),
             random_channel_color, np_surface[:, :, channel])
-        dvd_background_color[channel] = random_channel_color
-    return pygame.surfarray.make_surface(np_surface)
+        color_to_be_changed[channel] = random_channel_color
+    return pygame.surfarray.make_surface(np_surface), color_to_be_changed
 
 
-while not exit:
+while True:
     screen.fill(BG_COLOR)
 
-    if ((x + img_size[0] >= screen_info.current_w) or (x <= 0)) and ((y + img_size[1] >= screen_info.current_h) or (y <= 0)):
-        time.sleep(9999999999)
-    if (x + img_size[0] >= screen_info.current_w) or (x <= 0):
-        logo = random_color_change(logo)
+    if ((x + img_size[0] + x_speed >= screen_width) or (x + x_speed <= 0)) and (
+            (y + y_speed + img_size[1] >= screen_height) or (y + y_speed <= 0)):
+        x -= 1.5 * x_speed
+        y -= 1.5 * x_speed
+    if (x + img_size[0] >= screen_width) or (x <= 0):
+        logo, dvd_background_color = random_color_change(logo, dvd_background_color)
         x_speed = -x_speed
-    if (y + img_size[1] >= screen_info.current_h) or (y <= 0):
-        logo = random_color_change(logo)
+    if (y + img_size[1] >= screen_height) or (y <= 0):
+        logo, dvd_background_color = random_color_change(logo, dvd_background_color)
         y_speed = -y_speed
+
     x += x_speed
     y += y_speed
     screen.blit(logo, (x, y))
-    pygame.display.update()
-    clock.tick(60)
-    for event in pygame.event.get():
-        if event.type == pygame.QUIT:
-            exit = True
+    screen_np = pygame.surfarray.array3d(screen)
+    st.image(screen_np, caption='Idle Bouncing DVD logo')
 
-pygame.quit()