Add code preview '_layout.html' version

app.py

@@ -241,7 +241,7 @@ def process_and_generate(image_input, image_path_from_state, sidebar_prompt, hea
         os.unlink(final_image_path)
     
     # Return all the outputs, including for the state objects
-    return layout_preview, final_preview, layout_code, final_code, gr.update(value=package_url, visible=True)
+    return layout_preview, final_preview, final_code, layout_code, final_code, gr.update(value=package_url, visible=True)
 
 
 with gr.Blocks() as demo:
@@ -314,7 +314,7 @@ with gr.Blocks() as demo:
     generate_btn.click(
         process_and_generate,
         [active_image, active_image_path_state, sidebar_prompt, header_prompt, navigation_prompt, main_content_prompt],
-        [html_preview, html_preview_with_placeholder, layout_code_state, final_code_state, download_button],
+        [html_preview, html_preview_with_placeholder, html_code_output, layout_code_state, final_code_state, download_button],
         show_progress="full"
     )