Update app.py

app.py

@@ -1,6 +1,8 @@
 import streamlit as st
 from openai import OpenAI
 import sqlite3
+import pandas as pd  # Add to top of file
+
 
 # App UI
 st.set_page_config(page_title="Zero SQL", layout="wide")
@@ -133,9 +135,9 @@ if submitted:
 
             st.subheader("Results")
             if results:
+                df = pd.DataFrame(results, columns=column_names)
                 st.dataframe(
-                    data=results,
-                    columns=columns,
+                    data=df,
                     use_container_width=True,
                     hide_index=True
                 )