more refactor

app.py

@@ -5,6 +5,7 @@ from nbconvert import HTMLExporter
 from traitlets.config import Config
 import os
 import shutil
+import base64
 
 # Configuration for HTMLExporter
 config = Config()
@@ -21,11 +22,29 @@ def reset_tmp_folder():
         shutil.rmtree(TMP_DIR)
     os.makedirs(TMP_DIR)
 
+
+def embed_figures(html_body, resources):
+    for key, value in resources['outputs'].items():
+        b64_figure = base64.b64encode(value).decode('utf-8')
+        img_tag = f'data:image/png;base64,{b64_figure}'
+        html_body = html_body.replace(key, img_tag)
+    return html_body
+
 def parse_notebook():
     reset_tmp_folder()
-    notebook_data = next(ds_iter)
-    notebook_string = notebook_data["text"]
-    notebook_id = notebook_data["id"].split("/")[-1]
+
+    found_notebook = False
+    counter = 0
+    while not found_notebook:
+        counter+=1
+        print(counter)
+        notebook_data = next(ds_iter)
+        notebook_string = notebook_data["text"]
+        notebook_id = notebook_data["id"].split("/")[-1]
+
+        if len(notebook_string)>1 * 1024 * 1024:
+            found_notebook = True
+
     out_path = os.path.join(TMP_DIR, notebook_id)
     
     # Save the notebook string to a file
@@ -34,25 +53,17 @@ def parse_notebook():
     
     notebook_parsed = nbformat.reads(notebook_string, as_version=4)
     (notebook_body, resources) = html_exporter.from_notebook_node(notebook_parsed)
-    
-    print("Resources:", resources["outputs"])
+    notebook_body = embed_figures(notebook_body, resources)
+    print("Resources:", resources["outputs"].keys())
     return notebook_body, out_path
 
 
 with gr.Blocks() as demo:
     gr.Markdown("# Kaggle Notebooks")
-    
-    with gr.Row():
-        button = gr.Button("Show next!")
-        #button_download = gr.Button("Download notebook")
-    
-    with gr.Row():
-        with gr.Column():
-            html = gr.HTML("")
+    button = gr.Button("Show next!")
+    file = gr.File()
+    html = gr.HTML("")
 
-    with gr.Row():
-        with gr.Column():
-            file = gr.File()
     
     button.click(fn=parse_notebook, inputs=[], outputs=[html, file])
     demo.load(fn=parse_notebook, inputs=[], outputs=[html, file])