Update script.py

script.py

@@ -29,7 +29,7 @@ def _get_metadata():
         cs.append(cols)
 
     con.close()
-    return {'table': ts, 'columns': cs}
+    return pd.DataFrame({'table': ts, 'columns': cs})
 
 st.title("English to SQL via LangChain")
 st.subheader("table metadata")
@@ -44,8 +44,8 @@ for i, query in enumerate(QUERIES):
     output = db_chain(query)
     sql, result = output["intermediate_steps"][1], output["result"]
 
-    st.subheader("query %s" % i)
+    st.subheader("query %s" % 1 + i)
     
     st.write("Q: %s" % query)
-    st.write("(sql: %s)" % sql)
+    st.code("(sql: %s)" % sql)
     st.write("A: %s" % result)