| import streamlit as st |
| from sqlalchemy import text |
|
|
| from utils.db_connection import get_connection |
| from utils.inference import predict_theme |
|
|
|
|
| def _insert(product, rating, review, sentiment, theme): |
| conn = get_connection() |
| with conn.session as session: |
| session.execute( |
| text("INSERT INTO user_input (product, rating, review, sentiment, theme) " |
| "VALUES (:product, :rating, :review, :sentiment, :theme)"), |
| {"product": product, "rating": rating, "review": review, |
| "sentiment": sentiment, "theme": theme}, |
| ) |
| session.commit() |
|
|
|
|
| def run(): |
| st.header("📝 Submit a Review") |
| st.write("Add a new customer review to the database. The app predicts its theme on the way " |
| "in, so the dashboard sees new reviews already categorized.") |
|
|
| with st.form("review_form"): |
| product = st.text_input("Product") |
| rating = st.slider("Rating", min_value=1, max_value=5, value=5) |
| review = st.text_area("Review") |
| submit = st.form_submit_button("Submit", type="primary") |
|
|
| if submit: |
| if not product.strip() or not review.strip(): |
| st.warning("Please complete all fields.") |
| return |
|
|
| |
| sentiment, theme = None, None |
| try: |
| _, ranked = predict_theme(review.strip()) |
| if ranked: |
| sentiment, theme = ranked[0][0], ranked[0][1] |
| except Exception: |
| pass |
|
|
| try: |
| _insert(product.strip(), rating, review.strip(), sentiment, theme) |
| if theme: |
| st.success(f"Saved! Predicted theme: **{theme}** ({sentiment.lower()} review)") |
| else: |
| st.success("Review saved. (Theme will be assigned later.)") |
| except Exception as e: |
| st.error("Could not save to the database - check the Supabase connection / secrets.") |
| st.caption(f"Details: {e}") |
|
|
