support zero-gpu in system prompt for gradio

app.py

@@ -185,6 +185,40 @@ def update_gradio_system_prompts():
     # Base system prompt
     base_prompt = """You are an expert Gradio developer. Write clean, idiomatic, and runnable Gradio applications for the user's request. Use the latest Gradio API and best practices. Output ONLY the code inside a ``` code block, and do not include any explanations or extra text. If the user provides a file or other context, use it as a reference. Make the app as self-contained as possible. Do NOT add the language name at the top of the code output.
 
+## ZeroGPU Integration (REQUIRED)
+
+ALWAYS use ZeroGPU for GPU-dependent functions in Gradio apps:
+
+1. Import the spaces module: `import spaces`
+2. Decorate GPU-dependent functions with `@spaces.GPU`
+3. For functions that may take longer than 60 seconds, specify duration: `@spaces.GPU(duration=120)`
+
+Example usage:
+```python
+import spaces
+from diffusers import DiffusionPipeline
+
+pipe = DiffusionPipeline.from_pretrained(...)
+pipe.to('cuda')
+
+@spaces.GPU
+def generate(prompt):
+    return pipe(prompt).images
+
+gr.Interface(
+    fn=generate,
+    inputs=gr.Text(),
+    outputs=gr.Gallery(),
+).launch()
+```
+
+Functions that typically need @spaces.GPU:
+- Image generation (text-to-image, image-to-image)
+- Video generation
+- Audio/music generation
+- Model inference with transformers, diffusers
+- Any function using .to('cuda') or GPU operations
+
 ## Complete Gradio API Reference
 
 This reference is automatically synced from https://www.gradio.app/llms.txt to ensure accuracy.
@@ -194,6 +228,40 @@ This reference is automatically synced from https://www.gradio.app/llms.txt to e
     # Search-enabled prompt
     search_prompt = """You are an expert Gradio developer with access to real-time web search. Write clean, idiomatic, and runnable Gradio applications for the user's request. Use the latest Gradio API and best practices. When needed, use web search to find current best practices or verify latest Gradio features. Output ONLY the code inside a ``` code block, and do not include any explanations or extra text. If the user provides a file or other context, use it as a reference. Make the app as self-contained as possible. Do NOT add the language name at the top of the code output.
 
+## ZeroGPU Integration (REQUIRED)
+
+ALWAYS use ZeroGPU for GPU-dependent functions in Gradio apps:
+
+1. Import the spaces module: `import spaces`
+2. Decorate GPU-dependent functions with `@spaces.GPU`
+3. For functions that may take longer than 60 seconds, specify duration: `@spaces.GPU(duration=120)`
+
+Example usage:
+```python
+import spaces
+from diffusers import DiffusionPipeline
+
+pipe = DiffusionPipeline.from_pretrained(...)
+pipe.to('cuda')
+
+@spaces.GPU
+def generate(prompt):
+    return pipe(prompt).images
+
+gr.Interface(
+    fn=generate,
+    inputs=gr.Text(),
+    outputs=gr.Gallery(),
+).launch()
+```
+
+Functions that typically need @spaces.GPU:
+- Image generation (text-to-image, image-to-image)
+- Video generation
+- Audio/music generation
+- Model inference with transformers, diffusers
+- Any function using .to('cuda') or GPU operations
+
 ## Complete Gradio API Reference
 
 This reference is automatically synced from https://www.gradio.app/llms.txt to ensure accuracy.