Update app.py

app.py

@@ -2,56 +2,42 @@ import random
 import time
 import numpy as np
 import gradio as gr
+import colorsys
 
 from fastrtc import WebRTC
 
 def video_generation_handler(prompt):
-    if seed == -1: 
-        seed = random.randint(0, 2**32 - 1)
-    
-    random.seed(seed)
-    np.random.seed(seed)
-    
-    time.sleep(0.5)
-    
-    num_frames = 60 
+    num_frames = 120
     width, height = 832, 480
     
     for frame_idx in range(num_frames): 
-        # Create a frame with a random solid color that changes over time
-        hue = (frame_idx * 6) % 360  # Cycle through hues
-        saturation = 0.7 + 0.3 * np.sin(frame_idx * 0.1)  # Vary saturation
-        value = 0.8 + 0.2 * np.sin(frame_idx * 0.2)  # Vary brightness
+        hue = (frame_idx * 3) % 360
+        saturation = 0.7 + 0.3 * np.sin(frame_idx * 0.1)
+        value = 0.8 + 0.2 * np.sin(frame_idx * 0.2)
         
-        # Convert HSV to RGB
-        import colorsys
         r, g, b = colorsys.hsv_to_rgb(hue/360, saturation, value)
         
-        # Create frame with some noise for visual interest
-        frame_np = np.full((height, width, 3), [r*255, g*255, b*255], dtype=np.uint8)
+        rgb_frame = np.full((height, width, 3), [r*255, g*255, b*255], dtype=np.uint8)
         
-        # Add some random noise
-        noise = np.random.randint(-20, 20, (height, width, 3), dtype=np.int16)
-        frame_np = np.clip(frame_np.astype(np.int16) + noise, 0, 255).astype(np.uint8)
+        noise = np.random.randint(-15, 15, (height, width, 3), dtype=np.int16)
+        rgb_frame = np.clip(rgb_frame.astype(np.int16) + noise, 0, 255).astype(np.uint8)
         
-        # Add some simple animation (moving gradient)
         x_grad = np.linspace(0, 1, width)
         y_grad = np.linspace(0, 1, height)
         X, Y = np.meshgrid(x_grad, y_grad)
         
-        # Moving wave pattern
         wave = np.sin(X * 4 + frame_idx * 0.2) * np.sin(Y * 4 + frame_idx * 0.3)
-        wave_normalized = ((wave + 1) / 2 * 50).astype(np.uint8)
+        wave_normalized = ((wave + 1) / 2 * 40).astype(np.uint8)
         
-        # Apply wave to one color channel
-        frame_np[:, :, (frame_idx // 10) % 3] = np.clip(
-            frame_np[:, :, (frame_idx // 10) % 3].astype(np.int16) + wave_normalized, 
+        rgb_frame[:, :, (frame_idx // 20) % 3] = np.clip(
+            rgb_frame[:, :, (frame_idx // 20) % 3].astype(np.int16) + wave_normalized, 
             0, 255
         ).astype(np.uint8)
         
-        yield frame_np
+        bgr_frame = rgb_frame[:, :, ::-1]
         
-        time.sleep(0.033) #30fps
+        yield bgr_frame
+        time.sleep(0.033) # 30 fps
     
     print("🎉 Dummy stream finished!")
 
@@ -66,7 +52,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Dummy FastRTC Demo") as demo:
             with gr.Group():
                 prompt = gr.Textbox(
                     label="Prompt", 
-                    placeholder="Enter any prompt (this is just for demo, colors will be random)", 
+                    placeholder="Enter any prompt to change the random seed", 
                     lines=3,
                     value="A colorful animated demo"
                 )
@@ -75,7 +61,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Dummy FastRTC Demo") as demo:
             
         with gr.Column(scale=3):
             gr.Markdown("### 2. View Live Stream")
-            gr.Markdown("Click 'Start Generation' below to begin streaming random colored frames.")
+            gr.Markdown("Click 'Start Generation' to begin streaming frames.")
             
             webrtc_output = WebRTC(
                 label="Video Stream",