import streamlit as st
import sqlite3

st.title("Task 2: Insert Data")

db_name = st.text_input("Enter database name:", "example.db")
table_name = st.text_input("Enter table name:", "students")
data = st.text_area("Enter data (e.g., 1, 'John Doe'):")

if st.button("Insert Data"):
    try:
        conn = sqlite3.connect(db_name)
        cursor = conn.cursor()
        cursor.execute(f"INSERT INTO {table_name} VALUES ({data})")
        conn.commit()
        st.success("Data inserted successfully!")
    except Exception as e:
        st.error(f"Error: {e}")
    finally:
        conn.close()