Fix caching and index errors

app.py

@@ -3,6 +3,7 @@ import numpy as np
 import random
 import logging
 import sys
+import os
 
 # 设置日志记录
 logging.basicConfig(level=logging.INFO, 
@@ -10,6 +11,9 @@ logging.basicConfig(level=logging.INFO,
                     stream=sys.stdout)
 logger = logging.getLogger(__name__)
 
+# 确保缓存目录存在
+os.makedirs(".gradio/cached_examples", exist_ok=True)
+
 # 修复 Gradio JSON Schema 错误
 try:
     import gradio_client.utils
@@ -87,12 +91,17 @@ def generate_image(prompt):
         seed = random.randint(0, MAX_SEED)
         generator = torch.Generator().manual_seed(seed)
         
-        image = pipe(
-            prompt=prompt,
-            guidance_scale=0.0,
-            num_inference_steps=2,
-            generator=generator
-        ).images[0]
+        # 添加异常处理，防止数组索引越界
+        try:
+            image = pipe(
+                prompt=prompt,
+                guidance_scale=0.0,
+                num_inference_steps=2,
+                generator=generator
+            ).images[0]
+        except IndexError as e:
+            logger.error(f"Index error in pipe: {str(e)}")
+            return None
         
         logger.info("Image generation successful")
         return image
@@ -100,20 +109,26 @@ def generate_image(prompt):
         logger.error(f"Error generating image: {str(e)}")
         return None
 
-# 创建简单的 Gradio 界面
+# 创建简单的 Gradio 界面，禁用示例缓存
 demo = gr.Interface(
     fn=generate_image,
     inputs=gr.Textbox(label="Enter your prompt"),
     outputs=gr.Image(label="Generated Image"),
     title="SDXL Turbo Text-to-Image",
     description="Enter a text prompt to generate an image.",
-    examples=["A cute cat", "Sunset over mountains", "Astronaut in space"]
+    examples=["A cute cat", "Sunset over mountains"],  # 减少示例数量
+    cache_examples=False  # 禁用示例缓存以避免文件访问错误
 )
 
 # 启动应用
 if __name__ == "__main__":
     try:
         logger.info("Starting Gradio app")
-        demo.launch()
+        # 添加更多启动选项，以提高稳定性
+        demo.launch(
+            show_error=True,
+            max_threads=1,  # 降低并发以避免竞态条件
+            share=False
+        )
     except Exception as e:
         logger.error(f"Error launching app: {str(e)}")