new

app.py

@@ -493,7 +493,7 @@ with gr.Blocks(css="""
         return detailed_prompt_value
 
     # Step 2: 点击 "Generate Image" 按钮时
-    def generate_image(prompt, seed):
+    def generate_image(detailed_prompt_value, seed):
         global generated_image, generated_detailed_prompt
         # 如果没有生成过详细描述，先执行 refine-to-detailed
         if not generated_detailed_prompt:
@@ -503,7 +503,7 @@ with gr.Blocks(css="""
             detailed_prompt_value = f"Already refined: {prompt} (Seed: {seed})"
 
         generated_image = True
-        return text_to_image(detailed_prompt_value, seed)
+        return detailed_prompt_value, text_to_image(detailed_prompt_value, seed)
 
     # Step 3: 点击 "Generate Mesh" 按钮时
     def generate_mesh(image):
@@ -520,7 +520,7 @@ with gr.Blocks(css="""
     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)
+    btn_text2img.click(fn=generate_image, inputs=[detailed_prompt, seed1], outputs=[detailed_prompt, output_image1])
 
     # Generate Mesh
     btn_gen_mesh.click(fn=generate_mesh, inputs=[output_image1], outputs=[output_video1, download_1])