from tkinter import *
from tkinter import messagebox
import sqlite3
conn = sqlite3.connect('sondagem1.db')
c = conn.cursor()

c.execute('''
CREATE TABLE IF NOT EXISTS votos (
    id INTEGER PRIMARY KEY,
    partido TEXT NOT NULL
)
''')
conn.commit()

partido = {
    'PS': 0,
    'PSD': 0,
    'Chega': 0,
    'IL': 0,
    'BE': 0,
    'PCP': 0,
    'Livre': 0,
    'PAN': 0
}
def votar(partido_escolhido):
    c.execute('INSERT INTO votos (partido) VALUES (?)', (partido_escolhido,))
    conn.commit()
    messagebox.showinfo("Voto Registrado", f"Você votou no {partido_escolhido}!")
    atualizar_resultados()

def atualizar_resultados():
    for escolha in partido:
        partido[escolha] = 0
    c.execute('SELECT partido, COUNT(*) FROM votos GROUP BY partido')
    rows = c.fetchall()
    for row in rows:
        partido[row[0]] = row[1]
    resultados_texto = "\n".join([f"{escolha}: {contagem} votos" for escolha, contagem in partido.items()])
    resultados_label.config(text=resultados_texto)
root = Tk()
root.title("Sondagem de Partidos")
root.geometry('400x550')
root.configure(background='#09A3BA')

titulo = Label(root, text="Qual é seu partido favorito?", font=("Helvetica", 16))
titulo.pack(pady=10)

for escolha in partido:
    btn = Button(root, text=escolha, width=20, command=lambda c=escolha: votar(c))
    btn.pack(pady=5)
resultados_label = Label(root, text="", font=("Helvetica", 14))
resultados_label.pack(pady=10)
atualizar_resultados()
root.mainloop()
conn.close()