import os
import pandas as pd
import gradio as gr
from flask import Flask, request, jsonify

ruta_candidatos = 'Candidatos.txt'
candidatos = {}
with open(ruta_candidatos, "r") as f:
    for line in f:
        if line.strip():
            candidato, partido = line.split(":")
            candidatos[candidato.strip()] = partido.strip()

votos = pd.DataFrame(columns=["Candidato", "Votos"])

app = Flask(__name__)

@app.route("/votacion", methods=["POST"])
def votacion():
    nombre_amigo = request.form["nombre_amigo"]
    candidato = request.form["candidato"]

    if candidato not in candidatos:
        return jsonify({"Resultado": "Candidato no válido."})

    votos.loc[len(votos)] = [candidato, 1]
    return jsonify({"Resultado": "Gracias por tu voto."})

@app.route("/candidatos", methods=["GET"])
def obtener_candidatos():
    return jsonify(candidatos)

@app.route("/votos", methods=["GET"])
def obtener_votos():
    return votos.to_json(orient="records")

if __name__ == "__main__":
    app.run()