Update app.py

app.py

@@ -62,7 +62,7 @@ def init_app(api_key,
                             dtype=dtype,
                             device="cuda")
     return {
-        progress_box: "### Current Progress \n Model Initialized Successfully! Time to Run the Simulation",
+        progress_output: "### Current Progress \n Model Initialized Successfully! Time to Run the Simulation",
         log_output: console_logs.getvalue(),
         main_block_step_1: gr.update(visible=False),
         main_block_step_2: gr.update(visible=True),
@@ -70,6 +70,7 @@ def init_app(api_key,
         run_sim_btn: gr.update(visible=True)
     }
 
+
 def run_simulation(topic,
                    save=True,
                    num_responses=50,
@@ -121,15 +122,16 @@ 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
+    sys.stdout = restore_point
     return {
-        progress_box: "### Current Progress \n To see your results, visit the bot simulation Mastodon server. \n Link to Mastodon Server: [https://bot-simulation-research.app/home](https://bot-simulation-research.app/home)",
+        progress_output: "### Current Progress \n To see your results, visit the bot simulation Mastodon server. \n Link to Mastodon Server: [https://bot-simulation-research.app/home](https://bot-simulation-research.app/home)",
         log_output: console_logs.getvalue(),
         simulation_output_box: gr.update(visible=True),
         saved_file: tweet_filename,
         example_image: img_fullpath
     }
 
+
 def toggle_image_params(with_images):
     if with_images:
         return {
@@ -140,6 +142,7 @@ def toggle_image_params(with_images):
             image_params: gr.update(visible=False)
         }
 
+
 def add_token_func(add_token):
     image_tokens_list.append(add_token)
     return {
@@ -147,8 +150,27 @@ def add_token_func(add_token):
     }
 
 
+def reset_app():
+    sys.stdout = restore_point
+    console_logs = StringIO()
+    return {
+        progress_output: gr.update(value="### Current Progress <br>"),
+        log_output: console_logs.getvalue(),
+        api_key: gr.update(value=""),
+        topic: gr.update(value=""),
+        with_images: gr.update(value=False),
+        image_params: gr.update(visible=False),
+        main_block_step_1: gr.update(visible=True),
+        main_block_step_2: gr.update(visible=False),
+        simulation_output_box: gr.update(visible=False),
+        run_sim_btn: gr.update(visible=False),
+        init_btn: gr.update(visible=True)
+    }
+
+
 with gr.Blocks() as demo:
     gr.HTML(title)
+    demo_state = gr.State("Terms")
     with gr.Column(elem_id="main_block", visible=False) as main_block:
         with gr.Column(elem_id = "main_block_step_1", visible=True) as main_block_step_1:
             gr.HTML(value=subtitle_1)
@@ -194,7 +216,7 @@ with gr.Blocks() as demo:
             topic = gr.Textbox(label="Enter the topic to Generate tweets (Ex: \"The Earth is Flat\")")
             save = gr.Checkbox(label="Download Generated Tweets CSV? (Y/n)", value=True)
             with gr.Accordion("Additional Parameters", open=False):
-                num_responses = gr.Slider(label= "Number of total fake Tweets to generate",minimum=20, maximum=500, step=5)
+                num_responses = gr.Slider(label= "Number of total fake Tweets to generate", value=20, minimum=20, maximum=500, step=5)
                 n = gr.Slider(label="Number of Tweets to return per API call (smaller = More randomized answers)", value=10, minimum=1, maximum=50, step=1)
                 system_prompt = gr.Dropdown(label="",
                                             value=BotSimulator.assistant_prompt,
@@ -272,7 +294,7 @@ with gr.Blocks() as demo:
 
         with gr.Row(elem_id="progress_box") as progress_box:
             with gr.Column():
-                progress_box = gr.Markdown("### Current Progress <br>")
+                progress_output = gr.Markdown("### Current Progress <br>")
             with gr.Column():
                 log_output = gr.Textbox(every=0.5, label="Logs", lines=8)
 
@@ -293,9 +315,10 @@ with gr.Blocks() as demo:
                                      augment_mode,
                                      image_every_n_posts,
                                      image_subtoken,
+                                     image_tokens,
                                      news_company,
                                      text_model_name],
-                             outputs=[progress_box,
+                             outputs=[progress_output,
                                       log_output,
                                       simulation_output_box,
                                       saved_file,
@@ -311,16 +334,26 @@ with gr.Blocks() as demo:
                                      diffusion_model,
                                      keyword_model,
                                      dtype],
-                             outputs=[progress_box,
+                             outputs=[progress_output,
                                       log_output,
                                       main_block_step_1,
                                       main_block_step_2,
                                       init_btn,
                                       run_sim_btn])
-            
+
         with gr.Row(elem_id="reset_block", visible=True) as reset_block:
             reset_btn = gr.Button("Reset")
-            # reset_btn.click()
+            reset_btn.click(reset_app, inputs=[], outputs=[progress_output,
+                                                           log_output,
+                                                           api_key,
+                                                           topic,
+                                                           with_images,
+                                                           image_params,
+                                                           main_block_step_1,
+                                                           main_block_step_2,
+                                                           simulation_output_box,
+                                                           run_sim_btn,
+                                                           init_btn])
 
     with gr.Column(elem_id = "user_consent_container") as user_consent_block:
         accept_checkbox = gr.Checkbox(visible=False)