refactor

app.py

@@ -23,41 +23,38 @@ def reset_tmp_folder():
 
 def parse_notebook():
     reset_tmp_folder()
-    
     notebook_data = next(ds_iter)
     notebook_string = notebook_data["text"]
     notebook_id = notebook_data["id"].split("/")[-1]
+    out_path = os.path.join(TMP_DIR, notebook_id)
     
     # Save the notebook string to a file
-    with open(os.path.join(TMP_DIR, notebook_id), 'w') as f:
+    with open(out_path, 'w') as f:
         f.write(notebook_string)
     
     notebook_parsed = nbformat.reads(notebook_string, as_version=4)
     (notebook_body, resources) = html_exporter.from_notebook_node(notebook_parsed)
     
     print("Resources:", resources["outputs"])
-    return notebook_body
+    return notebook_body, out_path
 
-def download_notebook():
-    for filename in os.listdir(TMP_DIR):
-        if filename.endswith('.ipynb'):
-            path = os.path.join(TMP_DIR, filename)
-            print(path)
-            return path
 
 with gr.Blocks() as demo:
     gr.Markdown("# Kaggle Notebooks")
     
     with gr.Row():
         button = gr.Button("Show next!")
-        button_download = gr.Button("Download notebook")
+        #button_download = gr.Button("Download notebook")
     
     with gr.Row():
         with gr.Column():
             html = gr.HTML("")
+
+    with gr.Row():
+        with gr.Column():
+            file = gr.File()
     
-    button.click(fn=parse_notebook, inputs=[], outputs=[html])
-    button_download.click(fn=download_notebook, inputs=[], outputs=[])
-    demo.load(fn=parse_notebook, inputs=[], outputs=[html])
+    button.click(fn=parse_notebook, inputs=[], outputs=[html, file])
+    demo.load(fn=parse_notebook, inputs=[], outputs=[html, file])
 
 demo.launch()