pretty sql

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 import json
 from datasets import load_dataset
+import sqlparse
 
 st.set_page_config(page_title="BIRD SQL inspection", layout="wide")
 
@@ -26,5 +27,7 @@ st.markdown(f'<h2 style="color:blue;">Database {db_id} schema:</h2>', unsafe_all
 st.code(samples[index_example]["schema"])
 st.markdown(f'<h2 style="color:blue;">{index_example} Question:</h2>', unsafe_allow_html=True)
 st.code(samples[index_example]["question"])
+sql_str = samples[index_example]["SQL"]
+sql_str_pretty = sqlparse.format(sql_str, reindent=True, keyword_case='upper')
 st.markdown(f'<h2 style="color:blue;">{index_example} SQL:</h2>', unsafe_allow_html=True)
-st.code(samples[index_example]["SQL"])
+st.code(sql_str_pretty)

requirements.txt

@@ -1,2 +1,3 @@
 protobuf~=3.19.0
-altair==4.2.2
+altair==4.2.2
+sqlparse