diff --git a/sentiment.py b/sentiment.py index fd8f4dbcfbd89af8f0e39657a9927adf7643e53f..0f71ee22149b0496d23a61ba63bc6a59c8647068 100644 --- a/sentiment.py +++ b/sentiment.py @@ -5,27 +5,39 @@ import sys def main(): - ans = [] - sentiments = dict() - total = 0 - try: - with open('sentiment.txt', 'r') as f: - lines = f.read().splitlines() # split lines - user_search_value = raw_input('Please enter a word: ') - for line in f.readlines(): - line = line.strip().lower().split() - for words in line: - if words.find(user_search_value): - total += 1 - print(total) - for l in lines: - words = l.split() # split each string in line - ans.append(words) # appending to final ans - print(len(lines)) - var = 'The word ' + str(user_search_value) + ' appears ' + str(total) + ' times with an average sentiment of ' + str(sentiments) - print (var) - except IOError: - print("Error: can\'t find file or read data") + print ('Enter your choice: \n' + '1: Get the score of a word\n' + '2: Get the average score of words in a file (one word per line)\n' + '3: Find the highest/lowest scoring words in a file\n' + '4: Sort words from a file into positive.txt and negative.txt\n' + '5: Exit the program\n') + +choice = raw_input('Enter a number 1-5: ') + +if choice == 5: + print('Goodbye!') + + ans = [] + sentiments = dict() + total = 0 + try: + with open('sentiment.txt', 'r') as f: + lines = f.read().splitlines() # split lines + user_search_value = raw_input('Please enter a word: ') + for line in f.readlines(): + line = line.strip().lower().split() + for words in line: + if words.find(user_search_value): + total += 1 + print(total) + for l in lines: + words = l.split() # split each string in line + ans.append(words) # appending to final ans + print(len(lines)) + var = 'The word ' + str(user_search_value) + ' appears ' + str(total) + ' times with an average sentiment of ' + str(sentiments) + print (var) + except IOError: + print("Error: can\'t find file or read data") pass