main.py

from pattern.web import Twitter, Google
from vaderSentiment.vaderSentiment import SentimentIntensityAnalyzer
import re
import pandas as pd
import discord

##text = input("Enter Name of NFT Project To Analyze: ")

analyzer = SentimentIntensityAnalyzer()

new_words = {
    'wgmi': 4,
    'ngmi': -4,
    '🔥' : 4,
    'hacked' : -1,
    'rug': -2,
    'was rugged': -4,
    'wagmi': 4
}
analyzer.lexicon.update(new_words)

def get_sentiment(text):
    tweets = []
    twitter = Twitter(language='en')
    for tweets_list in twitter.search(  text + '-"#'+text+'"', cached=False, count=10000):
        tweets.append(tweets_list.text)

    tweets =  [re.sub(r'@[A-Za-z0-9]+','', str(x)) for x in tweets]
    tweets =  [re.sub(r'^https?:\/\/.*[\r\n]*','', str(x)) for x in tweets]

    tweet_score = analyzer.polarity_scores((tweets))
    tw_df = pd.DataFrame(tweet_score, index=[0])

    goog = []
    engine = Google()
    for result in engine.search(text + "NFT", cached=False, count=25):
        test = result.text.replace('<b>', '')
        test = test.replace('</b>', '')
        goog.append(test)

    goog_score = analyzer.polarity_scores((goog))
    goog_df = pd.DataFrame(goog_score, index=[0])

    tweet_mean = (tw_df['compound'].mean())
    goog_mean = (goog_df['compound'].mean())

    tweet_mean = (tweet_mean * 0.85) + (goog_mean * 0.15)

    if(tweet_mean == 0):
        return "Not Valid Search"

    if(tweet_mean > 0.9):
        return "🔥 Hot 🔥\n Score: " + str(tweet_mean)
    elif(tweet_mean < 0.9 and tweet_mean > 0.7):
        return "Mid \n Score: " + str(tweet_mean)
    return "🗑️ Trash 🗑️\n Score: " + str(tweet_mean)

client = discord.Client()

@client.event
async def on_ready():
    print('We have logged in as {0.user}'.format(client))

@client.event
async def on_message(message):
    if message.author == client.user:
        return

    if message.content.startswith('!search'):
        text = message.content[message.content.index(' ')+1:]
        await message.channel.send(get_sentiment(text))
    
    if message.content.startswith('!help'):
        await message.channel.send("> Scores go `-1.0` - `1.0` (With 1 being most positive and -1 being the most negative) \n > `Command`: `!search <NFT Project Name>`")

client.run('')