Update BotSimulator.py

BotSimulator.py

@@ -1335,101 +1335,4 @@ class TootBot:
         """
         model_id = model_id
         self.keyword_model = KeyphraseExtractionPipeline(model=model_id)
-        print(f"Keyword Extractor model {model_id} initialized")
-
-
-if __name__ == '__main__':
-    print("------------------------------------------------------------------")
-    print("Initializing Bot Simulation Application...")
-    print("------------------------------------------------------------------")
-    # model_class = input("Enter Model Class (Only supports \"openai\" as of 03/20/2023): ")
-    # if model_class == "":
-    #     model_class = "openai"
-    # model = input("Enter Model Name (default: \"gpt-3.5-turbo-0301\"): ")
-    # if model == "":
-    #     model = "gpt-3.5-turbo-0301"
-    # temperature = float(input("Enter Model Temperature (determinism parameter: 0.0-1.0): "))
-    model_class = "openai"
-    model = "gpt-3.5-turbo-0301"
-    temperature = 0.9
-    Tootbot = TootBot(model=model, model_class=model_class, temperature=temperature)
-
-    print("------------------------------------------------------------------")
-    print("Initializing Mastodon Credentials...")
-    print("------------------------------------------------------------------")
-    # username = input("Enter your Mastodon username: ")
-    # password = input("Enter your Mastodon password: ")
-    # redirect_uri = input("Enter your redirect_uri (if not known, use default. Click enter): ")
-    # if redirect_uri == "":
-    #     redirect_uri = "http://localhost:8080"
-    # code= "<AUTHORIZATION_CODE>"
-    username="research-team@bot-simulation-research.app"
-    password="DavidSamYungjun"
-    redirect_uri="http://localhost:8080"
-    client_id="mastodon/clientcred.secret"
-    to_file="mastodon/usercred.secret"
-    # client_id = input("Enter Mastodon Client Secret filename: ")
-    # to_file = input("Save results to file (default = \"usercred.secret\"): ")
-    # if to_file == "":
-    #     to_file = "usercred.secret"
-    Tootbot.mastodon_login(username=username,
-                       password=password,
-                       redirect_uri=redirect_uri,
-                       client_id=client_id,
-                       to_file=to_file)
-
-    print("------------------------------------------------------------------")
-    print("Initializing Models")
-    print("------------------------------------------------------------------")
-    diff_model = input("Enter Diffusion Model Name (Huggingface): ")
-    if diff_model == "":
-        diff_model = "stabilityai/stable-diffusion-2-1-base"
-    keyword_model = input("Enter keyword extraction model name (if none, default is used): ")
-    if keyword_model == "":
-        keyword_model = "ml6team/keyphrase-extraction-distilbert-inspec"
-    text_fail_classifier = "models/text-failed-classifier"
-    dtype = input("Enter torch dtype (valid responses: [\"float32\", \"float16\"]): ")
-    if dtype == "":
-        dtype = "float32"
-    device = input("Enter device (\"gpu\" or \"cpu\"): ")
-    if device == "":
-        device = "gpu"
-    Tootbot.init_models(diffusion_model=diff_model, 
-                        keyword_model=keyword_model,
-                        text_fail_classifier=text_fail_classifier,
-                        dtype="float32",
-                        device="cuda")
-    print("------------------------------------------------------------------")
-    print("Running Text generation")
-    print("------------------------------------------------------------------")
-    topic = input("Enter the topic for your toots/tweets: ")
-    num_responses = int(input("Enter number of toots/tweets to generate: "))
-
-    save = input("Save results? (Y/n) ")
-    if save == "Y" or save == "y":
-        save = True
-    elif save == "N" or save == "n":
-        save = False
-    with_images = input("Run with images? (Y/n) ")
-    if with_images == "Y" or with_images == "y":
-        with_images = True
-    elif with_images == "N" or with_images == "n":
-        with_images = False
-    use_keyword_extractor = input("Use keyword extractor? (Y/n) ")
-    if use_keyword_extractor == "Y" or use_keyword_extractor == "y":
-        use_keyword_extractor = True
-    elif use_keyword_extractor == "N" or use_keyword_extractor == "n":
-        use_keyword_extractor = False
-    image_tokens = input("Enter image tokens (ex: [black and white, year 2023]): ")
-    image_tokens = [im.strip() for im in image_tokens.strip('][').replace("\"", "").split(',')]
-    n = 50
-    image_every_n_posts = 10
-    folder = "images"
-    topic_filename = topic.replace(" ", "-") + ".csv"
-    Tootbot.reset_bot()
-    new_data_df = Tootbot.run_bot_simulation(topic, num_responses=num_responses, n=n, save=True, filename=topic_filename)
-    Tootbot.post(with_images=with_images, use_keyword_extractor=use_keyword_extractor, image_every_n_posts=image_every_n_posts, folder=folder, image_tokens=image_tokens)
-    print("------------------------------------------------------------------")
-    print("APPLICATION FINISHED SUCCESSFULLY")
-    print("View results at https://bot-simulation-research.app/home")
-    print("------------------------------------------------------------------")
+        print(f"Keyword Extractor model {model_id} initialized")