app.py original

app.py

@@ -1,87 +1,26 @@
 import streamlit as st
-import os
-import psycopg2
-from sqlalchemy import create_engine, text
-from dotenv import load_dotenv
+from pipeline import text_to_sql
 
-# Load environment variables
-load_dotenv()
+st.title("SQLCoder Text-to-SQL App")
 
-st.title("Database Connection Test")
+nl_query = st.text_input("Enter your query:", value="List 11 names of ships type schooner")
 
-def test_db_connection():
-    """Test database connection and basic query"""
-    try:
-        conn_str = os.getenv("SUPABASE_CONN_STR")
-        if not conn_str:
-            return False, "SUPABASE_CONN_STR environment variable not found"
-        
-        st.write(f"Connection string configured: {bool(conn_str)}")
-        
-        # Test basic connection
-        conn = psycopg2.connect(conn_str)
-        cursor = conn.cursor()
-        
-        # Test simple query
-        cursor.execute("SELECT version();")
-        version = cursor.fetchone()
-        
-        # Test schema access
-        cursor.execute("""
-            SELECT table_name, column_name, data_type
-            FROM information_schema.columns
-            WHERE table_schema = 'public'
-            ORDER BY table_name, ordinal_position
-            LIMIT 10;
-        """)
-        schema_sample = cursor.fetchall()
-        
-        # Test vessels table specifically
-        cursor.execute("SELECT COUNT(*) FROM vessels;")
-        vessel_count = cursor.fetchone()[0]
-        
-        cursor.close()
-        conn.close()
-        
-        return True, {
-            "version": version[0] if version else "Unknown",
-            "schema_sample": schema_sample,
-            "vessel_count": vessel_count
-        }
-        
-    except Exception as e:
-        return False, str(e)
-
-if st.button("Test Database Connection"):
-    success, result = test_db_connection()
-    
-    if success:
-        st.success("✅ Database connection successful!")
-        st.write("**Database Version:**", result["version"])
-        st.write("**Total vessels:**", result["vessel_count"])
-        st.write("**Schema Sample (first 10 columns):**")
-        for table, column, dtype in result["schema_sample"]:
-            st.write(f"- {table}.{column} ({dtype})")
-    else:
-        st.error(f"❌ Database connection failed: {result}")
-
-# Manual SQL test
-st.subheader("Manual SQL Test")
-manual_sql = st.text_area("Enter SQL to test:", value="SELECT name, vessel_type FROM vessels LIMIT 5;")
-
-if st.button("Execute SQL"):
-    try:
-        conn_str = os.getenv("SUPABASE_CONN_STR")
-        conn = psycopg2.connect(conn_str)
-        cursor = conn.cursor()
-        cursor.execute(manual_sql)
-        results = cursor.fetchall()
-        cursor.close()
-        conn.close()
-        
-        st.success(f"Query executed successfully! Found {len(results)} rows:")
-        for i, row in enumerate(results):
-            st.write(f"Row {i+1}: {row}")
+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")
             
-    except Exception as e:
-        st.error(f"SQL execution failed: {e}")
+            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.")