Update app.py

app.py

@@ -7,13 +7,12 @@ from textblob import TextBlob
 sentiment_analyzer = pipeline("sentiment-analysis")
 
 def analyze_youtube_content(youtube_url, transcript_text=""):
-    """Main function to analyze YouTube content"""
+    """Analyze YouTube content"""
     results = {}
     
-    # If URL is provided, get video info
+    # Get video info
     if youtube_url:
         try:
-            # Create a YouTube object
             yt = pytube.YouTube(youtube_url)
             results["video_info"] = {
                 "title": yt.title,
@@ -25,24 +24,19 @@ def analyze_youtube_content(youtube_url, transcript_text=""):
                 "message": str(e)
             }
     
-    # If transcript is provided, analyze it
+    # Analyze transcript
     if transcript_text:
-        # Analyze sentiment with TextBlob
+        # TextBlob sentiment
         blob = TextBlob(transcript_text)
-        textblob_sentiment = blob.sentiment
+        sentiment = blob.sentiment
         
-        # Analyze sentiment with Hugging Face
+        # Hugging Face sentiment
         hf_result = sentiment_analyzer(transcript_text[:512])[0]
         
         results["sentiment"] = {
-            "textblob": {
-                "polarity": round(textblob_sentiment.polarity, 2),
-                "assessment": "positive" if textblob_sentiment.polarity > 0 else "negative" if textblob_sentiment.polarity < 0 else "neutral"
-            },
-            "huggingface": {
-                "label": hf_result["label"],
-                "score": round(hf_result["score"], 4)
-            }
+            "polarity": round(sentiment.polarity, 2),
+            "assessment": "positive" if sentiment.polarity > 0 else "negative" if sentiment.polarity < 0 else "neutral",
+            "huggingface": hf_result["label"]
         }
     
     return results
@@ -59,6 +53,6 @@ demo = gr.Interface(
     description="Analyze viral moments from YouTube videos using ML models"
 )
 
-# Launch the app with MCP server enabled
+# Launch with MCP server enabled
 if __name__ == "__main__":
-    demo.launch(mcp_server=True)
+    demo.launch(mcp_server=True)