Update app.py

app.py

@@ -1,9 +1,10 @@
+# app.py
 import streamlit as st
 from transformers import pipeline
 import time
 
 st.set_page_config(
-    page_title="Cosmetic Review AI Analyst",
+    page_title="Cosmetic Review Analyst",
     layout="wide",
     initial_sidebar_state="expanded",
 )
@@ -36,9 +37,9 @@ def load_css():
 @st.cache_resource(show_spinner=False)
 def load_models():
     summarizer = pipeline(
-        "text-generation",
-        model="Lin0He/text-summary-gpt2-short",
-        max_length=150,
+        "summarization",
+        model="Falconsai/text_summarization",
+        max_length=200,
         temperature=0.7
     )
     
@@ -52,36 +53,13 @@ def load_models():
 def main():
     load_css()
     st.title("💄 Cosmetic Review AI Analyst")
-    st.subheader("AI-Powered E-commerce Review Insight Engine")
-    
-    with st.sidebar:
-        st.header("Configuration Parameters")
-        
-        with st.container(border=True):
-            st.markdown("""
-            **Adjust parameters before analysis:**
-            - `Max Summary Length`: Control generated summary's word count (50-200)
-            - `Confidence Threshold`: Set minimum confidence for positive classification
-            """)
-        
-        st.markdown("### 🔧 Summary Settings")
-        max_length = st.slider(
-            "Max Summary Length", 
-            50, 200, 120,
-            help="Maximum number of tokens in generated summary"
-        )
-        
-        st.markdown("### 🎚️ Sentiment Threshold")
-        confidence_threshold = st.slider(
-            "Sentiment Confidence Threshold", 
-            0.5, 1.0, 0.8,
-            help="Minimum confidence score for positive sentiment classification"
-        )
+    st.warning("⚠️ Please keep reviews under 200 words for optimal analysis")
     
     user_input = st.text_area(
-        "Input cosmetic product review (Supports both Chinese/English)", 
+        "Input cosmetic product review (Chinese/English supported)", 
         height=200,
-        placeholder="Example: This serum is amazing! After one week of use, my skin looks visibly brighter..."
+        placeholder="Example: This serum transformed my skin in just 3 days...",
+        help="Maximum 200 characters recommended"
     )
     
     if st.button("Start Analysis", use_container_width=True):
@@ -89,47 +67,50 @@ def main():
             st.error("⚠️ Please input valid review content")
             return
             
-        with st.spinner('🔍 AI is analyzing...'):
-            with st.spinner('🚀 Loading AI models...'):
+        with st.spinner('🔍 Analyzing...'):
+            try:
                 summarizer, classifier = load_models()
-            
-            st.toast('Models ready!', icon='✅')
-            
-            with st.expander("Original Review", expanded=True):
-                st.write(user_input)
-            
-            summary = summarizer(user_input, max_length=max_length)[0]['generated_text']
-            with st.container():
-                col1, col2 = st.columns([1, 3])
-                with col1:
-                    st.subheader("📝 Smart Summary")
-                with col2:
-                    st.markdown(f"```\n{summary}\n```")
-            
-            results = classifier(summary)
-            positive_score = results[0][1]['score']
-            label = "Positive 👍" if positive_score > confidence_threshold else "Negative 👎"
-            
-            with st.container():
-                st.subheader("📊 Sentiment Report")
-                col1, col2 = st.columns(2)
-                with col1:
-                    st.metric("Sentiment", label)
-                    st.write(f"Confidence: {positive_score:.2%}")
-                with col2:
-                    progress_color = "#4CAF50" if label=="Positive 👍" else "#FF5252"
-                    st.markdown(f"""
-                    <div style="
-                        background: {progress_color}10;
-                        border-radius: 10px;
-                        padding: 15px;
-                    ">
-                        <div style="font-size: 14px; color: {progress_color}; margin-bottom: 8px;">Sentiment Intensity</div>
-                        <div style="height: 8px; background: #eee; border-radius: 4px;">
-                            <div style="width: {positive_score*100}%; height: 100%; background: {progress_color}; border-radius: 4px;"></div>
+                
+                with st.expander("Original Review", expanded=True):
+                    st.write(user_input)
+                
+                # Text summarization
+                summary = summarizer(user_input, max_length=200)[0]['summary_text']
+                with st.container():
+                    col1, col2 = st.columns([1, 3])
+                    with col1:
+                        st.subheader("📝 Summary")
+                    with col2:
+                        st.markdown(f"```\n{summary}\n```")
+                
+                # Sentiment analysis
+                results = classifier(summary)
+                positive_score = results[0][1]['score']
+                label = "Positive 👍" if positive_score > 0.5 else "Negative 👎"
+                
+                with st.container():
+                    st.subheader("📊 Sentiment Analysis")
+                    col1, col2 = st.columns(2)
+                    with col1:
+                        st.metric("Verdict", label)
+                        st.write(f"Confidence: {positive_score:.2%}")
+                    with col2:
+                        progress_color = "#4CAF50" if label=="Positive 👍" else "#FF5252"
+                        st.markdown(f"""
+                        <div style="
+                            background: {progress_color}10;
+                            border-radius: 10px;
+                            padding: 15px;
+                        ">
+                            <div style="font-size: 14px; color: {progress_color}; margin-bottom: 8px;">Intensity</div>
+                            <div style="height: 8px; background: #eee; border-radius: 4px;">
+                                <div style="width: {positive_score*100}%; height: 100%; background: {progress_color}; border-radius: 4px;"></div>
+                            </div>
                         </div>
-                    </div>
-                    """, unsafe_allow_html=True)
+                        """, unsafe_allow_html=True)
+                        
+            except Exception as e:
+                st.error(f"Analysis failed: {str(e)}")
 
 if __name__ == "__main__":
-    main() 
+    main()