Update app.py

app.py

@@ -444,33 +444,91 @@ from googleapiclient.discovery import build
 
 
 #############################
-import gradio as gr
-from youtube_transcript_api import YouTubeTranscriptApi, TranscriptsDisabled, NoTranscriptFound
-from textblob import TextBlob
 import requests
-import re
+from textblob import TextBlob
+from youtube_transcript_api import YouTubeTranscriptApi, TranscriptsDisabled, NoTranscriptFound
+import gradio as gr
 
 # Helper Functions
 def extract_video_id(url):
     """
-    Extracts the video ID from a YouTube URL.
+    Extract the video ID from a YouTube URL.
     """
-    pattern = r"(?:v=|\/)([0-9A-Za-z_-]{11}).*"
-    match = re.search(pattern, url)
+    import re
+    match = re.search(r"(?:v=|\/)([0-9A-Za-z_-]{11})", url)
     return match.group(1) if match else None
 
 def clean_text_for_analysis(text):
     """
-    Cleans text for sentiment analysis.
+    Clean text for sentiment analysis.
     """
-    return re.sub(r"[^a-zA-Z0-9 .,!?']", "", text)
+    import re
+    return re.sub(r"[^a-zA-Z0-9\s]", "", text)
+
+def extract_subtitle_info(cleaned_text):
+    """
+    Extract additional information from subtitles (e.g., keywords, summary).
+    """
+    from collections import Counter
+    words = cleaned_text.split()
+    common_words = Counter(words).most_common(5)
+    return f"Most common words: {', '.join([word for word, _ in common_words])}"
 
 def get_video_metadata(video_id):
     """
-    Fetches video metadata using YouTube Data API.
+    Fetch metadata for a YouTube video using the YouTube Data API.
+    """
+    YOUTUBE_API_KEY = "YOUR_YOUTUBE_API_KEY"  # Replace with your YouTube Data API key
+    try:
+        response = requests.get(
+            f"https://www.googleapis.com/youtube/v3/videos",
+            params={
+                "part": "snippet",
+                "id": video_id,
+                "key": YOUTUBE_API_KEY
+            }
+        ).json()
+        if "items" in response and response["items"]:
+            snippet = response["items"][0]["snippet"]
+            return {
+                "title": snippet.get("title", "N/A"),
+                "description": snippet.get("description", "N/A"),
+                "channelTitle": snippet.get("channelTitle", "N/A")
+            }
+        return {}
+    except Exception as e:
+        return {"error": str(e)}
+
+def get_recommendations(keywords, max_results=5):
+    """
+    Fetches related video recommendations based on the provided keywords.
     """
-    # Dummy implementation for metadata fetching.
-    return {"description": "No metadata available."}
+    YOUTUBE_API_KEY = "YOUR_YOUTUBE_API_KEY"  # Replace with your YouTube Data API key
+    if not keywords.strip():
+        return "Please provide search keywords"
+    try:
+        response = requests.get(
+            "https://www.googleapis.com/youtube/v3/search",
+            params={
+                "part": "snippet",
+                "q": f"educational {keywords}",
+                "type": "video",
+                "maxResults": max_results,
+                "relevanceLanguage": "en",
+                "key": YOUTUBE_API_KEY
+            }
+        ).json()
+
+        results = []
+        for item in response.get("items", []):
+            title = item["snippet"]["title"]
+            channel = item["snippet"]["channelTitle"]
+            video_id = item["id"]["videoId"]
+            results.append(f"\ud83d\udcfa {title}\n\ud83d\udc64 {channel}\n\ud83d\udd17 https://youtube.com/watch?v={video_id}\n")
+
+        return "\n".join(results) if results else "No recommendations found"
+    except Exception as e:
+        return f"Error: {str(e)}"
 
 def process_youtube_video(url="", keywords=""):
     try:
@@ -485,7 +543,7 @@ def process_youtube_video(url="", keywords=""):
         if url.strip():
             video_id = extract_video_id(url)
             if not video_id:
-                return None, "Invalid YouTube URL", "N/A", ""
+                return None, "Invalid YouTube URL", "N/A", "", ""
 
             thumbnail = f"https://img.youtube.com/vi/{video_id}/maxresdefault.jpg"
 
@@ -506,27 +564,27 @@ def process_youtube_video(url="", keywords=""):
                 cleaned_text = clean_text_for_analysis(text)
 
                 # Sentiment analysis
