Update app.py from anycoder

app.py

@@ -1,10 +1,24 @@
 import gradio as gr
 import numpy as np
-from PIL import Image
+from PIL import Image, ImageDraw, ImageFont
 import io
 import time
 import random
+import base64
 from typing import List, Dict, Any
+from functools import wraps
+
+# ZeroGPU decorator
+def zero_gpu(func):
+    """
+    Decorator to indicate that this function can run on CPU-only environments.
+    This is useful for deployment on platforms without GPU access.
+    """
+    @wraps(func)
+    def wrapper(*args, **kwargs):
+        return func(*args, **kwargs)
+    wrapper.zero_gpu = True
+    return wrapper
 
 # Configuration
 CONFIG = {
@@ -21,6 +35,7 @@ class ImageSequenceGenerator:
         self.playback_interval: Any = None
         self.is_generating: bool = False
 
+    @zero_gpu
     def generate_random_image(self, prompt: str) -> str:
         """Generate a random colorful image with the prompt text"""
         width, height = CONFIG["image_size"]
@@ -107,23 +122,27 @@ class ImageSequenceGenerator:
         img_str = f"data:image/png;base64,{base64.b64encode(buffered.getvalue()).decode()}"
         return img_str
 
+    @zero_gpu
     def add_image_to_buffer(self, image_data: str):
         """Add image to buffer"""
         self.images.append(image_data)
         return self.get_state()
 
+    @zero_gpu
     def remove_image_from_buffer(self, index: int):
         """Remove image from buffer"""
         if 0 <= index < len(self.images):
             self.images.pop(index)
         return self.get_state()
 
+    @zero_gpu
     def clear_buffer(self):
         """Clear the entire buffer"""
         self.images = []
         self.stop_sequence()
         return self.get_state()
 
+    @zero_gpu
     def play_sequence(self):
         """Start playing the sequence"""
         if self.is_playing or len(self.images) < CONFIG["required_images"]:
@@ -133,6 +152,7 @@ class ImageSequenceGenerator:
         self.current_frame = 0
         return self.get_state()
 
+    @zero_gpu
     def stop_sequence(self):
         """Stop playing the sequence"""
         if not self.is_playing:
@@ -142,6 +162,7 @@ class ImageSequenceGenerator:
         self.current_frame = 0
         return self.get_state()
 
+    @zero_gpu
     def get_current_frame(self):
         """Get the current frame for display"""
         if not self.is_playing or not self.images:
@@ -151,6 +172,7 @@ class ImageSequenceGenerator:
         self.current_frame = (self.current_frame + 1) % len(self.images)
         return frame_data
 
+    @zero_gpu
     def get_state(self) -> Dict[str, Any]:
         """Get current state for UI updates"""
         return {
@@ -161,6 +183,7 @@ class ImageSequenceGenerator:
             "images": self.images
         }
 
+    @zero_gpu
     def generate_image(self, prompt: str, negative_prompt: str, steps: int, guidance_scale: float, seed: int, batch_size: int):
         """Generate a new image"""
         if self.is_generating:
@@ -335,7 +358,7 @@ def create_gradio_app():
     }
     """
 
-    with gr.Blocks(css=custom_css) as demo:
+    with gr.Blocks() as demo:
         gr.Markdown("""
         <div style="text-align: center; margin-bottom: 2rem;">
             <h1 style="color: #6c5ce7; font-size: 2.5rem; margin-bottom: 0.5rem;">
@@ -540,16 +563,12 @@ def create_gradio_app():
         # Timer for playback
         timer = gr.Timer(1.0 / CONFIG["playback_fps"])
 
+        @zero_gpu
         def update_player_display():
             if generator.is_playing and generator.images:
                 frame_data = generator.get_current_frame()
                 if frame_data:
                     # Convert base64 to PIL Image
-                    from PIL import Image
-                    import io
-                    import base64
-
-                    # Extract base64 data
                     header, encoded = frame_data.split(",", 1)
                     binary_data = base64.b64decode(encoded)
                     image = Image.open(io.BytesIO(binary_data))