import streamlit as st
import sqlite3

st.title("Task 5: Delete Data")

db_name = st.text_input("Enter database name:", "example.db")
delete_query = st.text_area("Enter SQL delete query (e.g., DELETE FROM students WHERE id=1):")

if st.button("Delete Data"):
    try:
        conn = sqlite3.connect(db_name)
        cursor = conn.cursor()
        cursor.execute(delete_query)
        conn.commit()
        rows_affected = cursor.rowcount
        if rows_affected == 0:
            st.warning("No data deleted. Please check your query.")
        else:
            st.success("Data deleted successfully!")
    except Exception as e:
        st.error(f"Error: {e}")
    finally:
        conn.close()