Fixed

README.md

@@ -1,2 +1,30 @@
 # InteractiveProgramming
+
 This is the base repo for the interactive programming project for Software Design, Spring 2016 at Olin College.
+
+Here is what you will need to install:
+
+Twitter-mining:
+```
+$ sudo pip3 install python-twitter
+```
+Pygame:
+```
+$ sudo apt-get build-dep python-pygame
+$ sudo apt-get install python-dev
+$ sudo pip3 install pygame
+```
+Word-processing:
+```
+$ sudo pip install -U nltk
+```
+
+First, download all the image files.
+
+Then, download either the pickle files or the twitter_textmining.py file (and run the subsequent file for the pickle).
+
+Download phrase_extractor.py and main.py.
+
+Run main.py. Hover over the candidate's names in order to see lines drawn, depicting the sentiment of candidates' tweets towards other candidates.
+
+Link to write-up: [finished write-up](https://github.com/prava-d/InteractiveProgramming/blob/fixed/MP4Write-Up.pdf)

interactive_play.py

@@ -0,0 +1,38 @@
+import tkinter
+
+class App:
+    def __init__(self, root):
+        def do_work(self):
+            if self.mouse_pressed:
+                self.Hover1.bind("<Enter>", lambda event, h=self.Hover1: h.configure(bg="red"))
+                self.Hover1.bind("<Leave>", lambda event, h=self.Hover1: h.configure(bg="SystemButtonFace"))
+
+                self.Hover2.bind("<Enter>", lambda event, h=self.Hover2: h.configure(bg="yellow"))
+                self.Hover2.bind("<Leave>", lambda event, h=self.Hover2: h.configure(bg="SystemButtonFace"))
+            else:
+                self.Hover1.unbind("<Enter>")
+                self.Hover1.unbind("<Leave>")
+                self.Hover2.unbind("<Enter>")
+                self.Hover2.unbind("<Leave>")
+
+
+
+    def do_work(self):
+        if self.mouse_pressed:
+            self.Hover1.bind("<Enter>", lambda event, h=self.Hover1: h.configure(bg="red"))
+            self.Hover1.bind("<Leave>", lambda event, h=self.Hover1: h.configure(bg="SystemButtonFace"))
+
+            self.Hover2.bind("<Enter>", lambda event, h=self.Hover2: h.configure(bg="yellow"))
+            self.Hover2.bind("<Leave>", lambda event, h=self.Hover2: h.configure(bg="SystemButtonFace"))
+
+    def OnMouseDown(self, event):
+        self.mouse_pressed = True
+        self.do_work()
+
+    def OnMouseUp(self, event):
+        self.mouse_pressed = False
+        self.do_work()
+
+root=tkinter.Tk()
+app = App(root)
+root.mainloop()

main.py

@@ -0,0 +1,285 @@
+"""
+Display candidate names on a circle and show the candidate's
+sentiments when mouse is on the candidate's name
+"""
+
+import pygame
+import math
+import pickle
+import phrase_extractor as pe
+
+SCREEN_WIDTH = 1200
+SCREEN_HEIGHT = 800
+
+# Define some colors
+RED     = (255, 0 , 0)
+BLUE    = (0, 0, 255)
+POWDER_BLUE = (176, 224, 230)
+FIREBRICK = (178, 34, 34)
+
+# Define colors for sentiment
+NEGATIVE = (255, 0 , 0)    # red
+POSITIVE = (0, 225, 0)     # green
+NEUTRAL  = (125, 125, 125) # gray
+NO_INFO  = (0, 255, 255)   # cyan
+
+class Candidate:
+    def __init__(self, name, image_file, pickle_file):
+        self.name = name
+        self.picture = image_file
+        f = open(pickle_file, 'rb')
+        self.quotes = pickle.load(f)
+        f.close()
+        self.sentiment = []
+        self.screen_rect = None
+        self.screen_xy = None
+
+    def update_sentiment(self, candidate):
+        names = candidate.name.split()
+        names.append(candidate.name)
+
+        sentences = []
+        for quote in self.quotes:
+            sentences += pe.get_sentences(quote, names)
+
+        score = 0
+        mood = NO_INFO
+        for s in sentences:
+            quote = pe.get_quote(s)
+            score += quote.tone
+        if len(sentences) > 0:
+            if score > 0:
+                mood = POSITIVE
+            elif score < 0:
+                mood = NEGATIVE
+            else:
+                mood = NEUTRAL
+        self.sentiment.append((candidate, mood))
+
+
+def get_sentiments(candidates):
+    """
+    Update sentiments for all candidates
+    """
+    for candidate1 in candidates:
+        for candidate2 in candidates:
+            if candidate1 != candidate2:
+                candidate1.update_sentiment(candidate2)
+
+
+def show_sentiments(screen, candidate):
+    """
+    Show sentiment lines for the candidate
+    """
+    # Screen width and height
+    screen_width = screen.get_width()
+    screen_height = screen.get_height()
+
+    # Photo
+    photo_size = (int(screen_height / 7), int(screen_height / 6))
+    image = pygame.image.load(candidate.picture).convert()
+    image = pygame.transform.scale(image, photo_size)
+    screen.blit(image, [10, 10])
+
+    # lines
+    for (to_candidate, mood) in candidate.sentiment:
+        pygame.draw.line(screen, mood, candidate.screen_xy, to_candidate.screen_xy, 2)
+
+
+def show_candidates(screen, candidates):
+    """
+    Display candidates names and returns screen position details
+    """
+    # Screen width and height
+    screen_width = screen.get_width()
+    screen_height = screen.get_height()
+
+    # Display legend
+    font_size = int(screen_height * 0.02)
+    font = pygame.font.SysFont('Calibri', font_size, True, False)
+
+    x = int(screen_width * 0.85)
+    y = int(screen_height - 11 * font_size)
+    t = font.render('Legend:', True, BLUE)
+    screen.blit(t, [x, y])
+
+    x = int(screen_width * 0.9)
+    y = int(screen_height - 10 * font_size)
+    t = font.render('Positive', True, POSITIVE)
+    screen.blit(t, [x, y])
+
+    y = int(screen_height - 9 * font_size)
+    t = font.render('Negative', True, NEGATIVE)
+    screen.blit(t, [x, y])
+
+    y = int(screen_height - 8 * font_size)
+    t = font.render('Neutral', True, NEUTRAL)
+    screen.blit(t, [x, y])
+
+    y = int(screen_height - 7 * font_size)
+    t = font.render('No Info', True, NO_INFO)
+    screen.blit(t, [x, y])
+
+    # Draw a circle
+    center = (int(screen_width/2), int(screen_height/2))
+    radius = int(screen_height * 0.4)
+    width = int(screen_height * 0.02)
+    pygame.draw.circle(screen, FIREBRICK, center, radius, width)
+
+    # Select the font to use, size, bold, italics
+    font_size = int(screen_height * 0.03)
+    font = pygame.font.SysFont('Calibri', font_size, True, False)
+
+    a = 0
+    for candidate in candidates:
+        # Name length on screen in pixels (approx)
+        name_len = int(len(candidate.name) * font_size / 2)
+
+        angle = a * 2 * math.pi/len(candidates)
+
+        # Find screen position to write names around the circle
+        x = center[0] + int(radius * math.cos(angle))
+        y = center[1] + int(radius * math.sin(angle))
+
+        # Draw a small red circle
+        candidate.screen_xy = (x,y)
+        pygame.draw.circle(screen, POWDER_BLUE, candidate.screen_xy, 5)
+
+        # Calculate the position for writing name on the screen
+        if x < center[0]:
+            x -= name_len
+        elif x == center[0]:
+            x -= int(name_len / 2)
+
+        if y < center[1]:
+            y -= font_size
+        elif y == center[1]:
+            y -= int(font_size / 2)
+
+        # Draw the text
+        candidate.screen_rect = pygame.Rect(x, y, name_len, font_size)
+        t = font.render(candidate.name, True, POWDER_BLUE)
+        screen.blit(t, [x, y])
+        a += 1
+
+
+def show_popup(screen, candidate):
+    """
+    Display candidate's details
+    """
+    # Screen width and height
+    screen_width = screen.get_width()
+    screen_height = screen.get_height()
+
+    # popup window
+    w = screen_width / 1.9
+    x = (screen_width - w) / 2
+    h = screen_height / 1.8
+    y = (screen_height - h) / 2
+    pygame.draw.rect(screen, win_color, (x, y, w, h), 0)
+
+    # Photo
+    photo_size = (int(screen_height / 7), int(screen_height / 7))
+    image = pygame.image.load(candidate.picture).convert()
+    image = pygame.transform.scale(image, photo_size)
+    screen.blit(image, [int(x * 1.2), int(y * 1.1)])
+
+    # Name
+    nx = int(x * 1.7)
+    ny = int(y * 1.3)
+    font_size = int(screen_height * 0.03)
+    font = pygame.font.SysFont('Calibri', font_size, True, False)
+    t = font.render(candidate.name, True, RED)
+    screen.blit(t, [nx, ny])
+
+    # Details
+    x = int(x * 1.1)
+    y = int(y * 1.8)
+    font_size = int(screen_height * 0.035)
+    line_size = int(screen_height * 0.06)
+    font = pygame.font.SysFont('Calibri', font_size, False, False)
+    text = candidate.quotes[0]
+    while len(text) > 0:
+        if len(text) > line_size:
+            line = text[0:line_size]
+            text = text[line_size:]
+        else:
+            line = text
+            text = ''
+
+        t = font.render(line.strip(), True, text_color)
+        screen.blit(t, [x, y])
+        y += font_size
+
+
+
+def main():
+    """
+    This is our main program.
+    """
+
+    candidates = [
+        Candidate('Bernie Sanders', 'Bernie Sanders.jpg', 'bernietwitters.pickle'),
+        Candidate('Hillary Clinton', 'Hillary Clinton.jpg', 'clintontwitters.pickle'),
+        Candidate('Ted Cruz', 'Ted Cruz.jpg', 'cruztwitters.pickle'),
+        Candidate('Tim Kaine', 'Tim Kaine.jpg', 'kainetwitters.pickle'),
+        Candidate('John Kasich', 'John Kasich.jpg', 'kasichtwitters.pickle'),
+        Candidate('Martin O\'Malley', 'Martin O\'Malley.jpg', 'malleytwitters.pickle'),
+        Candidate('Mike Pence', 'Mike Pence.jpg', 'pencertwitter.pickle'),
+        Candidate('Donald Trump', 'Donald Trump.jpg', 'trumptwitters.pickle')
+        ]
+
+    get_sentiments(candidates)
+
+    pygame.init()
+
+    # Set the height and width of the screen
+    size = [SCREEN_WIDTH, SCREEN_HEIGHT]
+    screen = pygame.display.set_mode(size)
+    background = pygame.image.load('background.jpg').convert()
+    background = pygame.transform.scale(background, size)
+
+    pygame.display.set_caption("2016 Presidential Candidates")
+
+
+
+    # Loop until the user clicks the close button.
+    done = False
+
+    # Used to manage how fast the screen updates
+    clock = pygame.time.Clock()
+
+    screen.blit(background, background.get_rect())
+    show_candidates(screen, candidates)
+
+    while not done:
+
+        # --- Event Processing
+        for event in pygame.event.get():
+            if event.type == pygame.QUIT:
+                done = True
+            elif event.type == pygame.MOUSEMOTION:
+                on_candidate = False
+                p = pygame.mouse.get_pos()
+                for candidate in candidates:
+                    if candidate.screen_rect.collidepoint(p):
+                        #show_popup(screen, candidate)
+                        show_sentiments(screen, candidate)
+                        on_candidate = True
+                if not on_candidate:
+                    screen.blit(background, background.get_rect())
+                    show_candidates(screen, candidates)
+
+        # --- Wrap-up
+        # Limit to 10 frames per second
+        clock.tick(10)
+
+        # Go ahead and update the screen with what we've drawn.
+        pygame.display.flip()
+
+
+    # Close everything down
+    pygame.quit()
+
+if __name__ == "__main__":
+    main()