Spaces:

User1342
/

WatchTower

Runtime error

App Files Files Community

User1342 commited on Oct 31, 2022

Commit

06188f8

•

1 Parent(s): 8cd7a79

Update app.py

Browse files

Files changed (1) hide show

app.py +68 -38

app.py CHANGED Viewed

@@ -14,6 +14,8 @@ consumer_secret = os.getenv('CONSUMER_SECRET')
 my_access_token = os.getenv('ACCESS_TOKEN')
 my_access_secret = os.getenv('ACCESS_SECRET')
 bearer = os.getenv('BEARER')
 html_data = '''
 <!DOCTYPE html>
 <html>
@@ -133,7 +135,7 @@ def get_client_from_tokens(oauth_verifier, oauth_token):
     '''
     new_oauth1_user_handler = tweepy.OAuth1UserHandler(
         consumer_token, consumer_secret,
-        callback="https://hf.space/embed/User1342/WatchTower/"
     )
     new_oauth1_user_handler.request_token = {
         "oauth_token": oauth_token,
@@ -158,11 +160,36 @@ def get_client_from_tokens(oauth_verifier, oauth_token):
     return their_client
 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.
-    TODO: Datasets not implemented.
     :param client:
     :param threshold:
     :param dataset:
@@ -178,36 +205,30 @@ def block_users(client, threshold, dataset):
         for user in users:
             print("Reviewing user {}".format(user))
-            if user["threshold"] >= threshold:
-                user_id = str(user["username"])
-                finished = False
-                while not finished:
-                    try:
-                        print("preparing to block {}".format(user_id))
-                        client.block(target_user_id=user_id)
-                        print("User blocked")
-                    except tweepy.errors.TooManyRequests as e:
-                        print("time out error")
-                        print(e)
-                        finished = True
-                        # time.sleep(240)
-                        continue
-                    except tweepy.errors.BadRequest as e:
-                        print("bad request error")
-                        print(e)
-                        finished = True
                         continue
-                    except:
-                        time.sleep(240)
                         continue
-                    time.sleep(1)
-                    finished = True
-                    me = client.get_me()
-                    print("{} blocked {}".format(me.data["username"], user))
-                    num_users_blocked = num_users_blocked + 1
     return num_users_blocked
@@ -231,9 +252,20 @@ def chat(selected_option=None, radio_score=None, url_params=None):
     if radio_score != None and selected_option != None:
         if client != None:
             history.append(
-                ["Model tuned to a '{}%' threshold and is using the '{}' dataset.".format(radio_score, selected_option),
-                 "{} Account blocking initialised".format(str(selected_option).capitalize())])
             num_users_blocked = block_users(client, radio_score, selected_option)
             history.append(
                 ["Blocked {} user account(s).".format(num_users_blocked), "Thank you for using Watchtower."])
@@ -315,7 +347,7 @@ def get_target_website():
     '''
     oauth1_user_handler = tweepy.OAuth1UserHandler(
         consumer_token, consumer_secret,
-        callback="https://hf.space/embed/User1342/WatchTower/"
     )
     target_website = oauth1_user_handler.get_authorization_url(signin_with_twitter=True)
@@ -359,12 +391,10 @@ with block:
                         get_target_website())).style(
                 )
                 with gr.Row().style(mobile_collapse=True, equal_height=True):
-                    radio = gr.CheckboxGroup(value="Violent", choices=["Violent", "Hate Speech", "Misinformation"],
-                                                 interactive=False, label="Behaviour To Block").style(
-                    rounded=(False, True, True, False),
-                )
-                    slider = gr.Slider(value=5, interactive=True, label="Threshold Confidence Tolerance")
                 chatbot = gr.Chatbot(value=chat_history, label="Watchtower Output").style(
                     rounded=(False, True, True, False),

 my_access_token = os.getenv('ACCESS_TOKEN')
 my_access_secret = os.getenv('ACCESS_SECRET')
 bearer = os.getenv('BEARER')
 html_data = '''
 <!DOCTYPE html>
 <html>
     '''
     new_oauth1_user_handler = tweepy.OAuth1UserHandler(
         consumer_token, consumer_secret,
+        callback="http://127.0.0.1:7860/'#https://hf.space/embed/User1342/WatchTower/"
     )
     new_oauth1_user_handler.request_token = {
         "oauth_token": oauth_token,
     return their_client
+def block_user(user_id, user, reason):
+    finished = False
+    while not finished:
+        try:
+            print("preparing to block {}".format(user_id))
+            client.block(target_user_id=user_id)
+            print("User blocked")
+        except tweepy.errors.TooManyRequests as e:
+            print("time out error")
+            print(e)
+            finished = True
+            # time.sleep(240)
+            continue
+        except tweepy.errors.BadRequest as e:
+            print("bad request error")
+            print(e)
+            finished = True
+            continue
+        except:
+            time.sleep(240)
+            continue
+        time.sleep(1)
+        finished = True
+        me = client.get_me()
+        print("{} blocked {}, for {}".format(me.data["username"], user, reason))
 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.
     :param client:
     :param threshold:
     :param 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"])
+                        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"])
+                        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"])
+                        block_user(user_id, user, "Hate Speech")
+                        num_users_blocked = num_users_blocked + 1
                         continue
     return num_users_blocked
     if radio_score != None and selected_option != None:
         if client != None:
+            # Extract the list to a string representation
+            if type(selected_option) is list:
+                block_type = ""
+                for b_type in selected_option:
+                    block_type = block_type + " + " + b_type.capitalize()
+                block_type = "'" + block_type[3:] + "'"
+            else:
+                block_type = selected_option
+            # 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(
                 ["Blocked {} user account(s).".format(num_users_blocked), "Thank you for using Watchtower."])
     '''
     oauth1_user_handler = tweepy.OAuth1UserHandler(
         consumer_token, consumer_secret,
+        callback="http://127.0.0.1:7860/"#"https://hf.space/embed/User1342/WatchTower/"
     )
     target_website = oauth1_user_handler.get_authorization_url(signin_with_twitter=True)
                         get_target_website())).style(
                 )
                 with gr.Row().style(mobile_collapse=True, equal_height=True):
+                    radio = gr.CheckboxGroup(value=["Violent", "Hate Speech"], choices=["Violent", "Hate Speech", "Misinformation"],
+                                                 interactive=False, label="Behaviour To Block").style(rounded=(False, True, True, False))
+                    slider = gr.Slider(value=10, interactive=True, label="Threshold Confidence Tolerance")
                 chatbot = gr.Chatbot(value=chat_history, label="Watchtower Output").style(
                     rounded=(False, True, True, False),