final02.py

from flask import Flask, request,  render_template_string
import pandas as pd 
import sqlite3
import plotly.express as px 
import plotly.io as pio
import os

#configura o plotly para abrir os arquivos no navegador por padrão
pio.renderers.default = 'browser'

# carregar o drinks.csv
script_dir = os.path.dirname(os.path.abspath(__file__))
df = pd.read_csv(os.path.join(script_dir,'drinks.csv'))

# cria o banco de dados em sql e popular com os dados do arquivo csv
conn = sqlite3.connect('consumo_alcool.db')
df.to_sql('drinks', conn, if_exists="replace", index=False)
conn.commit()
conn.close()

#inicia o flask
app = Flask(__name__)
html_template = '''
    <h1>Dashboard - Consumo de Alcool</h1>
    <h2>Menu </h2>
        <ul>
            <li> <a href='/grafico1'> Média de consumo por tipo de bebida</a> </li>
            <li> <a href='/grafico2'> Comparativo  entre os tipos de bebidas</a> </li>
            <li> <a href='/comparar'> Comparar</a> </li>
        </ul>
    '''

# rota inicial com links para os graficos1
@app.route('/grafico1')
def grafico1():
    conn = sqlite3.connect('consumo_alcool.db')
    df = pd.read_sql_query('SELECT AVG(beer_servings) AS cerveja, AVG (spirit_servings) AS destilados, AVG(wine_servings) AS vinhos FROM drinks', conn)
    df_melted = df_melted(var_name='Bebidas', value_name="Média de Porções")
    fig = px.bar(df_melted, x="Bebidas", y="Média de Porções", title="Média de consumo global por tipo")
    return fig.to_html

@app.route('/grafico2')
def grafico2():
    conn = sqlite3.connect("consumo_alcool.db")
    df = pd.read_sql_query("SELECT beer_servings, spirit_servings, wine_servings FROM drinks", conn)
    conn.close()
    medias = df.mean().reset_index()
    medias_columns = ['Tipo', 'Média']
    fig = px.pie(medias,names='Tipo', values='Média', title = "Proporção média entre tipos de bebidas")
    return fig.to_html() +  '<br><a href="/">Voltar ao inicio</a>'

@app.route("/comparar", methods = ['GET', 'POST'])
def comparar():
    opcoes = ["beer_servings", "spirit_servings", "wine_servings", "total_litres_of_pure_alcohol"]
    if request.method == "POST":
        eixo_x = request.form.get('eixo_x')
        eixo_y = request.form.get('eixo_y')
        if eixo_x == eixo_y:
            return "<h3>Selecione variaveis diferentes!.</h3>"
        
        conn =sqlite3.connect("Consumo_alcool.db")
        df = pd.read_sql_query("SELECT country, {}, {} FROM drinks".format(eixo_x,eixo_y), conn)
        fig =px.scatter(df, x=eixo_x, y=eixo_y, title=f"Comparação entre {eixo_x} e {eixo_y}")
        fig.update_traces(textposition="top center")
        return fig.to_html() + "<br><a href='/'> Voltar ao incio</a>"
    
    return render_template_string('''
        <h2>Comparar Campos</h2>
        <form method="POST">
            <label for="eixo_x"> Eixo X: </label>    
            <select name="eixo_x">
                {% for cal in opcoes %}
                    <option value="{{col}}"> {{col}}  </option>  
                {%endfor%}  
            <select><br><br> 
            <label for="eixo_y"> Eixo Y: </label>    
            <select name="eixo_y">
                {% for cal in opcoes %}
                    <option value="{{col}}"> {{col}}  </option>  
                {%endfor%}  
            <select><br><br> 
            <input type="submit" value=" - Comparar -">      
        </form>     
    ''', opcoes=opcoes)

if __name__ == '__main__':
        app.run(debug=True)