dev:sqlite

app.py

@@ -21,9 +21,7 @@ std_edp = 'https://huggingface.co/datasets/strauss-oak/afa-bd/resolve/main/tb-st
 
 conn = sqlite3.connect("afa-db.db")
 
-sql_example = """SELECT alu_id, alu_nome, alu_uf, alu_cep, progresso_alfabetizacao 
-                FROM tb_students
-                WHERE progresso_alfabetizacao >= 6 LIMIT 100"""
+sql_example = """SELECT alu_id, alu_nome, alu_uf, alu_cep, progresso_alfabetizacao FROM tb_students WHERE progresso_alfabetizacao >= 6 LIMIT 100"""
 
 txt_example = """Qual um dos estudantes com maior nota no estado de SP?"""
 
@@ -51,6 +49,14 @@ def refresh_data_students():
     df_std = df_std[:10]
     return df_std, n_std
 
+def execute_sql(sql):
+    data = None
+    try:
+        data = pd.read_sql(sql, conn)
+    except Exception as err:
+        st.error(err)
+    return data
+
 st.info('Amostras de Dados:')
 e1 = st.expander('Professores:', expanded=False)
 with e1:
@@ -74,10 +80,16 @@ st.info('Consultas:')
 e4 = st.expander('Linguagem SQL:', expanded=False)
 with e4:
     sql = st.text_input("Digite uma instrução SQL válida:", sql_example)
+    if st.button("Processar Consulta"):
+        df = execute_sql(sql)
+        if df is not None:
+            st.dataframe(df)
 
 e5 = st.expander('Linguagem Natural:', expanded=False)
 with e5:
     txt = st.text_input("Digite um texto de consulta válido:", txt_example)
+    if st.button("Processar Consulta"):
+        st.write("Em construção")
 
 
 #sql = """SELECT alu_id, alu_nome, alu_uf, alu_cep, progresso_alfabetizacao FROM tb_students