Update app.py

app.py

@@ -0,0 +1,183 @@
+import streamlit as st
+from transformers import pipeline
+import pandas as pd
+from typing import List, Dict
+import json
+import plotly.express as px
+from datetime import datetime
+
+# Initialize sentiment analysis pipeline
+@st.cache_resource
+def load_model():
+    return pipeline("sentiment-analysis")
+
+class GraicieAnalyzer:
+    def __init__(self):
+        """Initialize the analyzer"""
+        self.sentiment_pipeline = load_model()
+        
+    def analyze_post(self, post_text: str) -> Dict:
+        """Analyze a single post and return all metrics"""
+        # Get base sentiment
+        sentiment_result = self.sentiment_pipeline(post_text)[0]
+        
+        # Determine style based on post characteristics
+        style = self._determine_style(post_text)
+        
+        # Determine tone based on content
+        tones = self._determine_tone(post_text, sentiment_result)
+        
+        # Calculate rating
+        rating = self._calculate_rating(post_text)
+        
+        return {
+            "post": post_text,
+            "style": style,
+            "tones": tones,
+            "rating": rating,
+            "sentiment": sentiment_result["label"],
+            "confidence": f"{sentiment_result['score']:.2%}"
+        }
+    
+    def _determine_style(self, text: str) -> str:
+        """Determine post style based on characteristics"""
+        text_lower = text.lower()
+        
+        if "!" in text or "?" in text or "😱" in text or "🤯" in text:
+            return "Hooking"
+        elif len(text.split()) <= 10:
+            return "Short & Sweet"
+        elif text.count('🏆') > 0 or text.count('📈') > 0:
+            return "Audience-focused"
+        elif text.count('😊') > 0 or "i feel" in text_lower or "just" in text_lower:
+            return "Authentic"
+        elif text.count('📸') > 0 or text.count('😍') > 0:
+            return "Visually appealing"
+        elif "?" in text or "what do you think" in text_lower:
+            return "Controversial"
+        else:
+            return "Storytelling"
+
+    def _determine_tone(self, text: str, sentiment: Dict) -> List[str]:
+        """Determine post tone"""
+        tones = []
+        text_lower = text.lower()
+        
+        if sentiment["label"] == "POSITIVE":
+            tones.append("Optimistic")
+        else:
+            tones.append("Pessimistic")
+            
+        if "?" in text:
+            tones.append("Curious")
+        if "!" in text:
+            tones.append("Assertive")
+        if any(word in text_lower for word in ["learn", "guide", "how to", "tips"]):
+            tones.append("Informative")
+        if any(emoji in text for emoji in ["😂", "😆", "🤣"]):
+            tones.append("Entertaining")
+            
+        return tones
+
+    def _calculate_rating(self, text: str) -> int:
+        """Calculate content rating"""
+        text_lower = text.lower()
+        # Simple rating system based on content markers
+        rating = 1  # Start with family-friendly
+        
+        sensitive_words = ["damn", "hell", "stupid", "idiot"]
+        very_sensitive_words = ["fuck", "shit", "ass"]
+        adult_themes = ["drunk", "sex", "nsfw", "18+"]
+        
+        if any(word in text_lower for word in sensitive_words):
+            rating = max(rating, 2)
+        if any(word in text_lower for word in very_sensitive_words):
+            rating = max(rating, 4)
+        if any(word in text_lower for word in adult_themes):
+            rating = max(rating, 5)
+            
+        return rating
+
+def main():
+    # Set page config
+    st.set_page_config(
+        page_title="Project Graicie - Social Media Content Analyzer",
+        page_icon="📱",
+        layout="wide"
+    )
+    
+    # Title and description
+    st.title("🤖 Project Graicie")
+    st.markdown("""
+    ### AI-Powered Social Media Content Analysis
+    Analyze your social media content to understand its style, tone, and appropriateness. 
+    Get insights to improve your social media presence!
+    """)
+    
+    # Initialize analyzer
+    analyzer = GraicieAnalyzer()
+    
+    # Create columns for better layout
+    col1, col2 = st.columns([2, 1])
+    
+    with col1:
+        # Input area
+        st.subheader("📝 Content Analysis")
+        post_text = st.text_area(
+            "Enter your social media post here:",
+            height=150,
+            placeholder="Type or paste your post content here..."
+        )
+        
+        if st.button("🔍 Analyze Content", use_container_width=True):
+            if post_text:
+                with st.spinner("Analyzing your content..."):
+                    # Get analysis
+                    result = analyzer.analyze_post(post_text)
+                    
+                    # Display results in an organized way
+                    st.success("Analysis Complete!")
+                    
+                    # Create three columns for results
+                    res_col1, res_col2, res_col3 = st.columns(3)
+                    
+                    with res_col1:
+                        st.metric("Style", result["style"])
+                        
+                    with res_col2:
+                        st.metric("Sentiment", f"{result['sentiment']} ({result['confidence']})")
+                        
+                    with res_col3:
+                        st.metric("Content Rating", f"{result['rating']}/5")
+                    
+                    # Show tones
+                    st.subheader("📊 Detected Tones")
+                    tone_html = " ".join([f'<span style="background-color: #e6f3ff; padding: 5px 10px; margin: 5px; border-radius: 15px;">{tone}</span>' for tone in result["tones"]])
+                    st.markdown(f"<div style='margin-top: 10px;'>{tone_html}</div>", unsafe_allow_html=True)
+                    
+            else:
+                st.warning("Please enter some content to analyze!")
+    
+    with col2:
+        # Tips and information
+        st.subheader("💡 Tips")
+        st.info("""
+        **For better results:**
+        - Write naturally
+        - Include relevant emojis
+        - Add hashtags if relevant
+        - Keep your audience in mind
+        """)
+        
+        # Rating guide
+        st.subheader("📋 Rating Guide")
+        st.markdown("""
+        1️⃣ Family-friendly  
+        2️⃣ Minor concerns  
+        3️⃣ Teen-appropriate  
+        4️⃣ Mature content  
+        5️⃣ Adult/Sensitive content
+        """)
+
+if __name__ == "__main__":
+    main()