Sentiment Value Agent

app/models/agents/sentiment_value_agent.rb

@@ -0,0 +1,83 @@
+require 'csv'
+
+module Agents
+	class SentimentValueAgent < Agent
+		description <<-MD
+			The SentimentValueAgent generates `good-bad` (psychological valence or happiness index),`active-passive` (arousal), and `strong-weak` (dominance) score. It will output a value between 1 and 9. Make sure the content this agent is running on have sufficient length.
+			Add more stuff
+		MD
+
+		event_description <<-MD
+			Events look like:
+			{
+				:valence   => 4.5
+				:arousal   => 4.5
+				:dominance => 4.5
+			}
+		MD
+
+		default_schedule "every_1h"
+
+		def default_options
+			{
+
+			}
+		end
+
+		def working?
+			true
+		end
+
+		def receive(incoming_events)
+            incoming_events.each do |event|
+                self.memory[:queue] ||= []
+                self.memory[:queue] << event.payload
+            end
+        end
+
+        def validate_options
+        end
+
+        def sentiment_hash
+        	anew = {}
+        	CSV.foreach Rails.root.join('app/assets/anew.csv') do |row|
+        		anew[row[0]] = [row[2],row[4],row[6]].map {|val| val.to_f}
+        	end
+        	anew
+        end
+
+        def sentiment_values(anew,text)
+        	valence, arousal, dominance, freq = [0] * 4
+        	text.downcase.strip.gsub(/[^a-z ]/,"").split.each do |word|
+        		if anew.has_key?(word)
+        			valence   += anew[word][0]
+        			arousal   += anew[word][1]
+        			dominance += anew[word][2]
+        			freq      += 1
+        		end
+        	end
+        	if valence != 0
+        		[valence/freq, arousal/freq, dominance/freq]
+        	else
+        		["Insufficient data for meaningful answer"] * 3
+        	end
+
+
+        end
+
+        def check
+        	if self.memory[:queue] && self.memory[:queue].length > 0
+        		anew = sentiment_hash
+        		self.memory[:queue].each do |text|
+        			if text[:message]
+        				sent_values = sentiment_values anew, text[:message]
+        				create_event :payload => {:content => text[:message], :valence => sent_values[0], :arousal => sent_values[1], :dominance => sent_values[2]}
+        			end
+        		end
+        		self.memory[:queue] = []
+        	end
+        end
+
+
+	end
+end