|
|
# app.py
|
|
|
import streamlit as st
|
|
|
import pandas as pd
|
|
|
from datetime import datetime
|
|
|
from database import db
|
|
|
from summarizer import summarizer
|
|
|
from config import config
|
|
|
import json
|
|
|
|
|
|
# Page config
|
|
|
st.set_page_config(
|
|
|
page_title="Nederlandse Politieke Kompas", page_icon="🇳🇱", layout="wide"
|
|
|
)
|
|
|
|
|
|
|
|
|
def main():
|
|
|
st.title("🇳🇱 Nederlandse Politieke Kompas")
|
|
|
st.markdown(
|
|
|
"Ontdek welke politieke partij het beste bij jouw idealen past door te stemmen op echte Tweede Kamer moties."
|
|
|
)
|
|
|
|
|
|
# Initialize session state
|
|
|
if "session_id" not in st.session_state:
|
|
|
st.session_state.session_id = None
|
|
|
if "current_motion_index" not in st.session_state:
|
|
|
st.session_state.current_motion_index = 0
|
|
|
if "motions" not in st.session_state:
|
|
|
st.session_state.motions = []
|
|
|
if "show_results" not in st.session_state:
|
|
|
st.session_state.show_results = False
|
|
|
|
|
|
# Sidebar configuration
|
|
|
with st.sidebar:
|
|
|
st.header("Instellingen")
|
|
|
|
|
|
motion_count = st.slider(
|
|
|
"Aantal moties",
|
|
|
min_value=5,
|
|
|
max_value=25,
|
|
|
value=config.DEFAULT_MOTION_COUNT,
|
|
|
)
|
|
|
|
|
|
policy_area = st.selectbox("Beleidsgebied", config.POLICY_AREAS)
|
|
|
|
|
|
margin_range = st.slider(
|
|
|
"Controversiële moties (%)",
|
|
|
min_value=0,
|
|
|
max_value=100,
|
|
|
value=(
|
|
|
config.DEFAULT_WINNING_MARGIN_MIN,
|
|
|
config.DEFAULT_WINNING_MARGIN_MAX,
|
|
|
),
|
|
|
)
|
|
|
|
|
|
if st.button("Start Nieuwe Sessie"):
|
|
|
start_new_session(motion_count, policy_area, margin_range)
|
|
|
|
|
|
if st.button("Genereer AI Samenvattingen"):
|
|
|
with st.spinner("Genereren van samenvattingen..."):
|
|
|
summarizer.update_motion_summaries()
|
|
|
st.success("Samenvattingen bijgewerkt!")
|
|
|
|
|
|
# Main content
|
|
|
if not st.session_state.session_id:
|
|
|
show_welcome_screen(motion_count, policy_area, margin_range)
|
|
|
elif st.session_state.show_results:
|
|
|
show_results()
|
|
|
else:
|
|
|
show_motion_interface()
|
|
|
|
|
|
|
|
|
def start_new_session(motion_count, policy_area, margin_range):
|
|
|
"""Start a new voting session"""
|
|
|
# Get filtered motions
|
|
|
motions = db.get_filtered_motions(
|
|
|
policy_area=policy_area,
|
|
|
min_margin=margin_range[0] / 100,
|
|
|
max_margin=margin_range[1] / 100,
|
|
|
limit=motion_count,
|
|
|
)
|
|
|
|
|
|
if len(motions) < motion_count:
|
|
|
st.warning(
|
|
|
f"Slechts {len(motions)} moties gevonden met de geselecteerde criteria."
|
|
|
)
|
|
|
|
|
|
# Create session
|
|
|
session_id = db.create_session(motion_count)
|
|
|
|
|
|
# Update session state
|
|
|
st.session_state.session_id = session_id
|
|
|
st.session_state.motions = motions[:motion_count]
|
|
|
st.session_state.current_motion_index = 0
|
|
|
st.session_state.show_results = False
|
|
|
|
|
|
st.rerun()
|
|
|
|
|
|
|
|
|
def show_welcome_screen(motion_count, policy_area, margin_range):
|
|
|
"""Show welcome screen with start button"""
|
|
|
col1, col2, col3 = st.columns([1, 2, 1])
|
|
|
|
|
|
with col2:
|
|
|
st.markdown("### Welkom bij de Nederlandse Politieke Kompas!")
|
|
|
|
|
|
st.markdown(f"""
|
|
|
**Jouw instellingen:**
|
|
|
- 📊 **{motion_count} moties** uit het beleidsgebied **{policy_area}**
|
|
|
- 🎯 **Controversiële moties** tussen {margin_range[0]}% en {margin_range[1]}% marge
|
|
|
|
|
|
Klik op "Start Nieuwe Sessie" in de zijbalk om te beginnen met stemmen.
|
|
|
""")
|
|
|
|
|
|
st.info(
|
|
|
"💡 **Tip**: Kies 'Alle' als beleidsgebied voor een breed overzicht van verschillende onderwerpen."
|
|
|
)
|
|
|
|
|
|
|
|
|
def show_motion_interface():
|
|
|
"""Show motion voting interface"""
|
|
|
if not st.session_state.motions:
|
|
|
st.error("Geen moties gevonden. Start een nieuwe sessie.")
|
|
|
return
|
|
|
|
|
|
current_index = st.session_state.current_motion_index
|
|
|
total_motions = len(st.session_state.motions)
|
|
|
|
|
|
# Progress bar
|
|
|
progress = (current_index) / total_motions
|
|
|
st.progress(progress, text=f"Motie {current_index + 1} van {total_motions}")
|
|
|
|
|
|
if current_index >= total_motions:
|
|
|
st.session_state.show_results = True
|
|
|
st.rerun()
|
|
|
return
|
|
|
|
|
|
motion = st.session_state.motions[current_index]
|
|
|
|
|
|
# Motion display
|
|
|
st.header(f"Motie {current_index + 1}: {motion['title']}")
|
|
|
|
|
|
# Policy area tag
|
|
|
st.markdown(f"**Beleidsgebied:** {motion['policy_area']}")
|
|
|
|
|
|
# Layman explanation (prominent)
|
|
|
if motion.get("layman_explanation"):
|
|
|
st.markdown("### 📝 Uitleg in begrijpelijke taal:")
|
|
|
st.markdown(f"*{motion['layman_explanation']}*")
|
|
|
|
|
|
# Original description (collapsible)
|
|
|
motion_text = motion.get("body_text") or motion.get("description", "")
|
|
|
if motion_text:
|
|
|
label = (
|
|
|
"📋 Volledige motietekst"
|
|
|
if motion.get("body_text")
|
|
|
else "📋 Originele motiebeschrijving"
|
|
|
)
|
|
|
with st.expander(label):
|
|
|
st.write(motion_text)
|
|
|
|
|
|
# Voting buttons
|
|
|
st.markdown("### 🗳️ Hoe zou jij stemmen?")
|
|
|
|
|
|
col1, col2, col3 = st.columns(3)
|
|
|
|
|
|
with col1:
|
|
|
if st.button("✅ Voor", use_container_width=True, type="primary"):
|
|
|
cast_vote("Voor")
|
|
|
|
|
|
with col2:
|
|
|
if st.button("❌ Tegen", use_container_width=True):
|
|
|
cast_vote("Tegen")
|
|
|
|
|
|
with col3:
|
|
|
if st.button("🚫 Geen stem", use_container_width=True):
|
|
|
cast_vote("Geen stem")
|
|
|
|
|
|
|
|
|
def cast_vote(vote_choice):
|
|
|
"""Record user vote and move to next motion"""
|
|
|
current_motion = st.session_state.motions[st.session_state.current_motion_index]
|
|
|
|
|
|
# Save vote to database
|
|
|
db.update_user_vote(st.session_state.session_id, current_motion["id"], vote_choice)
|
|
|
|
|
|
# Move to next motion
|
|
|
st.session_state.current_motion_index += 1
|
|
|
st.rerun()
|
|
|
|
|
|
|
|
|
def show_results():
|
|
|
"""Show voting results and party matches"""
|
|
|
st.header("🎯 Jouw Resultaten")
|
|
|
|
|
|
# Calculate party matches
|
|
|
party_matches = db.calculate_party_matches(st.session_state.session_id)
|
|
|
|
|
|
if not party_matches:
|
|
|
st.error("Geen resultaten beschikbaar.")
|
|
|
return
|
|
|
|
|
|
# Party ranking table
|
|
|
st.subheader("📊 Partij Overeenkomsten (van hoog naar laag)")
|
|
|
|
|
|
df = pd.DataFrame(party_matches)
|
|
|
df.columns = ["Partij", "Overeenkomst %", "Eens", "Totaal"]
|
|
|
|
|
|
# Style the dataframe
|
|
|
def color_agreement(val):
|
|
|
if val >= 80:
|
|
|
return "background-color: #d4edda"
|
|
|
elif val >= 60:
|
|
|
return "background-color: #fff3cd"
|
|
|
else:
|
|
|
return "background-color: #f8d7da"
|
|
|
|
|
|
styled_df = df.style.applymap(color_agreement, subset=["Overeenkomst %"])
|
|
|
st.dataframe(styled_df, use_container_width=True, hide_index=True)
|
|
|
|
|
|
# Top match highlight
|
|
|
top_match = party_matches[0]
|
|
|
st.success(
|
|
|
f"🏆 **Beste match:** {top_match['party']} ({top_match['agreement_percentage']}% overeenkomst)"
|
|
|
)
|
|
|
|
|
|
# Detailed motion overview
|
|
|
st.subheader("📋 Gedetailleerd Overzicht per Motie")
|
|
|
show_detailed_motion_results()
|
|
|
|
|
|
# New session button
|
|
|
if st.button("🔄 Start Nieuwe Sessie"):
|
|
|
# Clear session state
|
|
|
for key in ["session_id", "motions", "current_motion_index", "show_results"]:
|
|
|
if key in st.session_state:
|
|
|
del st.session_state[key]
|
|
|
st.rerun()
|
|
|
|
|
|
|
|
|
def show_detailed_motion_results():
|
|
|
"""Show detailed voting results for each motion"""
|
|
|
import duckdb
|
|
|
|
|
|
conn = duckdb.connect(config.DATABASE_PATH)
|
|
|
|
|
|
# Get user votes
|
|
|
user_data = conn.execute(
|
|
|
"""
|
|
|
SELECT user_votes FROM user_sessions WHERE session_id = ?
|
|
|
""",
|
|
|
(st.session_state.session_id,),
|
|
|
).fetchone()
|
|
|
|
|
|
if not user_data:
|
|
|
return
|
|
|
|
|
|
user_votes = json.loads(user_data[0])
|
|
|
|
|
|
# Get motion details
|
|
|
motion_ids = list(user_votes.keys())
|
|
|
if motion_ids:
|
|
|
placeholders = ",".join(["?" for _ in motion_ids])
|
|
|
motions = conn.execute(
|
|
|
f"""
|
|
|
SELECT id, title, layman_explanation, body_text, description, voting_results FROM motions
|
|
|
WHERE id IN ({placeholders})
|
|
|
""",
|
|
|
motion_ids,
|
|
|
).fetchall()
|
|
|
|
|
|
for (
|
|
|
motion_id,
|
|
|
title,
|
|
|
layman_explanation,
|
|
|
body_text,
|
|
|
description,
|
|
|
voting_results_json,
|
|
|
) in motions:
|
|
|
voting_results = json.loads(voting_results_json)
|
|
|
user_vote = user_votes[str(motion_id)]
|
|
|
|
|
|
with st.expander(f"**{title}** (Jouw stem: {user_vote})"):
|
|
|
# Show layman explanation prominently
|
|
|
if layman_explanation:
|
|
|
st.markdown("**📝 Uitleg:**")
|
|
|
st.markdown(f"*{layman_explanation}*")
|
|
|
|
|
|
# Show full motion body text if available, otherwise description
|
|
|
motion_text = body_text or description
|
|
|
if motion_text:
|
|
|
st.markdown("**📋 Motiebeschrijving:**")
|
|
|
st.write(motion_text)
|
|
|
|
|
|
# Create voting overview
|
|
|
parties_voor = [p for p, v in voting_results.items() if v == "voor"]
|
|
|
parties_tegen = [p for p, v in voting_results.items() if v == "tegen"]
|
|
|
|
|
|
col1, col2 = st.columns(2)
|
|
|
with col1:
|
|
|
st.markdown("**Voor:**")
|
|
|
st.write(", ".join(parties_voor) if parties_voor else "Geen")
|
|
|
|
|
|
with col2:
|
|
|
st.markdown("**Tegen:**")
|
|
|
st.write(", ".join(parties_tegen) if parties_tegen else "Geen")
|
|
|
|
|
|
conn.close()
|
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
main()
|
|
|
|