-                sentiment = TextBlob(cleaned_text).sentiment
+                sentiment = TextBlob(cleaned_text).sentiment  # Use cleaned text for sentiment analysis
                 sentiment_label = f"{'Positive' if sentiment.polarity > 0 else 'Negative' if sentiment.polarity < 0 else 'Neutral'} ({sentiment.polarity:.2f})"
 
-                # Generate summary (Dummy implementation)
-                summary = f"Summary: {cleaned_text[:200]}..."
+                # Generate summary
+                summary = f"Summary: {cleaned_text[:400]}..."
 
-                # Extract subtitle information (Dummy implementation)
-                subtitle_info = "Subtitle processed successfully."
+                # Extract subtitle information
+                subtitle_info = extract_subtitle_info(cleaned_text)
 
             except TranscriptsDisabled:
                 metadata = get_video_metadata(video_id)
-                summary = metadata.get("description", "⚠️ This video has disabled subtitles.")
+                summary = metadata.get("description", "\u26a0\ufe0f This video has disabled subtitles.")
                 sentiment_label = "N/A"
                 subtitle_info = "No subtitles available for analysis."
             except NoTranscriptFound:
                 metadata = get_video_metadata(video_id)
-                summary = metadata.get("description", "⚠️ No English transcript available.")
+                summary = metadata.get("description", "\u26a0\ufe0f No English transcript available.")
                 sentiment_label = "N/A"
                 subtitle_info = "No subtitles available for analysis."
             except Exception as e:
-                return thumbnail, f"⚠️ Error processing transcript: {str(e)}", "N/A", ""
+                return thumbnail, f"\u26a0\ufe0f Error processing transcript: {str(e)}", "N/A", "", ""
 
         # If keywords are provided, get recommendations
         if keywords.strip():
@@ -535,44 +593,18 @@ def process_youtube_video(url="", keywords=""):
         return thumbnail, summary, sentiment_label, subtitle_info, recommendations
 
     except Exception as e:
-        return None, f"Error: {str(e)}", "N/A", ""
-
-def get_recommendations(keywords, max_results=5):
-    """
-    Fetches related video recommendations based on the provided keywords.
-    """
-    YOUTUBE_API_KEY = "AIzaSyD_SDF4lC3vpHVAMnBOcN2ZCTz7dRjUc98"  # Replace with your YouTube Data API key
-    if not keywords.strip():
-        return "Please provide search keywords"
-    try:
-        response = requests.get(
-            "https://www.googleapis.com/youtube/v3/search",
-            params={
-                "part": "snippet",
-                "q": f"educational {keywords}",
-                "type": "video",
-                "maxResults": max_results,
-                "relevanceLanguage": "en",
-                "key": YOUTUBE_API_KEY
-            }
-        ).json()
+        return None, f"Error: {str(e)}", "N/A", "", ""
 
-        results = []
-        for item in response.get("items", []):
-            title = item["snippet"]["title"]
-            channel = item["snippet"]["channelTitle"]
-            video_id = item["id"]["videoId"]
-            results.append(f"📺 {title}\n👤 {channel}\n🔗 https://youtube.com/watch?v={video_id}\n")
-
-        return "\n".join(results) if results else "No recommendations found"
-    except Exception as e:
-        return f"Error: {str(e)}"
+# User credentials (replace with a secure method in production)
+USER_CREDENTIALS = {
+    "admin": "admin"
+}
 
 # Gradio Interface
 with gr.Blocks(theme=gr.themes.Soft()) as app:
     # Login Page
     with gr.Group() as login_page:
-        gr.Markdown("# 🎓 Educational Learning Management System")
+        gr.Markdown("# \ud83c\udf93 Educational Learning Management System")
         username = gr.Textbox(label="Username")
         password = gr.Textbox(label="Password", type="password")
         login_btn = gr.Button("Login", variant="primary")
@@ -582,48 +614,40 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     with gr.Group(visible=False) as main_page:
         with gr.Row():
             with gr.Column(scale=1):
