use API access instaed of downloding the model

app.py

@@ -2,25 +2,58 @@ import streamlit as st
 from pipeline import text_to_sql
 
 st.title("SQLCoder Text-to-SQL App")
+st.write("Powered by defog/sqlcoder-7b-2 🚀")
 
-nl_query = st.text_input("Enter your query:", value="List 11 names of ships type schooner")
+# Sample queries for user guidance
+st.sidebar.header("Sample Queries")
+sample_queries = [
+    "List 11 names of ships type schooner",
+    "Show me the 5 oldest ships",
+    "What are the different types of vessels?",
+    "Count the number of ships by type",
+    "Show ships built after 1900"
+]
 
-if st.button("Generate & Execute"):
-    if nl_query:
+selected_sample = st.sidebar.selectbox("Choose a sample query:", [""] + sample_queries)
+
+# Main input
+nl_query = st.text_input(
+    "Enter your natural language query:", 
+    value=selected_sample if selected_sample else "List 11 names of ships type schooner",
+    help="Ask questions about your database in plain English"
+)
+
+if st.button("🔄 Generate & Execute SQL"):
+    if nl_query.strip():
         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)
+            try:
+                sql, results = text_to_sql(nl_query)
+                
+                # Display results
+                st.success("Query executed successfully!")
+                
+                # Show generated SQL
+                st.subheader("Generated SQL:")
+                st.code(sql, language="sql")
+                
+                # Show results
+                st.subheader("Results:")
+                if results:
+                    # Convert results to a more readable format
+                    if isinstance(results[0], tuple):
+                        # If results are tuples, display as table
+                        st.write(f"Found {len(results)} rows:")
+                        for i, row in enumerate(results[:50]):  # Show first 50 rows
+                            st.write(f"Row {i+1}: {row}")
+                        if len(results) > 50:
+                            st.info(f"Showing first 50 rows out of {len(results)} total results.")
+                    else:
+                        st.write(results)
+                else:
+                    st.info("Query executed successfully but returned no results.")
+                    
+            except Exception as e:
+                st.error(f"Error: {str(e)}")
+                st.write("Please try rephrasing your query or check if the requested data exists in the database.")
     else:
-        st.error("Enter a query.")
+        st.warning("Please enter a query to proceed.")