Spaces:

User1342
/

Ivory

Runtime error

App Files Files Community

User1342 commited on Nov 12, 2022

Commit

f83a470

1 Parent(s): 4bb0c0a

Update app.py

Browse files

Files changed (1) hide show

app.py +68 -75

app.py CHANGED Viewed

@@ -1,15 +1,19 @@
-import json
-import os
-from pprint import pprint
-from mastodon import Mastodon #Mastodon.py
 auth_button_text = '''<a href="{}"><button class="w3-button w3-light-grey w3-padding-large w3-section " onclick="document.getElementById('download').style.display='block'">
-        <i class=""></i> Login with Masterdon! 🐘
       </button></a><br>'''
 def client_log_in():
-    secret_file = open("ivory-secret.txt","w")
     secret_file.write("{}\n".format(os.getenv('app_secret_1')))
     secret_file.write("{}\n".format(os.getenv('app_secret_2')))
     secret_file.write("https://infosec.exchange\n")
@@ -24,45 +28,43 @@ def client_log_in():
     mastodon.log_in(
         os.getenv('username'),
         os.getenv('password'),
-        scopes=["write:blocks","write:mutes"],
         redirect_uri="https://user1342-ivory.hf.space/",
     )
     return mastodon
 def get_auth_url(mastodon):
-    return  mastodon.auth_request_url(client_id="ivory-secret.txt", scopes=["write:blocks","write:mutes"], redirect_uris="https://user1342-ivory.hf.space/")
 def login_from_code(code):
     mastodon = Mastodon(
         client_id="ivory-secret.txt",
         api_base_url='https://infosec.exchange'
     )
     mastodon.log_in(code=code,
-        scopes=["write:blocks","write:mutes"],
-                           redirect_uri="https://user1342-ivory.hf.space/"
-                           )
     return mastodon
-def get_posts_from_user(masterdom, user_id):
-    return masterdom.account_statuses(id=user_id)
-#!/usr/bin/env python
 # coding: utf-8
-import json
-import os
-import re
-import time
-from random import random
-import socket
-from threading import Thread
-from time import sleep
 html_data = '''
 <!DOCTYPE html>
@@ -134,11 +136,12 @@ function showDivs(n) {
 # Imports
 import json
 import os
-import time
 import gradio as gr
 # Setup the gradio block and add some generic CSS
-block = gr.Blocks(css=".container { max-width: 800px; margin: auto; } h1 { margin: 0px; padding: 5px 0; line-height: 50px; font-size: 60pt; }.close-heading {margin: 0px; padding: 0px;} .close-heading p { margin: 0px; padding: 0px;}", title="WatchTower")
 # Chat history variable used for the chatbot prompt on the 'getting started' page.
 chat_history = []
@@ -146,26 +149,24 @@ chat_history = []
 def get_client_from_tokens(code):
     '''
-    This function is used for generating a Tweepy client object based on Oauth verifier and token paramiters
-    :param oauth_verifier:
-    :param oauth_token:
-    :return: A Tweepy client object
     '''
-    return  login_from_code(code)
 def block_user(user_id, user, reason):
     finished = False
     blocked = True
     while not finished:
         client.account_block(user_id)
         print("Blocked {}, for {}".format(user, reason))
         return blocked
 def block_users(client, threshold, dataset):
     '''
     Used for blocking a series of users based on the threshold and datasets provided. Here the users folder is used.
@@ -184,37 +185,25 @@ def block_users(client, threshold, dataset):
         for user in users:
             print("Reviewing user {}".format(user))
-            if "threshold" in user:
-                # old type of dataset being used, only 'violent' data available
-                if "Violent" in dataset:
-                    if user["threshold"] >= threshold:
-                        user_id = str(user["username"])
-                        if block_user(user_id, user, "Violent - old dataset"):
-                            num_users_blocked = num_users_blocked + 1
-            else:
-                # modern dataset being used
-                if "Violent" in dataset:
-                    if user["violence-threshold"] >= threshold:
-                        user_id = str(user["username"])
-                        if block_user(user_id, user, "Violent"):
-                            num_users_blocked = num_users_blocked + 1
-                        continue
-                if "Hate Speech"  in dataset:
-                    if user["toxicity-threshold"] >= threshold:
-                        user_id = str(user["username"])
-                        if block_user(user_id, user, "Hate Speech"):
-                            num_users_blocked = num_users_blocked + 1
-                        continue
     return num_users_blocked
 def chat(selected_option=None, radio_score=None, url_params=None):
     '''
-    This function is used to initialise blocking users once the user has authenticated with Twitter.
     :param selected_option:
     :param radio_score:
     :param url_params:
@@ -243,7 +232,8 @@ def chat(selected_option=None, radio_score=None, url_params=None):
             # Display to user, set options
             history.append(
-                ["Model tuned to a '{}%' threshold and is using the {} dataset.".format(radio_score, block_type.capitalize()),
                  "{} Account blocking initialised".format(block_type.capitalize())])
             num_users_blocked = block_users(client, radio_score, selected_option)
             history.append(
@@ -278,7 +268,7 @@ target_website = None
 def update_target_website():
     '''
