import streamlit as st
from pipeline import text_to_sql

st.title("SQLCoder Text-to-SQL App")

nl_query = st.text_input("Enter your query:", value="List 11 names of ships type schooner")

if st.button("Generate & Execute"):
    if nl_query:
        with st.spinner("Generating SQL and executing query..."):
            sql, results = text_to_sql(nl_query)
            
            st.write("**Generated SQL:**")
            st.code(sql, language="sql")
            
            st.write("**Results:**")
            if isinstance(results, list) and results:
                st.write(f"Found {len(results)} rows:")
                for i, row in enumerate(results[:20]):  # Show first 20
                    st.write(f"Row {i+1}: {row}")
                if len(results) > 20:
                    st.info(f"Showing first 20 rows out of {len(results)} total results.")
            else:
                st.write(results)
    else:
        st.error("Enter a query.")