Create app.py

app.py

@@ -0,0 +1,204 @@
+import joblib
+import os
+import os.path
+from os import path
+import discord
+import pandas as pd
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from transformers import pipeline
+from discord.ext import commands
+from discord.ext import commands
+
+
+
+positive = ['admiration', 'surprise','curiosity','desire','amusement', 'approval', 'caring', 'excitement','gratitude','joy', 'love', 'optimism', 'pride', 'relief']
+negative = ['anger', 'annoyance', 'confusion','disappointment', 'disapproval', 'disgust', 'embarrassment', 'fear',  'grief', 'nervousness',  'realization', 'remorse', 'sadness']
+emotions=["admiration", "amusement", "anger", "annoyance", "approval", "caring", "confusion", "curiosity", "desire", "disappointment", "disapproval", "disgust", "embarrassment", "excitement", "fear", "gratitude", "grief", "joy", "love", "nervousness", "optimism", "pride", "realization", "relief", "remorse", "sadness", "surprise"]  
+admiration_list=[] 
+admiration_idx=0
+admiration_val=0
+amusement_list=[] 
+amusement_idx=0
+amusement_val=0
+anger_list=[] 
+anger_idx=0
+anger_val=0
+annoyance_list=[]
+annoyance_idx=0
+annoyance_val=0
+approval_list=[]
+approval_idx=0
+approval_val=0
+caring_list=[] 
+caring_idx=0
+caring_val=0
+confusion_list=[] 
+confusion_idx=0
+confusion_val=0
+curiosity_list=[] 
+curiosity_idx=0
+curiosity_val=0
+desire_list=[] 
+desire_idx=0
+desire_val=0
+disappointment_list=[] 
+disappointment_idx=0
+disappointment_val=0
+disapproval_list=[] 
+disapproval_idx=0
+disapproval_val=0
+disgust_list=[]
+disgust_idx=0
+disgust_val=0
+embarrassment_list=[]
+embarrassment_idx=0
+embrassment_val=0
+excitement_list=[] 
+excitement_idx=0
+excitement_val=0
+fear_list=[] 
+fear_idx=0
+fear_val=0
+gratitude_list=[] 
+gratitude_idx=0
+gratitude_val=0
+grief_list=[] 
+grief_idx=0
+grief_val=0
+joy_list=[] 
+joy_idx=0
+joy_val=0
+love_list=[]
+love_idx=0
+love_val=0
+nervousness_list=[]
+nervousness_idx=0
+nervousness_val=0 
+optimism_list=[] 
+optimism_idx=0
+optimism_val=0
+pride_list=[] 
+pride_idx=0
+pride_val=0
+realization_list=[] 
+realization_idx=0
+realization_val=0
+relief_list=[] 
+relief_idx=0
+relief_val=0
+remorse_list=[] 
+remorse_idx=0
+remorse_val=0
+sadness_list=[] 
+sadness_idx=0
+sadness_val=0
+surprise_list=[]
+surprise_idx=0
+surprise_val=0
+neutral_list=[]
+neutral_idx=0
+neutral_val=0
+basepath="/content/drive/MyDrive/raahee_activity_recommendations/"
+t=0
+for emotion in emotions:
+  filepath=basepath+emotion
+  responsepath=filepath+"_responses.xlsx"
+  if(os.path.exists(responsepath)):
+     df=pd.read_excel(responsepath)
+     emotion+="_list"
+     for i in df['Unnamed: 0']:
+        globals()[emotion].append(i)
+
+  else:
+    gifpath=filepath+"_gifs.xlsx"
+    if(os.path.exists(gifpath)):
+       df=pd.read_excel(gifpath) 
+       emotion+="_list"
+       for i in df['Unnamed: 0']:
+         globals()[emotion].append(i)  
+
+    else:
+       moviepath=filepath+"_movies.xlsx"
+       if(os.path.exists(moviepath)):
+         df=pd.read_excel(moviepath)
+         emotion+="_list"
+         val=emotion
+         val+="_val"
+         globals()[val]=1
+         t=1
+         for i in df['Unnamed: 0']:
+           globals()[emotion].append(i) 
+       else:
+         bookpath=filepath+"_books.xlsx"
+         t=2
+         val=emotion
+         val+="_val"
+         globals()[val]=2
+         if(os.path.exists(bookpath)):
+           df=pd.read_excel(bookpath)
+           emotion+="_list"
+           for i in df['Unnamed: 0']:
+             globals()[emotion].append(i) 
+
+
+client = discord.Client()
+bot = commands.Bot(command_prefix='!')
+
+
+
+#classifier = pipeline("text-classification",model='bhadresh-savani/bert-base-go-emotion')
+
+   
+def getEmotion(message):
+  return classifier(message)[0]["label"]    
+
+@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('$hello'):
+        await message.channel.send('Hello!')
+    elif message.content.startswith('!pm'):
+        await message.channel.send('')
+         
+    else:
+       print(message.author)
+       t=0
+       msg = message.content
+      
+       print(msg)
+       response = getEmotion(msg)
+       print(response)
+       ans=""
+       greeting=""
+       val=response+"_list"
+       idx=response+"_idx"
+       temp=response+"_val"
+       print(idx)
+       print(temp)
+       print(globals()[temp])
+       if response in positive:
+            greeting = "Great to hear that <@{0.author.id}> !!!\n".format(message)
+       elif response in negative:
+            greeting = "<@{0.author.id}> I can completely understand what you're going through\n".format(message)
+       if(globals()[temp]==1):
+          greeting+="This movie made my day, hope it turns out the same for you too !\n"
+       if(globals()[temp]==2):
+           greeting+="I really enjoyed reading this book, I guess you must read it too!\n"
+       t=0
+       if(globals()[idx]<len(globals()[val])):
+          ans=globals()[val][globals()[idx]]
+          globals()[idx]+=1
+       else:
+         globals()[idx]=0
+         if(len(globals()[val])!=0):
+          ans=globals()[val][0]
+         globals()[idx]+=1 
+
+       await message.channel.send(greeting+ans)    
+
+client.run(os.getenv("TOKEN"))