Upload app.py

app.py

@@ -84,6 +84,7 @@ def run_simulation(topic,
                    news_company="CNN",
                    text_model_name=""):
     console_logs = StringIO()
+    sys.stdout = console_logs
     filename = topic.replace(" ", "-") + ".csv"
     tweet_folder="data/fake-tweets"
     tweet_failfolder="data/failed-tweets"
@@ -120,10 +121,11 @@ def run_simulation(topic,
     subfolder = topic.replace(" ", "-")
     img_filename = f"{image_folder}/{subfolder}/tweet-img-row-0.png"
     img_fullpath = os.path.join(os.path.dirname(__file__), img_filename)
-
+    sys.stdout=restore_point
     return {
-        progress_box: "Simulation finished. To see your results, visit the bot simulation Mastodon server: https://bot-simulation-research.app/home",
+        progress_box: "Simulation finished. To see your results, visit the bot simulation Mastodon server.",
         log_output: console_logs.getvalue(),
+        mastodon_link: gr.update(visible=True),
         simulation_output_box: gr.update(visible=True),
         saved_file: tweet_filename,
         example_image: img_fullpath
@@ -273,6 +275,7 @@ with gr.Blocks() as demo:
         with gr.Row(elem_id="progress_box") as progress_box:
             progress_box = gr.Textbox(label="Current Progress", show_label=True)
             log_output = gr.Textbox(every=0.5, label="Logs", lines=8)
+            mastodon_link = gr.Markdown("Link to Mastodon Server: [https://bot-simulation-research.app/home](https://bot-simulation-research.app/home)", visible=False)
 
         with gr.Row(elem_id="simulation_output_box", visible=False) as simulation_output_box:
             saved_file = gr.File(label="List of Generated Tweets (CSV)")
@@ -296,6 +299,7 @@ with gr.Blocks() as demo:
                                     ],
                              outputs=[progress_box,
                                       log_output,
+                                      mastodon_link,
                                       simulation_output_box,
                                       saved_file,
                                       example_image