Update app.py

app.py

@@ -551,31 +551,18 @@ def create_ui(generator: FontMoodGenerator):
             
             # STEP 4: Export and Use
             with gr.Step("💾 Export & Use Your Fonts", id=3):
+
+                theme_css_display_step4 = gr.HTML()
                 gr.Markdown("""
                     ### Step 4: Get the code and use your fonts
                     Copy the CSS code below to use your font palette in your own projects.
                 """)
-                
                 css_code_output = gr.Code(
                     language="css", 
                     label="Your Font Palette CSS",
                     value="/* Generate a palette first to see CSS code here */",
                     lines=15
-                )
-                palette_preview = gr.HTML(label="Font Palette Preview")
-
-                # Update the palette preview when font data changes
-                def update_palette_preview(font_data_json):
-                    if not font_data_json:
-                        return ""
-                    top_hits = [{"corpus_id": item["corpus_id"], "score": item["score"]} for item in font_data_json]
-                    return generator._format_palette_as_html(top_hits)
-                
-                font_data_hidden.change(
-                    fn=update_palette_preview,
-                    inputs=font_data_hidden,
-                    outputs=palette_preview
-                )    
+                ) 
 
                 # Update CSS code when font data changes
                 def update_css_code(font_data_json):
@@ -589,6 +576,15 @@ def create_ui(generator: FontMoodGenerator):
                     inputs=font_data_hidden,
                     outputs=css_code_output
                 )
+
+                def apply_theme_to_step4(theme_css):
+                    return theme_css
+                
+                theme_css_hidden.change(
+                    fn=apply_theme_to_step4,
+                    inputs=theme_css_hidden,
+                    outputs=theme_css_display_step4
+                )
                 
                 gr.Markdown("""
                     **🚀 Next Steps:**