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 fpdf2 import FPDF
+from fpdf2 import FPDF   # Correct import
 from youtube_comment_downloader import YoutubeCommentDownloader
 import re
 import os
@@ -79,15 +79,6 @@ def fetch_youtube_comments(url, limit=120):
         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)
@@ -180,7 +171,7 @@ with gr.Blocks(title="CommentSurvey AI", theme=gr.themes.Soft()) as demo:
             if not url or not url.strip():
                 return None, "**❌ Please paste a YouTube URL**", None, None, None, None
 
-            comments = fetch_comments_from_url(url)
+            comments = fetch_youtube_comments(url)
 
             if len(comments) == 0:
                 return None, "**❌ Could not fetch comments.** Try a different public YouTube video.", None, None, None, None