Update app.py

app.py

@@ -43,18 +43,9 @@ def scrape_comments(video_id):
     return comments_df
 
 
-# Function to extract video ID from YouTube URL
-def extract_video_id(video_url):
-    match = re.search(r'(?<=v=)[\w-]+', video_url)
-    if match:
-        return match.group(0)
-    else:
-        st.error("Invalid YouTube video URL")
-
-# Function to fetch YouTube comments for a video ID
-def fetch_comments(video_id):
-    # Example using youtube-comment-scraper-python library
-    comments = scrape_comments(video_id)
+# Function to fetch video metadata using YouTube API
+def fetch_video_info(video_id):
+    video_id = extract_video_id(video_id)
 
     request = youtube.videos().list(
         part="snippet",
@@ -65,12 +56,27 @@ def fetch_comments(video_id):
     if response['items']:
         video_info = response['items'][0]['snippet']
         channel_name = video_info['channelTitle']
-        video_description = video_info['description']
+        video_title = video_info['title']
+        return channel_name, video_title
+    
+    else:
+        raise ValueError("Video not found")
+
+        
+# Function to extract video ID from YouTube URL
+def extract_video_id(video_url):
+    match = re.search(r'(?<=v=)[\w-]+', video_url)
+    if match:
+        return match.group(0)
     else:
-        channel_name = "Unknown"
-        video_description = "No description available"
+        st.error("Invalid YouTube video URL")
 
-    return comments, channel_name, video_description
+# Function to fetch YouTube comments for a video ID
+def fetch_comments(video_id):
+    # Example using youtube-comment-scraper-python library
+    comments_df = scrape_comments(video_id)
+
+    return comments_df
 
 # Function to analyze sentiment for a single comment
 def analyze_sentiment(comment):
@@ -153,8 +159,10 @@ def main():
             comments_df['sentiment'] = comments_df['comment'].apply(lambda x: analyze_sentiment(x[:512]))
             sentiment_counts = comments_df['sentiment'].value_counts()
 
+            channel_name, video_title = fetch_video_info(video_id)
+            
             st.write(f"**Channel Name:** {channel_name}")
-            st.write(f"**Video Description:** {video_description}")
+            st.write(f"**Video Description:** {video_title}")
             
             st.write("Based on top :100: comments from this video")
             # Create pie chart