Update app.py

app.py

@@ -3,7 +3,7 @@ import json
 import plotly.express as px
 import pandas as pd
 from groq import Groq
-from fpdf import FPDF
+from fpdf2 import FPDF
 from youtube_comment_downloader import YoutubeCommentDownloader
 import re
 import os
@@ -65,7 +65,7 @@ def clean_youtube_url(url):
         return f"https://www.youtube.com/watch?v={video_id}"
     return url
 
-def fetch_youtube_comments(url, limit=150):
+def fetch_youtube_comments(url, limit=120):
     try:
         clean_url = clean_youtube_url(url)
         downloader = YoutubeCommentDownloader()
@@ -76,14 +76,23 @@ def fetch_youtube_comments(url, limit=150):
                 break
         return comments
     except Exception as e:
-        print(f"Fetch error: {e}")
+        print(f"Error fetching YouTube: {e}")
         return []
 
+def fetch_comments_from_url(url):
+    if not url or not url.strip():
+        return []
+    url_lower = url.lower()
+    if any(x in url_lower for x in ["youtube", "youtu.be", "pie.yt"]):
+        return fetch_youtube_comments(url)
+    else:
+        return ["[Please use a valid YouTube link]"]
+
 def analyze_comments_with_groq(comments, post_context=""):
     try:
         client = Groq(api_key=GROQ_API_KEY)
         comments_text = "\n\n".join([f"Comment {i+1}: {c[:250]}" for i, c in enumerate(comments[:130])])
-        
+
         user_prompt = f"""
 Post Context: {post_context}
 Analyze these comments and return proper JSON:
@@ -119,19 +128,19 @@ def create_pdf_report(analysis_result, poll_question):
     pdf.set_font('Arial', 'B', 12)
     pdf.cell(0, 10, 'Summary:', 0, 1, 'L')
     pdf.set_font('Arial', '', 11)
-    pdf.multi_cell(0, 6, analysis_result.get('summary', 'No summary available.'))
+    pdf.multi_cell(0, 5, analysis_result.get('summary', 'No summary available.'))
     pdf.ln(10)
 
     pdf.set_font('Arial', 'B', 12)
     pdf.cell(0, 10, 'Top Themes:', 0, 1, 'L')
     pdf.set_font('Arial', '', 11)
-    for theme in analysis_result.get('top_themes', [])[:8]:
-        pdf.cell(0, 6, f"• {theme}", 0, 1, 'L')
+    for theme in analysis_result.get('top_themes', []):
+        pdf.cell(0, 7, f'- {theme}', 0, 1, 'L')
 
     pdf.output("CommentSurvey_Report.pdf")
     return "CommentSurvey_Report.pdf"
 
-# ====================== GRADIO INTERFACE ======================
+# ====================== GRADIO APP ======================
 with gr.Blocks(title="CommentSurvey AI", theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 📊 CommentSurvey AI\n**Turn YouTube Comments into Smart Surveys & Insights**")
 
@@ -139,7 +148,7 @@ with gr.Blocks(title="CommentSurvey AI", theme=gr.themes.Soft()) as demo:
         with gr.Column(scale=4):
             url_input = gr.Textbox(
                 label="🌐 Paste YouTube Video Link",
-                placeholder="https://youtu.be/xxxxxxxxxx",
+                placeholder="https://youtu.be/dAAzqMSknkI",
                 lines=1
             )
             analyze_btn = gr.Button("🚀 Analyze Comments", variant="primary", size="large")
@@ -168,9 +177,13 @@ with gr.Blocks(title="CommentSurvey AI", theme=gr.themes.Soft()) as demo:
             if not GROQ_API_KEY:
                 return None, "**❌ Groq API Key is missing!**", None, None, None, None
 
-            comments = fetch_youtube_comments(url)
+            if not url or not url.strip():
+                return None, "**❌ Please paste a YouTube URL**", None, None, None, None
+
+            comments = fetch_comments_from_url(url)
+
             if len(comments) == 0:
-                return None, "**❌ Could not fetch comments.** Try a different public video.", None, None, None, None
+                return None, "**❌ Could not fetch comments.** Try a different public YouTube video.", None, None, None, None
 
             result = analyze_comments_with_groq(comments)