Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 import streamlit as st
 from transformers import pipeline
 import time
@@ -36,7 +35,6 @@ def load_css():
 
 @st.cache_resource(show_spinner=False)
 def load_models():
-    # 移除所有Streamlit UI组件
     summarizer = pipeline(
         "text-generation",
         model="Lin0He/text-summary-gpt2-short",
@@ -92,7 +90,6 @@ def main():
             return
             
         with st.spinner('🔍 AI is analyzing...'):
-            # 将模型加载与UI反馈分离
             with st.spinner('🚀 Loading AI models...'):
                 summarizer, classifier = load_models()
             
@@ -122,4 +119,17 @@ def main():
                 with col2:
                     progress_color = "#4CAF50" if label=="Positive 👍" else "#FF5252"
                     st.markdown(f"""
-                    <div style="
+                    <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>
+                        </div>
+                    </div>
+                    """, unsafe_allow_html=True)
+
+if __name__ == "__main__":
+    main()