import tkinter as tk
from pc_twitter import reptile
from utils.createBrowserDriver import create

def button_click():
    user_input = entry.get()
    cookies = browser.get_cookies()
    session_id = browser.session_id
    if len(user_input) > 0:
        label.configure(text="Button Clicked!")
        reptile(browser)

browser = create("https://twitter.com/home?query=%E6%B0%B4")
window = tk.Tk()

window.geometry("400x300")

# 创建输入框
entry = tk.Entry(window)
entry.pack()

label = tk.Label(window, text="Hello, Tkinter!")
label.pack()

button = tk.Button(window, text="开始爬取", command=button_click)
button.pack()

window.mainloop()