Fixing Preview

app.py

@@ -95,6 +95,7 @@ def preview_notebook():
     html_path = f"{finetuning_notebook}.html"
     with open(html_path, 'w') as f:
         f.write(body)
+
     return f'<iframe src="file={html_path}" width="100%" height="250px"></iframe>'
 
 
@@ -158,7 +159,6 @@ def generate_code(components: dict[Component, Any]):
 
     report_to = get_value(components, REPORT_TO_ID)
 
-    gr.Warning(f"{output_dir}")
     if not check_valid_input(output_dir):
         gr.Warning("No output_dir is given")
 
@@ -190,7 +190,7 @@ def generate_code(components: dict[Component, Any]):
 
     return gr.Button(
         visible=True), f'''<div class="a_custom"><a href="file={file_name}" download={file_name}>
-        💾️ Download {finetuning_notebook}.ipynb</a> </div> '''
+        💾️ Download {finetuning_notebook}.ipynb</a> </div> ''', "<div></div>"
 
 
 with gr.Blocks(css=css, theme=gr.themes.Soft(text_size='lg', font=["monospace"],
@@ -290,7 +290,7 @@ 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)
 
-    greet_btn.click(fn=generate_code, inputs=all_components, outputs=[preview_btn, download_btn])
+    greet_btn.click(fn=generate_code, inputs=all_components, outputs=[preview_btn, download_btn, iframe])
 
     preview_btn.click(fn=preview_notebook, inputs=None, outputs=iframe)
 

utils/notebook_generator.py

@@ -87,10 +87,10 @@ def create_datasets_cells(cells: list, dataset: FTDataSet, seed: int):
 **Conversational format:**
 
 
-```python {"messages": [{"role": "system", "content": "You are..."}, {"role": "user", "content": "..."}, 
-{"role": "assistant", "content": "..."}]} {"messages": [{"role": "system", "content": "You are..."}, {"role": "user", 
-"content": "..."}, {"role": "assistant", "content": "..."}]} {"messages": [{"role": "system", "content": "You 
-are..."}, {"role": "user", "content": "..."}, {"role": "assistant", "content": "..."}]} """
+```python{"messages": [{"role": "system", "content": "You are..."}, {"role": "user", "content": "..."}, {"role": "assistant", "content": "..."}]}
+{"messages": [{"role": "system", "content": "You are..."}, {"role": "user", "content": "..."}, {"role": "assistant", "content": "..."}]}
+{"messages": [{"role": "system", "content": "You are..."}, {"role": "user", "content": "..."}, {"role": "assistant", "content": "..."}]}
+"""
     text_cell1 = nbf.v4.new_markdown_cell(text)
     text_cell2 = nbf.v4.new_markdown_cell(text_format)
     code = f"""