-    Updates the URL used to authenticate WatchTower with Twitter.
     #TODO this function is full of old code and can be optimised.
     :return:
     '''
@@ -297,9 +287,9 @@ def update_target_website():
     chatbot.value = chat_history
     chatbot.update(value=chat_history)
-    twitter_auth_button.value = auth_button_text.format(
         get_target_website())
-    twitter_auth_button.update(
         value=auth_button_text.format(
             get_target_website()))
@@ -318,12 +308,14 @@ get_window_url_params = """
         }
     """
 def get_chatbot_text():
-    return [('Welcome to Watchtower.', 'Log in via Twitter and configure your blocking options above.')]
 def get_target_website():
     '''
-    A wrapper function used for retrieving the URL a user will use to authenticate WatchTower with Masterdon.
     :return: auth url
     '''
@@ -331,13 +323,11 @@ def get_target_website():
     return get_auth_url(mastodon)
-# The Gradio HTML component used for the 'sign in with Twitter' button
 # The main chunk of code that uses Gradio blocks to create the UI
 html_button = None
 with block:
     gr.HTML('''
 <meta name="viewport" content="width=device-width, initial-scale=1">
 <link rel="stylesheet" href="https://www.w3schools.com/w3css/4/w3.css">
@@ -351,22 +341,25 @@ with block:
     with gr.Group():
         with gr.Row().style(equal_height=True):
             with gr.Box():
-                #gr.Label(value="WatchTower", visible=True, interactive=False)
                 url_params = gr.JSON({}, visible=False, label="URL Params").style(
                 )
                 text_input = gr.Text(label="Input", visible=False).style()
                 text_output = gr.Text(label="Output", visible=False).style()
-                html_button = twitter_auth_button = gr.HTML(
                     value=auth_button_text.format(
                         get_target_website())).style(
                 )
                 with gr.Row().style(equal_height=True):
-                    radio = gr.CheckboxGroup(value=["Violent", "Hate Speech"], choices=["Violent", "Hate Speech", "Misinformation"],
-                                                 interactive=False, label="Behaviour To Block").style()
                     slider = gr.Slider(value=30, interactive=True, label="Threshold Confidence Tolerance").style()
