new

app.py

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