new

app.py

@@ -472,12 +472,55 @@ with gr.Blocks(css="""
 
     # Button Click Events
     # Text2
-    btn_text2detailed.click(fn=text_to_detailed, inputs=[prompt, seed1], outputs=detailed_prompt)
-    btn_text2img.click(fn=text_to_image, inputs=[detailed_prompt, seed1], outputs=output_image1)
-    btn_gen_mesh.click(fn=bundle_image_to_mesh, inputs=[output_image1,], outputs=[output_video1, download_1]).then(
-        lambda: gr.Button(interactive=True),
-        outputs=[download_1],
-    )
+    # btn_text2detailed.click(fn=text_to_detailed, inputs=[prompt, seed1], outputs=detailed_prompt)
+    # btn_text2img.click(fn=text_to_image, inputs=[detailed_prompt, seed1], outputs=output_image1)
+    # btn_gen_mesh.click(fn=bundle_image_to_mesh, inputs=[output_image1,], outputs=[output_video1, download_1]).then(
+    #     lambda: gr.Button(interactive=True),
+    #     outputs=[download_1],
+    # )
+
+
+    # 状态变量，用于保存是否已经生成了详细提示和图像
+    generated_detailed_prompt = False
+    generated_image = False
+
+    # Step 1: 点击 "Refine to detailed prompt" 按钮时
+    def refine_to_detailed(prompt, seed):
+        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):
+        # 如果没有生成过详细描述，先执行 refine-to-detailed
+        if not generated_detailed_prompt:
+            detailed_prompt_value = text_to_detailed(prompt, seed)
+            generated_detailed_prompt = True
+        else:
+            detailed_prompt_value = f"Already refined: {prompt} (Seed: {seed})"
+
+        generated_image = True
+        return text_to_image(detailed_prompt_value, seed)
+
+    # Step 3: 点击 "Generate Mesh" 按钮时
+    def generate_mesh(image):
+        # 如果没有生成图像，先执行生成图像
+        if not generated_image:
+            detailed_prompt_value = text_to_detailed(prompt, seed1)
+            image = text_to_image(detailed_prompt_value, seed1)
+            generated_image = True
+
+        # 生成网格
+        return bundle_image_to_mesh(image)
+    # Button Click Events
+    btn_text2detailed.click(fn=refine_to_detailed, inputs=[prompt, seed1], outputs=detailed_prompt)
+
+    # Generate Image
+    btn_text2img.click(fn=generate_image, inputs=[prompt, seed1], outputs=output_image1)
+
+    # Generate Mesh
+    btn_gen_mesh.click(fn=generate_mesh, inputs=[output_image1], outputs=[output_video1, download_1])
+        
     with gr.Row():
         pass
     with gr.Row():