Check the output_dir's value

app.py

@@ -8,6 +8,8 @@ from utils.components_creator import *
 
 finetuning_notebook = "Finetuning_NoteBook"
 
+notebook = None
+
 css = """
 .container {
     align-items: center;
@@ -87,6 +89,7 @@ def get_value(components: dict[Component, Any], elem_id: str) -> Any:
 
 def preview_notebook():
     html_exporter = HTMLExporter()
+    global notebook
     (body, resources) = html_exporter.from_notebook_node(notebook)
 
     html_path = f"{finetuning_notebook}.html"
@@ -96,6 +99,8 @@ def preview_notebook():
 
 
 def generate_code(components: dict[Component, Any]):
+    global notebook
+    notebook = nbf.v4.new_notebook()
     create_install_libraries_cells(notebook['cells'])
     flash_attention_value = get_value(components, FLASH_ATTENTION_ID)
     if flash_attention_value:
@@ -285,7 +290,6 @@ with gr.Blocks(css=css, theme=gr.themes.Soft(text_size='lg', font=["monospace"],
         preview_btn = gr.Button(f"👀 Preview {finetuning_notebook}.ipynb", variant="primary", visible=False)
         download_btn = gr.HTML(show_label=False, visible=True)
 
-    notebook = nbf.v4.new_notebook()
     greet_btn.click(fn=generate_code, inputs=all_components, outputs=[preview_btn, download_btn])
 
     preview_btn.click(fn=preview_notebook, inputs=None, outputs=iframe)