give view support

app.py

@@ -239,8 +239,12 @@ def execute_query(query):
     try:
         start_time = time.time()
         with st.spinner("Executing SQL query..."):
-            with st.session_state.engine.connect() as conn:
-                df = pd.read_sql(query, conn)
+            # Create a connection and begin a transaction
+            with st.session_state.engine.begin() as conn:
+                # Execute the query using text() to ensure proper SQL compilation
+                result = conn.execute(text(query))
+                # Convert the result to a pandas DataFrame
+                df = pd.DataFrame(result.fetchall(), columns=result.keys())
             execution_time = time.time() - start_time
             pprint(f"[Query Execution] Latency: {execution_time:.2f}s")
         return df
@@ -325,27 +329,45 @@ if connection_string and connection_string != st.session_state.connection_string
 
 # Table Selection Section
 if st.session_state.connection_string:
-    st.header("2. Table Selection")
+    st.header("2. Database Object Selection")
     inspector = inspect(st.session_state.engine)
+    
+    # Get both tables and views
     tables = inspector.get_table_names()
+    views = inspector.get_view_names()
+    
+    # Create a list of tuples with (name, type) for all database objects
+    db_objects = [(table, 'Table') for table in tables] + [(view, 'View') for view in views]
+    db_objects.sort(key=lambda x: x[0])  # Sort alphabetically by name
+    
+    # Create two columns for the selection
+    col1, col2 = st.columns([3, 1])
+    
+    with col1:
+        # Extract just the names for the selectbox
+        object_names = [obj[0] for obj in db_objects]
+        # Set default index to 'lsq_leads' if present, otherwise 0
+        default_index = object_names.index('lsq_leads') if 'lsq_leads' in object_names else 0
+        selected_object = st.selectbox("Select a table or view", object_names, index=default_index)
     
-    # Set default index to 'lsq_leads' if present, otherwise 0
-    default_index = tables.index('lsq_leads') if 'lsq_leads' in tables else 0
-    selected_table = st.selectbox("Select a table", tables, index=default_index)
+    with col2:
+        # Display the object type (Table/View)
+        object_type = next(obj_type for obj_name, obj_type in db_objects if obj_name == selected_object)
+        st.text_input("Type", value=object_type, disabled=True)
     
     # Create containers for schema and data
     schema_container = st.container()
     data_container = st.container()
     
-    # Always load table data if we have a selected table
-    if selected_table:
+    # Always load object data if we have a selection
+    if selected_object:
         # Update session state
-        if selected_table != st.session_state.selected_table:
-            st.session_state.selected_table = selected_table
+        if selected_object != st.session_state.selected_table:
+            st.session_state.selected_table = selected_object
         
         # Always fetch schema and sample data
-        st.session_state.table_schema = get_table_schema(selected_table)
-        st.session_state.sample_data = get_sample_data(selected_table)
+        st.session_state.table_schema = get_table_schema(selected_object)
+        st.session_state.sample_data = get_sample_data(selected_object)
         
         # Always display schema and sample data if available
         with schema_container: