pages/Task3_Query_data.py

import streamlit as st
import sqlite3
import pandas as pd

st.title("Task 3: Query Data")

db_name = st.text_input("Enter database name:", "example.db")
query = st.text_area("Enter SQL query (e.g., SELECT * FROM students):")

if st.button("Execute Query"):
    try:
        conn = sqlite3.connect(db_name)
        result = pd.read_sql_query(query, conn)
        st.write(result)
    except Exception as e:
        st.error(f"Error: {e}")
    finally:
        conn.close()