Update app.py

app.py

@@ -11,43 +11,19 @@ st.set_page_config(
     initial_sidebar_state="expanded"
 )
 
-# Custom CSS styling
-st.markdown(r"""
+# Custom CSS
+st.markdown("""
 <style>
-    .header {
-        font-size: 2.5em !important;
-        color: #FF4B4B !important;
-        margin-bottom: 30px !important;
-    }
-    .sidebar .sidebar-content {
-        background-color: #F0F2F6;
-    }
-    .stProgress > div > div > div > div {
-        background-color: #FF4B4B;
-    }
-    .day-card {
-        padding: 20px;
-        border-radius: 10px;
-        box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-        margin-bottom: 20px;
-        background-color: white;
-    }
-    body {
-        background-image: url('https://images.unsplash.com/photo-7NT4EDSI5Ok');
-        background-size: cover;
-        background-position: center;
-        background-repeat: no-repeat;
-        background-attachment: fixed;
-    }
-    .main > div {
-        background-color: rgba(255, 255, 255, 0.9);
-        padding: 20px;
-        border-radius: 10px;
-    }
+    .header { font-size: 2.5em !important; color: #FF4B4B !important; margin-bottom: 30px !important; }
+    .sidebar .sidebar-content { background-color: #F0F2F6; }
+    .stProgress > div > div > div > div { background-color: #FF4B4B; }
+    .day-card { padding: 20px; border-radius: 10px; box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1); margin-bottom: 20px; background-color: white; }
+    body { background-size: cover; background-position: center; background-repeat: no-repeat; background-attachment: fixed; }
+    .main > div { background-color: rgba(255, 255, 255, 0.9); padding: 20px; border-radius: 10px; }
 </style>
 """, unsafe_allow_html=True)
 
-# Initialize session state
+# Session state for content plan
 if 'content_plan' not in st.session_state:
     st.session_state.content_plan = None
 
@@ -61,41 +37,46 @@ with st.sidebar:
 
 # Main content
 st.markdown('<div class="header">🎥 YouTube Content Strategist</div>', unsafe_allow_html=True)
-st.write("Generate data-driven content plans using YouTube audience insights")
+st.write("Generate data-driven content plans using YouTube audience insights.")
 
 # Helper functions
-def get_channel_id(channel_url):
-    """Extract channel ID from URL."""
+def get_actual_channel_id(youtube, channel_url):
+    """Extract real channel ID from a handle (@username) or custom URL."""
     if "@" in channel_url:
-        return channel_url.split("@")[-1]
+        username = channel_url.split("@")[-1]
+        request = youtube.channels().list(part="id", forHandle=username)
     elif "channel/" in channel_url:
         return channel_url.split("channel/")[-1]
+    else:
+        return None
+    
+    response = request.execute()
+    if "items" in response and response["items"]:
+        return response["items"][0]["id"]
     return None
 
-def get_channel_videos(channel_id):
+def get_channel_videos(youtube, channel_id):
     """Fetch recent video IDs from the channel."""
-    youtube = googleapiclient.discovery.build("youtube", "v3", developerKey=YOUTUBE_API_KEY)
     request = youtube.search().list(
         part="id",
         channelId=channel_id,
-        maxResults=MAX_VIDEOS,
+        maxResults=min(MAX_VIDEOS, 50),  # Ensure within API limit
         order="date",
         type="video"
     )
     response = request.execute()
-    return [item['id']['videoId'] for item in response['items']]
+    return [item['id']['videoId'] for item in response.get("items", [])]
 
-def get_video_comments(video_id):
+def get_video_comments(youtube, video_id):
     """Fetch comments from a video."""
-    youtube = googleapiclient.discovery.build("youtube", "v3", developerKey=YOUTUBE_API_KEY)
     request = youtube.commentThreads().list(
         part="snippet",
         videoId=video_id,
-        maxResults=COMMENTS_PER_VIDEO,
+        maxResults=min(COMMENTS_PER_VIDEO, 100),  # Ensure within API limit
         textFormat="plainText"
     )
     response = request.execute()
-    return [item['snippet']['topLevelComment']['snippet']['textDisplay'] for item in response['items']]
+    return [item['snippet']['topLevelComment']['snippet']['textDisplay'] for item in response.get("items", [])]
 
 def analyze_comments(comments):
     """Analyze comments using Gemini API and generate a content plan."""
@@ -115,7 +96,7 @@ def analyze_comments(comments):
     response = model.generate_content(prompt)
     return response.text
 
-# Analysis execution
+# Get channel URL
 channel_url = st.text_input("Enter YouTube Channel URL:", placeholder="https://www.youtube.com/@ChannelName")
 
 if st.button("Generate Content Plan 🚀"):
@@ -123,29 +104,24 @@ if st.button("Generate Content Plan 🚀"):
         st.error("Please fill all required fields!")
     else:
         try:
-            # Initialize APIs
             youtube = googleapiclient.discovery.build("youtube", "v3", developerKey=YOUTUBE_API_KEY)
             genai.configure(api_key=GEMINI_API_KEY)
-            model = genai.GenerativeModel('gemini-1.5-pro')
 
-            # Get channel data
             with st.spinner("🔍 Analyzing channel..."):
-                channel_id = get_channel_id(channel_url)
+                channel_id = get_actual_channel_id(youtube, channel_url)
                 if not channel_id:
-                    st.error("Invalid channel URL")
+                    st.error("Invalid channel URL!")
                     st.stop()
+                
+                video_ids = get_channel_videos(youtube, channel_id)
 
-                video_ids = get_channel_videos(channel_id)
-            
-            # Get comments
             all_comments = []
             progress_bar = st.progress(0)
             for idx, video_id in enumerate(video_ids):
                 progress = (idx + 1) / len(video_ids)
                 progress_bar.progress(progress, text=f"📥 Collecting comments from video {idx+1}/{len(video_ids)}...")
-                all_comments.extend(get_video_comments(video_id))
-            
-            # Generate content plan
+                all_comments.extend(get_video_comments(youtube, video_id))
+
             with st.spinner("🧠 Analyzing comments and generating plan..."):
                 content_plan = analyze_comments(all_comments)
                 st.session_state.content_plan = content_plan
@@ -158,31 +134,24 @@ if st.button("Generate Content Plan 🚀"):
 if st.session_state.content_plan:
     st.markdown("## 📅 10-Day Content Plan")
     st.success("Here's your personalized content strategy based on audience insights!")
-    
-    # Create date cards
+
     start_date = datetime.now()
     for day in range(10):
         current_date = start_date + timedelta(days=day)
-        with st.expander(f"Day {day+1} - {current_date.strftime('%b %d')}", expanded=True if day==0 else False):
-            # Use .split('\n') instead of f-string splitting
-            plan_lines = st.session_state.content_plan.split('\n')
-            
-            # Calculate start index, accounting for potential inconsistent line counts
+        with st.expander(f"Day {day+1} - {current_date.strftime('%b %d')}", expanded=True if day == 0 else False):
+            plan_lines = st.session_state.content_plan.split("\n")
             start_index = day * 5
             if start_index + 4 < len(plan_lines):
-               st.markdown(f"""
-               
-                            <div class="day-card">
-                                <h3>{plan_lines[start_index] if start_index < len(plan_lines) else 'No Topic'}</h3>
-                                <p>🎯 Objective: {plan_lines[start_index+1] if start_index+1 < len(plan_lines) else 'N/A'}</p>
-                                <p>📹 Format: {plan_lines[start_index+2] if start_index+2 < len(plan_lines) else 'N/A'}</p>
-                                <p>💡 Engagement Strategy: {plan_lines[start_index+3] if start_index+3 < len(plan_lines) else 'N/A'}</p>
-                                <p>📈 Success Metrics: {plan_lines[start_index+4] if start_index+4 < len(plan_lines) else 'N/A'}</p>
-                            </div>
-                            """, unsafe_allow_html=True)
+                st.markdown(f"""
+                <div class="day-card">
+                    <h3>{plan_lines[start_index] if start_index < len(plan_lines) else 'No Topic'}</h3>
+                    <p>🎯 Objective: {plan_lines[start_index+1] if start_index+1 < len(plan_lines) else 'N/A'}</p>
+                    <p>📹 Format: {plan_lines[start_index+2] if start_index+2 < len(plan_lines) else 'N/A'}</p>
+                    <p>💡 Engagement Strategy: {plan_lines[start_index+3] if start_index+3 < len(plan_lines) else 'N/A'}</p>
+                    <p>📈 Success Metrics: {plan_lines[start_index+4] if start_index+4 < len(plan_lines) else 'N/A'}</p>
+                </div>
+                """, unsafe_allow_html=True)
 
-    
-    # Download button
     st.download_button(
         label="📥 Download Full Plan",
         data=st.session_state.content_plan,
@@ -190,22 +159,14 @@ if st.session_state.content_plan:
         mime="text/plain"
     )
 
-# Instructions
 with st.expander("ℹ️ How to use this tool"):
     st.markdown("""
     1. **Get API Keys**:
        - YouTube Data API: [Get it here](https://console.cloud.google.com/)
        - Gemini API: [Get it here](https://makersuite.google.com/)
-    
     2. **Enter Channel URL**:
-       - Supports both channel IDs (@ChannelName) and custom URLs
-    
     3. **Adjust Settings**:
-       - Control number of videos/comments analyzed in sidebar
-    
     4. **Generate Plan**:
-       - Click the rocket button to create your strategy
-    
-    5. **Implement & Track**:
-       - Download the plan and track performance in YouTube Analytics
-    """)
+    5. **Download & Track Results**.
+    """)
+