diff --git a/.gitignore b/.gitignore index 5651d1b..7b6a305 100644 --- a/.gitignore +++ b/.gitignore @@ -4,3 +4,6 @@ __pycache__ data *.ipynb secret_key.txt +launch.json +env/ +twitter.txt \ No newline at end of file diff --git a/Makefile b/Makefile index 64d3be4..8fe64cd 100644 --- a/Makefile +++ b/Makefile @@ -3,6 +3,7 @@ up: python arxiv_daemon.py --num 2000 python compute.py + python twitter_daemon.py # I use this to run the server fun: diff --git a/aslite/db.py b/aslite/db.py index 55f1916..3c0e238 100644 --- a/aslite/db.py +++ b/aslite/db.py @@ -129,6 +129,10 @@ def get_email_db(flag='r', autocommit=True): edb = SqliteDict(DICT_DB_FILE, tablename='email', flag=flag, autocommit=autocommit) return edb +def get_tweets_db(flag='r', autocommit=True): + assert flag in ['r', 'c'] + tdb = SqliteDict(DICT_DB_FILE, tablename='tweets', flag=flag, autocommit=autocommit) + return tdb # ----------------------------------------------------------------------------- """ our "feature store" is currently just a pickle file, may want to consider hdf5 in the future diff --git a/aslite/twitter.py b/aslite/twitter.py new file mode 100644 index 0000000..84db7cc --- /dev/null +++ b/aslite/twitter.py @@ -0,0 +1,89 @@ +""" +Periodically checks Twitter for tweets about arxiv papers we recognize +and logs the tweets into mongodb database "arxiv", under "tweets" collection. +""" + +import os +import re +import time +import math +import pickle +import datetime +import tweepy +import logging + +# settings +# ----------------------------------------------------------------------------- +sleep_time = 60*10 # in seconds, between twitter API calls. Default rate limit is 180 per 15 minutes + +# convenience functions +# ----------------------------------------------------------------------------- + +def extract_arxiv_pids(r): + pids = [] + for u in r.get("entities",{}).get("urls",[]): + m = re.search('arxiv.org/(?:abs|pdf)/(.+)', u.get('unwound_url',u.get('expanded_url',''))) + if m: + pids.append(m.group(1)) + if m: + rawid = m.group(1).strip(".pdf") + pids.append(rawid) + return pids + +def get_latest_or_loop(q, start_datetime=None): + if start_datetime is None: + start_datetime = datetime.datetime.utcnow() - datetime.timedelta(days=6, hours=23) + start = start_datetime.strftime('%Y-%m-%dT%H:%M:%SZ') + results = [] + next_token = None + + q = "url:arxiv.org lang:en" + bearer = open('twitter.txt', 'r').read().splitlines()[0] + client = tweepy.Client(bearer) + + while True: + try: + resp = client.search_recent_tweets(q, expansions=['author_id'], + max_results=100, + next_token=next_token, + start_time=start, + tweet_fields=['id', 'created_at', 'author_id', 'entities', 'lang', 'public_metrics'], + user_fields=['public_metrics']) + #logging.log(logging.INFO, "fetched %d tweets", len(resp.data)) + results.append(resp) + next_token = resp.meta.get('next_token', None) + if next_token is None: + break + except Exception as e: + print('there was some problem (waiting some time and trying again):') + print(e) + time.sleep(sleep_time) + return results + +def parse_tweets(results): + tweets = [] + for result in results: + authors = result.includes.get('users',[]) + for r in result.data: + arxiv_pids = extract_arxiv_pids(r) + if not arxiv_pids: continue # nothing we know about here, lets move on + author = next(a for a in authors if a.id == r.author_id) + + # create the tweet. intentionally making it flat here without user nesting + tweet = {} + tweet['id'] = str(r.id) + tweet['pids'] = arxiv_pids # arxiv paper ids mentioned in this tweet + tweet['inserted_at_date'] = datetime.datetime.utcnow().isoformat() + tweet['created_at_date'] = r.created_at.isoformat() + tweet['created_at_time'] = int(time.mktime(r.created_at.timetuple())) + tweet['lang'] = r.lang + tweet['text'] = r.text + tweet['retweet_count'] = r.public_metrics.get('retweet_count',0) + tweet['reply_count'] = r.public_metrics.get('reply_count',0) + tweet['like_count'] = r.public_metrics.get('like_count',0) + tweet['quote_count'] = r.public_metrics.get('quote_count',0) + tweet['user_screen_name'] = author.username + tweet['user_followers_count'] = author.get('public_metrics',{}).get('followers_count',0) + tweet['user_following_count'] = author.get('public_metrics',{}).get('following_count',0) + tweets.append(tweet) + return tweets \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 304d2b4..b873204 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,3 +3,6 @@ Flask==2.0.2 numpy==1.21.4 scikit-learn==1.0.1 sqlitedict==1.7.0 +tweepy + + diff --git a/serve.py b/serve.py index 6aecb8e..3c7d5c8 100644 --- a/serve.py +++ b/serve.py @@ -11,6 +11,8 @@ import os import re import time from random import shuffle +import math +from functools import cache import numpy as np from sklearn import svm @@ -20,13 +22,14 @@ from flask import render_template from flask import g # global session-level object from flask import session -from aslite.db import get_papers_db, get_metas_db, get_tags_db, get_last_active_db, get_email_db +from aslite.db import get_papers_db, get_metas_db, get_tags_db, get_last_active_db, get_email_db, get_tweets_db from aslite.db import load_features # ----------------------------------------------------------------------------- # inits and globals RET_NUM = 25 # number of papers to return per page +max_tweet_records = 15 app = Flask(__name__) @@ -62,6 +65,11 @@ def get_metas(): g._mdb = get_metas_db() return g._mdb +def get_tweets(): + if not hasattr(g, '_tweets'): + g._tweets = get_tweets_db() + return g._tweets + @app.before_request def before_request(): g.user = session.get('user', None) @@ -169,6 +177,95 @@ def svm_rank(tags: str = '', pid: str = '', C: float = 0.01): return pids, scores, words +@cache +def tprepro(tweet_text): + # take tweet, return set of words + t = tweet_text.lower() + t = re.sub(r'[^\w\s]','',t) # remove punctuation + ws = set([w for w in t.split() if not w.startswith('#')]) + return ws + + +def score_tweet(tweet): + # give people with more followers more vote, as it's seen by more people and contributes to more hype + float_vote = min(math.log10(tweet['user_followers_count'] + 1), 4.0)/2.0 + + # uprank tweets that have more likes, retweets, replies, and quotes + float_vote += math.log10(tweet['like_count'] + tweet['retweet_count'] + 1) + float_vote += math.log10(tweet['reply_count'] + tweet['quote_count'] + 1) + return float_vote + +def weight_tweet(tweet): + papers = get_papers() + weight = 10.0 + # some tweets are really boring, like an rt + if "arxiv" in tweet['user_screen_name'].lower(): + weight -= 1 + + if (tweet['text'].lower().startswith('rt') or + tweet['lang'] != 'en' or + len(tweet['text']) < 40): + weight -= 1 + + # good tweets make a comment, not just a boring RT, or exactly the post title. Detect these. + tweet_words = len(tprepro(tweet['text'])) + title_words = 0 + for pid in tweet['pids']: + if pid not in papers: + continue + title_words += len(tprepro(papers[pid]['title'])) + comment_words = tweet_words - title_words # how much does the tweet have other than just the actual title of the article? + + if comment_words < 3: + weight -= 1 + + return weight + +@cache +def tweets_rank(days=7): + try: + days = int(days) + except: + days = 7 + + tweets = get_tweets() + papers = get_papers() + tnow = time.time() + t0 = tnow - int(days)*24*60*60 + tweets_filter = [t for p,t in tweets.items() if t['created_at_time'] > t0] + raw_votes, votes, records_dict = {}, {}, {} + for tweet in tweets_filter: + # filter out bots + if "arxiv" in tweet['user_screen_name'].lower(): + continue + + for pid in set(tweet['pids']): + if pid not in papers: + continue + if not pid in records_dict: + records_dict[pid] = {'pid':pid, 'tweets':[], 'vote': 0.0, 'raw_vote': 0} # create a new entry for this pid + + float_vote = score_tweet(tweet) + weight = float_vote + weight_tweet(tweet) + + # add up the votes for papers + records_dict[pid]['tweets'].append({'screen_name':tweet['user_screen_name'], 'text':tweet['text'], 'weight':weight, 'id':tweet['id'] }) + votes[pid] = votes.get(pid, 0.0) + float_vote + raw_votes[pid] = raw_votes.get(pid, 0) + 1 + + # record the total amount of vote/raw_vote for each pid + for pid in votes: + records_dict[pid]['vote'] = votes[pid] # record the total amount of vote across relevant tweets + records_dict[pid]['raw_vote'] = raw_votes[pid] + + + pids = sorted(records_dict, key=lambda x: records_dict[x]['vote'], reverse=True) + scores = [records_dict[pid]['vote'] for pid in pids] + tweets = [records_dict[pid]['tweets'] for pid in pids] + + return pids, scores, tweets + + def search_rank(q: str = ''): if not q: return [], [] # no query? no results @@ -210,13 +307,14 @@ def main(): default_skip_have = 'no' # override variables with any provided options via the interface - opt_rank = request.args.get('rank', default_rank) # rank type. search|tags|pid|time|random + opt_rank = request.args.get('rank', default_rank) # rank type. search|tags|pid|time|tweets|random opt_q = request.args.get('q', '') # search request in the text box opt_tags = request.args.get('tags', default_tags) # tags to rank by if opt_rank == 'tag' opt_pid = request.args.get('pid', '') # pid to find nearest neighbors to opt_time_filter = request.args.get('time_filter', default_time_filter) # number of days to filter by opt_skip_have = request.args.get('skip_have', default_skip_have) # hide papers we already have? opt_svm_c = request.args.get('svm_c', '') # svm C parameter + opt_tweet_filter = request.args.get('tweet_filter', '') # days of tweets to filter opt_page_number = request.args.get('page_number', '1') # page number for pagination # if a query is given, override rank to be of type "search" @@ -232,6 +330,7 @@ def main(): # rank papers: by tags, by time, by random words = [] # only populated in the case of svm rank + tweets = [] # only populated in the case of tweet rank if opt_rank == 'search': pids, scores = search_rank(q=opt_q) elif opt_rank == 'tags': @@ -240,6 +339,8 @@ def main(): pids, scores, words = svm_rank(pid=opt_pid, C=C) elif opt_rank == 'time': pids, scores = time_rank() + elif opt_rank == 'tweets': + pids, scores, tweets = tweets_rank(days=opt_tweet_filter) elif opt_rank == 'random': pids, scores = random_rank() else: @@ -287,12 +388,15 @@ def main(): context['papers'] = papers context['tags'] = rtags context['words'] = words + context['tweets'] = tweets context['words_desc'] = "Here are the top 40 most positive and bottom 20 most negative weights of the SVM. If they don't look great then try tuning the regularization strength hyperparameter of the SVM, svm_c, above. Lower C is higher regularization." + context['words_desc'] = "Here are the top 15 most influential tweets about this paper." context['gvars'] = {} context['gvars']['rank'] = opt_rank context['gvars']['tags'] = opt_tags context['gvars']['pid'] = opt_pid context['gvars']['time_filter'] = opt_time_filter + context['gvars']['tweet_filter'] = opt_tweet_filter context['gvars']['skip_have'] = opt_skip_have context['gvars']['search_query'] = opt_q context['gvars']['svm_c'] = str(C) @@ -324,12 +428,26 @@ def inspect(): }) words.sort(key=lambda w: w['weight'], reverse=True) + # get the tweets for this paper + tdb = get_tweets() + tweets = [t for _, t in tdb.items() if pid in t['pids']] + for i, t in enumerate(tweets): + tweets[i]['votes'] = score_tweet(t) + tweets[i]['weight'] = weight_tweet(t) + + # crop the tweets to only some number of highest weight ones (for efficiency) + tweets.sort(reverse=True, key=lambda x: x['weight']) + if len(tweets) > max_tweet_records: + tweets = tweets[:max_tweet_records] + # package everything up and render paper = render_pid(pid) context = default_context() context['paper'] = paper context['words'] = words + context['tweets'] = tweets context['words_desc'] = "The following are the tokens and their (tfidf) weight in the paper vector. This is the actual summary that feeds into the SVM to power recommendations, so hopefully it is good and representative!" + context['tweets_desc'] = "The following are the most influential tweets and their scores." return render_template('inspect.html', **context) @app.route('/profile') @@ -492,3 +610,6 @@ def register_email(): edb[g.user] = email return redirect(url_for('profile')) + +if __name__ == '__main__': + app.run(debug=True) \ No newline at end of file diff --git a/static/tweet_list.js b/static/tweet_list.js new file mode 100644 index 0000000..5958003 --- /dev/null +++ b/static/tweet_list.js @@ -0,0 +1,31 @@ +'use strict'; + +const Tweet = props => { + const p = props.tweet; + // tweet, score + return ( + twttr.widgets.createTweet( + String(p.id), + document.getElementById('tweetwrap') + ) + ) +} + +const TweetList = props => { + const lst = props.tweets; + const tweets_desc = props.tweets_desc; + const tlst = lst.map((jtweet, ix) => ); + return ( +
+
{tweets_desc}
+
+ {tlst} +
+
+ ) +} + +var elt = document.getElementById('tweetwrap'); +if(elt) { + ReactDOM.render(, elt); +} diff --git a/templates/index.html b/templates/index.html index 0c814b7..2617b02 100644 --- a/templates/index.html +++ b/templates/index.html @@ -5,6 +5,7 @@ var papers = {{ papers | tojson }}; var tags = {{ tags | tojson }}; var words = {{ words | tojson }}; +var tweets = {{ tweets | tojson }}; var words_desc = {{ words_desc | tojson }}; var gvars = {{ gvars | tojson }}; @@ -43,6 +44,7 @@ var move_page = function(int_offset) { + @@ -58,6 +60,10 @@ var move_page = function(int_offset) { + + + + diff --git a/templates/inspect.html b/templates/inspect.html index 79b35bf..842e4e2 100644 --- a/templates/inspect.html +++ b/templates/inspect.html @@ -5,17 +5,38 @@ var paper = {{ paper | tojson }}; var words = {{ words | tojson }}; var words_desc = {{ words_desc | tojson }}; +var tweets = {{ tweets | tojson }}; +var tweets_desc = {{ tweets_desc | tojson }}; + + {% endblock %} {% block content %}
-
+
+
{% endblock %} {% block elements %} + {% endblock %} diff --git a/twitter_daemon.py b/twitter_daemon.py new file mode 100644 index 0000000..81de8c1 --- /dev/null +++ b/twitter_daemon.py @@ -0,0 +1,42 @@ +""" +This script is intended to wake up every 30 min or so (eg via cron), +it checks for any new arxiv papers via the arxiv API and stashes +them into a sqlite database. +""" + +import sys +import time +import random +import datetime +import logging +import argparse + +from aslite.twitter import get_latest_or_loop, parse_tweets +from aslite.db import get_papers_db, get_tweets_db + +if __name__ == '__main__': + + logging.basicConfig(level=logging.INFO, format='%(name)s %(levelname)s %(asctime)s %(message)s', datefmt='%m/%d/%Y %I:%M:%S %p') + + pdb = get_papers_db(flag='c') + tdb = get_tweets_db(flag='c') + prevn = len(tdb) + + def store(t): + tdb[t['id']] = t + + latest = 0 + for k,v in tdb.items(): + if v['created_at_time'] > latest: + latest = v['created_at_time'] + + if prevn > 0: + start = datetime.datetime.utcfromtimestamp(latest) + else: + start = None + + # fetch the latest tweets mentioning arxiv.org + results = get_latest_or_loop(start) + tweets = parse_tweets(results) + for t in tweets: + if t['id'] not in tdb: store(t)