Update app.py

app.py

@@ -3,12 +3,13 @@ import sys
 import subprocess
 import os
 sys.path.append("./")
+sys.path.append("../")
 import BotSimulator
 from BotSimulator import TootBot
 import openai
 from io import StringIO
 import multiprocessing
-restore_point = sys.stdout
+restore_point=sys.stdout
 
 
 if not os.path.exists("data/fake-tweets"):
@@ -25,6 +26,7 @@ subtitle_2 = """<h2 align="center">Run bot simulation</h2>"""
 image_tokens_list = ["black and white", "year 2023", "cartoon", "animated",
                      "comic", "propaganda", "news", "classic disney style",
                      "holliemengert artstyle"]
+topic_prompt_default = "Write a short tweet with less than 500 characters as if you were a real person with social media lingo and hashtags on this topic: "
 pool = multiprocessing.Pool()
 num_processes = pool._processes
 global tootbot_app
@@ -36,7 +38,8 @@ def init_app(api_key,
              temperature,
              diffusion_model,
              keyword_model,
-             dtype):
+             dtype,
+             topic_prompt):
 
     try:
         openai.api_key = api_key
@@ -46,7 +49,7 @@ def init_app(api_key,
 
     console_logs = StringIO()
     sys.stdout = console_logs
-    model_class = model_class.replace(" ","").lower()
+    model_class = model_class.replace(" ", "").lower()
     global tootbot_app
     tootbot_app = TootBot(model=model_name,
                           model_class=model_class,
@@ -63,7 +66,8 @@ def init_app(api_key,
     os.remove("client_cred.secret")
     os.environ["OPENAI_API_KEY"] = api_key
     os.environ["OPENAI_API_KEY_2"] = api_key
-    tootbot_app.init_models(diffusion_model=diffusion_model, 
+    tootbot_app.topic_prescript = topic_prompt.strip() + " "
+    tootbot_app.init_models(diffusion_model=diffusion_model,
                             keyword_model=keyword_model,
                             text_fail_classifier="davidna22/text-failed-classifier",
                             dtype=dtype,
@@ -93,7 +97,7 @@ def run_simulation(topic,
                    text_model_name="gpt-3.5-turbo-0301"):
     if topic == "":
         raise gr.Error("Topic must not be empty")
-             
+
     console_logs = StringIO()
     sys.stdout = console_logs
     filename = topic.replace(" ", "-") + ".csv"
@@ -168,6 +172,7 @@ def reset_app():
         log_output: console_logs.getvalue(),
         api_key: gr.update(value=""),
         topic: gr.update(value=""),
+        topic_prompt: gr.update(value=topic_prompt_default),
         with_images: gr.update(value=False),
         image_params: gr.update(visible=False),
         main_block_step_1: gr.update(visible=True),
@@ -182,7 +187,7 @@ 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:
+        with gr.Column(elem_id="main_block_step_1", visible=True) as main_block_step_1:
             gr.HTML(value=subtitle_1)
             api_key = gr.Textbox(label="Enter your API key")
             model_class = gr.Dropdown(value="Open AI",
@@ -199,6 +204,8 @@ with gr.Blocks() as demo:
                                               "gpt-4-32k-0314"])
             temperature = gr.Number(label="Set Your Temperature (0.0-2.0)", value=0.9)
             with gr.Accordion("Additional Parameters", open=False):
+                topic_prompt = gr.Textbox(label="Enter the Topic Prompt (Default Example below):",
+                                          value=topic_prompt_default)
                 diffusion_model = gr.Dropdown(value="stabilityai/stable-diffusion-2-1-base",
                                               choices=["stabilityai/stable-diffusion-2-1-base",
                                                        "stabilityai/stable-diffusion-2-base",
@@ -343,7 +350,8 @@ with gr.Blocks() as demo:
                                      temperature,
                                      diffusion_model,
                                      keyword_model,
-                                     dtype],
+                                     dtype,
+                                     topic_prompt],
                              outputs=[progress_output,
                                       log_output,
                                       main_block_step_1,
@@ -357,6 +365,7 @@ with gr.Blocks() as demo:
                                                            log_output,
                                                            api_key,
                                                            topic,
+                                                           topic_prompt,
                                                            with_images,
                                                            image_params,
                                                            main_block_step_1,
@@ -390,4 +399,4 @@ with gr.Blocks() as demo:
     accept_checkbox.change(fn=enable_inputs, inputs=[], outputs=[user_consent_block, main_block], queue=False)
 
     demo.queue(concurrency_count=num_processes)
-    demo.launch(share=False)
+    demo.launch(share=True)