-                gr.Markdown("### 📋 Navigation")
-                nav_dashboard = gr.Button("📊 Dashboard", variant="primary")
-                nav_students = gr.Button("👥 Students")
-                nav_teachers = gr.Button("👨‍🏫 Teachers")
-                nav_courses = gr.Button("📚 Courses")
-                nav_youtube = gr.Button("🎥 YouTube Tool")
-                logout_btn = gr.Button("🚪 Logout", variant="stop")
+                gr.Markdown("### \ud83d\udccb Navigation")
+                nav_dashboard = gr.Button("\ud83d\udcca Dashboard", variant="primary")
+                nav_youtube = gr.Button("\ud83c\udfa5 YouTube Tool")
+                logout_btn = gr.Button("\ud83d\udeaa Logout", variant="stop")
 
             with gr.Column(scale=3):
-                # Dashboard Content
-                dashboard_page = gr.Group()
-                with dashboard_page:
-                    gr.Markdown("## 📊 Dashboard")
-
                 # YouTube Tool Content
                 youtube_page = gr.Group(visible=False)
                 with youtube_page:
                     gr.Markdown("## Agent for YouTube Content Exploration")
                     with gr.Row():
-                        video_url = gr.Textbox(
-                            label="YouTube URL",
-                            placeholder="https://youtube.com/watch?v=..."
-                        )
-                        analyze_btn = gr.Button("🔍 Analyze Video", variant="primary")
+                        with gr.Column(scale=2):
+                            video_url = gr.Textbox(
+                                label="YouTube URL",
+                                placeholder="https://youtube.com/watch?v=..."
+                            )
+                            keywords = gr.Textbox(
+                                label="Keywords for Recommendations",
+                                placeholder="e.g., python programming, machine learning"
+                            )
+                            analyze_btn = gr.Button("\ud83d\udd0d Analyze Video", variant="primary")
+                            recommend_btn = gr.Button("\ud83d\udd0d Get Recommendations", variant="secondary")
+
+                        with gr.Column(scale=1):
+                            video_thumbnail = gr.Image(label="Video Preview")
 
                     with gr.Row():
-                        keywords = gr.Textbox(
-                            label="Keywords for Recommendations",
-                            placeholder="e.g., python programming, machine learning"
-                        )
-                        recommend_btn = gr.Button("🔍 Get Recommendations", variant="primary")
-
-                    with gr.Row():
-                        video_thumbnail = gr.Image(label="Video Preview")
-
-                    with gr.Row():
-                        summary = gr.Textbox(label="📝 Summary", lines=8)
-                        sentiment = gr.Textbox(label="😊 Content Sentiment")
-                        recommendations = gr.Textbox(label="🎯 Related Videos", lines=10)
+                        with gr.Column():
+                            summary = gr.Textbox(label="\ud83d\udd8b\ufe0f Summary", lines=8)
+                            sentiment = gr.Textbox(label="\ud83d\ude0a Content Sentiment")
+                        with gr.Column():
+                            recommendations = gr.Textbox(label="\ud83c\udfaf Related Videos", lines=10)
 
     def login_check(user, pwd):
-        USER_CREDENTIALS = {"admin": "admin"}
         if USER_CREDENTIALS.get(user) == pwd:
             return {
                 login_page: gr.update(visible=False),
@@ -633,12 +657,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
         return {
             login_page: gr.update(visible=True),
             main_page: gr.update(visible=False),
-            login_msg: "❌ Invalid credentials"
+            login_msg: "\u274c Invalid credentials"
         }
 
     def show_page(page_name):
         updates = {
-            dashboard_page: gr.update(visible=False),
             youtube_page: gr.update(visible=False)
         }
         updates[page_name] = gr.update(visible=True)
@@ -651,19 +674,18 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
         outputs=[login_page, main_page, login_msg]
     )
 
-    nav_dashboard.click(lambda: show_page(dashboard_page), outputs=list(show_page(dashboard_page).keys()))
     nav_youtube.click(lambda: show_page(youtube_page), outputs=list(show_page(youtube_page).keys()))
 
     analyze_btn.click(
         process_youtube_video,
-        inputs=[video_url, gr.Textbox(label="")],  # Empty input for keywords
+        inputs=[video_url, ""],
         outputs=[video_thumbnail, summary, sentiment, recommendations]
     )
 
     recommend_btn.click(
-        lambda keywords: (None, "", "", get_recommendations(keywords)),
+        lambda keywords: process_youtube_video("", keywords)[-1],
         inputs=[keywords],
-        outputs=[video_thumbnail, summary, sentiment, recommendations]
+        outputs=[recommendations]
     )
 
     logout_btn.click(
@@ -675,7 +697,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     )
 
 if __name__ == "__main__":
-    app.launch()
-
-
-
+    app.launch()