Upload streamlit_app.py

src/streamlit_app.py

@@ -1,10 +1,22 @@
 import streamlit as st
+import time
+import os
 from supabase import create_client
 from datetime import datetime
 from zoneinfo import ZoneInfo
 from supabase_auth.errors import AuthApiError
-import time
-import os
+from transformers import pipeline
+
+
+@st.cache_resource
+def load_sentiment_model():
+    return pipeline(
+        "sentiment-analysis",
+        model="wonrax/phobert-base-vietnamese-sentiment"
+    )
+sentiment_model = load_sentiment_model()
+
+
 # ========== CONFIG ==========
 st.set_page_config(
     page_title="Personal Diary",
@@ -42,6 +54,10 @@ def signup(email, password):
         "password": password
     })
 
+def analyze_sentiment(text: str) -> str:
+    result = sentiment_model(text)[0]
+    return result["label"]  # POS / NEG / NEU
+
 
 # ========== HEADER ==========
 st.title("📔 Personal Diary")
@@ -159,9 +175,11 @@ else:
 
         if st.button("💾 Lưu nhật ký", use_container_width=True):
             if content.strip():
+                sentiment = analyze_sentiment(content)
                 supabase.table("journals").insert({
                     "user_id": user.id,
                     "content": content,
+                    "sentiment": sentiment,
                     "created_at": datetime.utcnow().isoformat()
                 }).execute()
 
@@ -200,6 +218,13 @@ else:
                         key=row["id"],
                         height=120
                     )
+                    sentiment = row.get("sentiment", "N/A")
+                    emoji_map = {
+                        "Vui vẻ": "😊",
+                        "Buồn": "😔",
+                        "Trung tính": "😐"
+                    }
+                    st.caption(f"Cảm xúc: {emoji_map.get(sentiment, '❓')} {sentiment}")
 
                     col_u, col_d = st.columns(2)