Update app.py

app.py

@@ -36,11 +36,14 @@ def generate_article(prompt, progress=gr.Progress(track_tqdm=True)):
         do_generate_article=True,
         do_polish_article=True,
     )
-    print(response)
     runner.post_run()
     runner.summary()
+    with open(f'{prompt.replace(" ", "_")}/storm_gen_article.txt', 'r') as file:
+        content = file.read()
+    return content
 
 with gr.Blocks() as demo:
+    gr.Markdown("# Omnipedia article generation demo")
     prompt = gr.Textbox(label="Prompt")
     output = gr.Markdown(label="Output")
     btn = gr.Button("Generate")