twitter-pics/main.py at main · CodeDayBayArea/twitter-pics · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
import tweepy
from dotenv import load_dotenv
import os
import smtplib
from email.message import EmailMessage
import cv2
import time
from PIL import Image, ImageTk
import tkinter as tk
import tkinter.font as font
from math import floor
import imghdr
load_dotenv()

auth = tweepy.OAuthHandler(os.environ.get(
    "CONSUMER_KEY"), os.environ.get("CONSUMER_SECRET"))
auth.set_access_token(os.environ.get("API_TOKEN"),
                      os.environ.get("API_SECRET"))

api = tweepy.API(auth)

w1 = tk.Tk()
w1.title("Sketchy")
w1.geometry("2560x1600")
label = tk.Label(w1)
label.grid(row=0, column=1, sticky='nesw')
cap = cv2.VideoCapture(0)

scale_factor = 0.58

def getImage():
    ret, frame = cap.read()

    image = frame
    grey_img = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    blur = cv2.GaussianBlur(grey_img, (21, 21), 0)
    sketch = cv2.divide(grey_img, blur, scale=256.0)
    # sketch = cv2.resize(sketch, (w1.winfo_width(), w1.winfo_height()-0), interpolation = cv2.INTER_AREA)
    cv2.imwrite("sketch.png", sketch)

    background = Image.open("sketch.png").convert("RGBA")
    width, height = background.size
    foreground = Image.open("stamp.png").convert("RGBA")
    foreground = foreground.resize(
        (width, int(150*(height/width))))
    background.paste(foreground, (0, 0), mask=foreground)
    newBackground = background.resize((floor(background.size[0]*scale_factor), floor(background.size[1]*scale_factor)))
    newBackground.save("main.png")

    return ImageTk.PhotoImage((background))


def updateLabel():
    LE_IMAGE = getImage()
    label.imgtk = LE_IMAGE
    label.configure(image=LE_IMAGE)
   # Repeat after an interval to capture continiously
    label.after(5, updateLabel)


def postTweet():
    media = api.media_upload('./main.png')
    api.update_status("A person has appeared at #toorcamp", media_ids=[media.media_id_string])


def printPhoto():
    os.startfile("main.png", "print")

def showWindow():

    for i in range(4):
        w1.columnconfigure(i, weight=1)
    w1.rowconfigure(1, weight=1)
    myFont = font.Font(size=30)
    bird = tk.Button(w1, height=50,
                     text="Post to Twitter", command=postTweet)
    bird.grid(row=2, column=2, sticky='nesw')
    bird["font"] = myFont
    bird.pack()

    retake = tk.Button(w1, height=50, text="Retake")
    retake.grid(row=2, column=1, sticky='nesw')
    retake["font"] = myFont
    retake.pack()

    printer = tk.Button(w1, text="Print photo", command=printPhoto)
    printer.grid(row=2, column=1, sticky='nesw')

    updateLabel()

    w1.mainloop()


if __name__ == "__main__":
    showWindow()