Upload app.py

app.py

@@ -1,9 +1,9 @@
 import os
 import pandas as pd
 import gradio as gr
-from flask import Flask, request
+from flask import Flask, request, jsonify
 
-ruta_candidatos = os.path.join(os.path.dirname(__file__), "Candidatos.txt")
+ruta_candidatos = 'Candidatos.txt'
 candidatos = {}
 with open(ruta_candidatos, "r") as f:
     for line in f:
@@ -11,53 +11,28 @@ with open(ruta_candidatos, "r") as f:
             candidato, partido = line.split(":")
             candidatos[candidato.strip()] = partido.strip()
 
-votos = pd.DataFrame(columns=["IP", "Candidato", "Votos", "Nombre"])
+votos = pd.DataFrame(columns=["Candidato", "Votos"])
 
 app = Flask(__name__)
-access_granted = False
-
-@app.route("/")
-def index():
-    iface = gr.Interface(
-        fn=votacion,
-        inputs=["text", gr.inputs.Radio(list(candidatos.keys()))],
-        outputs="text",
-        title="Elecciones Presidenciales 2023",
-        description="Ingresa el nombre de tu amigo votante y selecciona un candidato."
-    )
-    return iface.launch(share=False)
-
-@app.route("/resultados", methods=["GET", "POST"])
-def resultados():
-    global access_granted
-
-    if request.method == "POST":
-        password = request.form.get("password")
-        if password == "notiene":
-            access_granted = True
-        else:
-            access_granted = False
-
-    if access_granted:
-        return votos.to_html()
-    else:
-        return '''
-            <form method="POST" action="/resultados">
-                <input type="password" name="password" placeholder="Ingrese la contraseña" required>
-                <button type="submit">Acceder</button>
-            </form>
-        '''
-
-def votacion(nombre_amigo, candidato):
-    ip = request.headers["x-forwarded-for"] if "x-forwarded-for" in request.headers else request.remote_addr
-    if ip in votos["IP"].values:
-        return "Ya has votado."
-    else:
-        if candidato not in candidatos:
-            return "Candidato no válido."
-        
-        votos.loc[len(votos)] = [ip, candidato, 1, nombre_amigo]
-        return "Gracias por tu voto."
+
+@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()