Upload app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr 
+mport gradio as gr 
 import sys
 import subprocess
 import os
@@ -82,7 +82,7 @@ def run_simulation(topic,
                    image_subtoken="Provide a realistic photograph. ",
                    image_tokens=[],
                    news_company="CNN",
-                   text_model_name=""):
+                   text_model_name="gpt-3.5-turbo-0301"):
     console_logs = StringIO()
     sys.stdout = console_logs
     filename = topic.replace(" ", "-") + ".csv"
@@ -259,16 +259,15 @@ with gr.Blocks() as demo:
                     news_company = gr.Dropdown(label="News Company Logo to use",
                                                value="CNN",
                                                choices=[
-                                                   "CNN"
-                                               ])
+                                                   "CNN"])
                     text_model_name = gr.Dropdown(value="gpt-3.5-turbo-0301",
-                                              label="Select Text Model to use",
-                                              choices=["gpt-3.5-turbo",
-                                                       "gpt-3.5-turbo-0301",
-                                                       "gpt-4",
-                                                       "gpt-4-0314",
-                                                       "gpt-4-32k",
-                                                       "gpt-4-32k-0314"])
+                                                  label="Select Text Model to use",
+                                                  choices=["gpt-3.5-turbo",
+                                                           "gpt-3.5-turbo-0301",
+                                                           "gpt-4",
+                                                           "gpt-4-0314",
+                                                           "gpt-4-32k",
+                                                           "gpt-4-32k-0314"])
             with_images.change(toggle_image_params, with_images, [image_params])
 
         with gr.Row(elem_id="progress_box") as progress_box:
@@ -295,14 +294,12 @@ with gr.Blocks() as demo:
                                      image_every_n_posts,
                                      image_subtoken,
                                      news_company,
-                                     text_model_name
-                                    ],
+                                     text_model_name],
                              outputs=[progress_box,
                                       log_output,
                                       simulation_output_box,
                                       saved_file,
-                                      example_image
-                                     ])
+                                      example_image])
 
         with gr.Row(elem_id="init_btn", visible=True) as init_btn:
             submit_btn = gr.Button("Initialize Bot")
@@ -319,8 +316,11 @@ with gr.Blocks() as demo:
                                       main_block_step_1,
                                       main_block_step_2,
                                       init_btn,
-                                      run_sim_btn
-                                     ])
+                                      run_sim_btn])
+            
+        with gr.Row(elem_id="reset_block", visible=True) as reset_block:
+            reset_btn = gr.Button("Reset")
+            # reset_btn.click()
 
     with gr.Column(elem_id = "user_consent_container") as user_consent_block:
         accept_checkbox = gr.Checkbox(visible=False)