fix

app.py

@@ -23,11 +23,11 @@ st.sidebar.header('Sample Selection')
 index_example = st.sidebar.number_input(f"Choose a sample from the existing {len(samples)} notebooks:", min_value=0, max_value=max(0, len(samples)-1), value=0, step=1)
 
 db_id = samples[index_example]["db_id"]
-st.markdown(f'<h2 style="color:blue;">Database {db_id} schema:</h2>', unsafe_allow_html=True)
-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.markdown(f'<h2 style="color:blue;">SQL:</h2>', unsafe_allow_html=True)
 st.code(sql_str_pretty)
+st.markdown(f'<h2 style="color:blue;">Database schema:</h2>', unsafe_allow_html=True)
+st.code(samples[index_example]["schema"])