-                chatbot = gr.Chatbot(label="Watchtower Output", value=[('Welcome to Watchtower.', 'Log in via Mastodon and configure your blocking options above.')]).style(color_map=["grey","purple"])
                 btn = gr.Button("Run WatchTower").style(full_width=True).style()
     btn.click(fn=button_pressed, inputs=[slider, url_params],
@@ -379,7 +372,7 @@ with block:
    </p>"""
     )
-# Setup callback for when page loads (used to set a new Twitter auth target webspage)
 block.__enter__()
 block.set_event_trigger(
     event_name="load", fn=update_target_website, inputs=None, outputs=[html_button], no_target=True
@@ -392,4 +385,4 @@ block.set_event_trigger(
 block.attach_load_events()
 # Launcg the page
-block.launch(enable_queue = True)

+from mastodon import Mastodon  # Mastodon.py
+# A HTML blob of text used to define the 'login in with' button
 auth_button_text = '''<a href="{}"><button class="w3-button w3-light-grey w3-padding-large w3-section " onclick="document.getElementById('download').style.display='block'">
+        <i class=""></i> Login with Mastodon! 🐘
       </button></a><br>'''
 def client_log_in():
+    '''
+     Generates an authenticated Masterdon client for the admin user. Used for retireving the target website URL
+    :return: authenticated Masterdon client variable
+    '''
+    # Setup a 'secret' file used for persisting session between requests.
+    secret_file = open("ivory-secret.txt", "w")
     secret_file.write("{}\n".format(os.getenv('app_secret_1')))
     secret_file.write("{}\n".format(os.getenv('app_secret_2')))
     secret_file.write("https://infosec.exchange\n")
     mastodon.log_in(
         os.getenv('username'),
         os.getenv('password'),
+        scopes=["write:blocks", "write:mutes"],
         redirect_uri="https://user1342-ivory.hf.space/",
     )
     return mastodon
 def get_auth_url(mastodon):
+    '''
+    Retrieves a URL for the user to visit to auth them with WatchTower.
+    :param mastodon: A admin masterdon instance.
+    :return: The target URL.
+    '''
+    return mastodon.auth_request_url(client_id="ivory-secret.txt", scopes=["write:blocks", "write:mutes"],
+                                     redirect_uris="https://user1342-ivory.hf.space/")
 def login_from_code(code):
+    '''
+    Used to create a masterdon client instance based on an authenticated user code (retrieved from the URL).
+    :param code: The code which will authenticate the user/ WatchTower.
+    :return: A masterdon client instance signed in as the user.
+    '''
     mastodon = Mastodon(
         client_id="ivory-secret.txt",
         api_base_url='https://infosec.exchange'
     )
     mastodon.log_in(code=code,
+                    scopes=["write:blocks", "write:mutes"],
+                    redirect_uri="https://user1342-ivory.hf.space/")
     return mastodon
+# !/usr/bin/env python
 # coding: utf-8
 html_data = '''
 <!DOCTYPE html>
 # Imports
 import json
 import os
 import gradio as gr
 # Setup the gradio block and add some generic CSS
+block = gr.Blocks(
+    css=".container { max-width: 800px; margin: auto; } h1 { margin: 0px; padding: 5px 0; line-height: 50px; font-size: 60pt; }.close-heading {margin: 0px; padding: 0px;} .close-heading p { margin: 0px; padding: 0px;}",
+    title="WatchTower")
 # Chat history variable used for the chatbot prompt on the 'getting started' page.
 chat_history = []
 def get_client_from_tokens(code):
     '''
+    This function is used for generating a masterdon client object based on the code URL paramiters
+    :param code
+    :return: A Masterdon client object
     '''
+    return login_from_code(code)
 def block_user(user_id, user, reason):
     finished = False
     blocked = True
     while not finished:
         client.account_block(user_id)
         print("Blocked {}, for {}".format(user, reason))
         return blocked
 def block_users(client, threshold, dataset):
     '''
     Used for blocking a series of users based on the threshold and datasets provided. Here the users folder is used.
         for user in users:
             print("Reviewing user {}".format(user))
+            if "Violent" in dataset:
+                if user["violence-threshold"] >= threshold:
+                    user_id = str(user["username"])
+                    if block_user(user_id, user, "Violent"):
+                        num_users_blocked = num_users_blocked + 1
+                    continue
+            if "Hate Speech" in dataset:
+                if user["toxicity-threshold"] >= threshold:
+                    user_id = str(user["username"])
+                    if block_user(user_id, user, "Hate Speech"):
+                        num_users_blocked = num_users_blocked + 1
+                    continue
     return num_users_blocked
 def chat(selected_option=None, radio_score=None, url_params=None):
     '''
+    This function is used to initialise blocking users once the user has authenticated with Mastodon.
     :param selected_option:
     :param radio_score:
     :param url_params:
             # Display to user, set options
             history.append(
+                ["Model tuned to a '{}%' threshold and is using the {} dataset.".format(radio_score,
+                                                                                        block_type.capitalize()),
                  "{} Account blocking initialised".format(block_type.capitalize())])
             num_users_blocked = block_users(client, radio_score, selected_option)
             history.append(
 def update_target_website():
     '''
+    Updates the URL used to authenticate WatchTower with Mastodon.
     #TODO this function is full of old code and can be optimised.
     :return:
     '''
     chatbot.value = chat_history
     chatbot.update(value=chat_history)
+    mastodon_auth_button.value = auth_button_text.format(
         get_target_website())
+    mastodon_auth_button.update(
         value=auth_button_text.format(
             get_target_website()))
         }
     """
 def get_chatbot_text():
+    return [('Welcome to Watchtower.', 'Log in via Mastodon and configure your blocking options above.')]
 def get_target_website():
     '''
+    A wrapper function used for retrieving the URL a user will use to authenticate WatchTower with Mastodon.
     :return: auth url
     '''
     return get_auth_url(mastodon)
+# The Gradio HTML component used for the 'sign in with Mastodon' button
 # The main chunk of code that uses Gradio blocks to create the UI
 html_button = None
 with block:
     gr.HTML('''
 <meta name="viewport" content="width=device-width, initial-scale=1">
 <link rel="stylesheet" href="https://www.w3schools.com/w3css/4/w3.css">
     with gr.Group():
         with gr.Row().style(equal_height=True):
             with gr.Box():
+                # gr.Label(value="WatchTower", visible=True, interactive=False)
                 url_params = gr.JSON({}, visible=False, label="URL Params").style(
                 )
                 text_input = gr.Text(label="Input", visible=False).style()
                 text_output = gr.Text(label="Output", visible=False).style()
+                html_button = mastodon_auth_button = gr.HTML(
                     value=auth_button_text.format(
                         get_target_website())).style(
                 )
                 with gr.Row().style(equal_height=True):
+                    radio = gr.CheckboxGroup(value=["Violent", "Hate Speech"],
+                                             choices=["Violent", "Hate Speech", "Misinformation"],
+                                             interactive=False, label="Behaviour To Block").style()
                     slider = gr.Slider(value=30, interactive=True, label="Threshold Confidence Tolerance").style()
+                chatbot = gr.Chatbot(label="Watchtower Output", value=[('Welcome to Watchtower.',
+                                                                        'Log in via Mastodon and configure your blocking options above.')]).style(
+                    color_map=["grey", "purple"])
                 btn = gr.Button("Run WatchTower").style(full_width=True).style()
     btn.click(fn=button_pressed, inputs=[slider, url_params],
    </p>"""
     )
+# Setup callback for when page loads (used to set a new Mastodon auth target webspage)
 block.__enter__()
 block.set_event_trigger(
     event_name="load", fn=update_target_website, inputs=None, outputs=[html_button], no_target=True
 block.attach_load_events()
 # Launcg the page
+block.launch(enable_queue=True)