new

app.py

@@ -29,6 +29,10 @@ subprocess.run(
         "pip install ./extension/renderutils_plugin-0.1.0-cp310-cp310-linux_x86_64.whl --force-reinstall --no-deps"
     )
 )
+# 状态变量，用于保存是否已经生成了详细提示和图像
+generated_detailed_prompt = False
+generated_image = False
+
 def install_cuda_toolkit():
     # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
     # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
@@ -480,20 +484,17 @@ with gr.Blocks(css="""
     # )
 
 
-    # 状态变量，用于保存是否已经生成了详细提示和图像
-    generated_detailed_prompt = False
-    generated_image = False
 
     # Step 1: 点击 "Refine to detailed prompt" 按钮时
     def refine_to_detailed(prompt, seed):
-        nonlocal generated_detailed_prompt
+        global generated_detailed_prompt
         detailed_prompt_value = text_to_detailed(prompt, seed)
         generated_detailed_prompt = True
         return detailed_prompt_value
 
     # Step 2: 点击 "Generate Image" 按钮时
     def generate_image(prompt, seed):
-        nonlocal generated_image, generated_detailed_prompt
+        global generated_image, generated_detailed_prompt
         # 如果没有生成过详细描述，先执行 refine-to-detailed
         if not generated_detailed_prompt:
             detailed_prompt_value = text_to_detailed(prompt, seed)
@@ -506,13 +507,13 @@ with gr.Blocks(css="""
 
     # Step 3: 点击 "Generate Mesh" 按钮时
     def generate_mesh(image):
-        nonlocal generated_detailed_prompt, generated_image
+        global generated_detailed_prompt, generated_image
         # 如果没有生成图像，先执行生成图像
         if not generated_image:
             detailed_prompt_value = text_to_detailed(prompt, seed1)
             image = text_to_image(detailed_prompt_value, seed1)
             generated_image = True
-
+        generated_detailed_prompt, generated_image = False, False
         # 生成网格
         return bundle_image_to_mesh(image)
     # Button Click Events