Spaces:
Sleeping
Sleeping
import streamlit as st | |
import sqlite3 | |
import pandas as pd | |
# Initialize SQLite database | |
conn = sqlite3.connect('word_classification.db') | |
c = conn.cursor() | |
# Create table if it doesn't exist | |
c.execute(''' | |
CREATE TABLE IF NOT EXISTS word_classification ( | |
word TEXT, | |
category TEXT | |
) | |
''') | |
# Function to save word and its category | |
def save_word(word, category): | |
c.execute("INSERT INTO word_classification (word, category) VALUES (?, ?)", | |
(word, category)) | |
conn.commit() | |
# Function to load words | |
def load_words(): | |
df = pd.read_sql_query("SELECT * FROM word_classification", conn) | |
return df | |
# Streamlit UI | |
st.title('Word Classification') | |
# Input fields for word and category | |
word = st.text_input('Enter Word') | |
category = st.selectbox('Select Category', ['Theme', 'Subtheme', 'Keywords']) | |
if st.button('Save Word'): | |
save_word(word, category) | |
st.success(f'Word "{word}" saved under category "{category}"!') | |
if st.button('View All Entries'): | |
df = load_words() | |
st.dataframe(df) | |
# Close the database connection when the app is done | |
conn.close() | |