Update app.py

app.py

@@ -451,34 +451,22 @@ import gradio as gr
 
 # Helper Functions
 def extract_video_id(url):
-    """
-    Extract the video ID from a YouTube 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):
-    """
-    Clean text for sentiment analysis.
-    """
     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):
-    """
-    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
+    YOUTUBE_API_KEY = "YOUR_YOUTUBE_API_KEY"
     try:
         response = requests.get(
             f"https://www.googleapis.com/youtube/v3/videos",
@@ -500,10 +488,7 @@ def get_video_metadata(video_id):
         return {"error": str(e)}
 
 def get_recommendations(keywords, max_results=5):
-    """
-    Fetches related video recommendations based on the provided keywords.
-    """
-    YOUTUBE_API_KEY = "YOUR_YOUTUBE_API_KEY"  # Replace with your YouTube Data API key
+    YOUTUBE_API_KEY = "YOUR_YOUTUBE_API_KEY"
     if not keywords.strip():
         return "Please provide search keywords"
     try:
@@ -524,177 +509,97 @@ def get_recommendations(keywords, max_results=5):
             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")
+            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)}"
 
-def process_youtube_video(url="", keywords=""):
+def process_youtube_video(url):
     try:
-        # Initialize variables
-        thumbnail = None
-        summary = "No transcript available"
-        sentiment_label = "N/A"
-        recommendations = ""
-        subtitle_info = "No additional information available"
+        video_id = extract_video_id(url)
+        if not video_id:
+            return None, "Invalid YouTube URL", "N/A", "", ""
 
-        # If URL is provided, process the YouTube video
-        if url.strip():
-            video_id = extract_video_id(url)
-            if not video_id:
-                return None, "Invalid YouTube URL", "N/A", "", ""
-
-            thumbnail = f"https://img.youtube.com/vi/{video_id}/maxresdefault.jpg"
+        thumbnail = f"https://img.youtube.com/vi/{video_id}/maxresdefault.jpg"
 
+        try:
+            transcript_list = YouTubeTranscriptApi.list_transcripts(video_id)
+            transcript = None
             try:
-                # Fetch transcript
-                transcript_list = YouTubeTranscriptApi.list_transcripts(video_id)
-                transcript = None
-                try:
-                    transcript = transcript_list.find_transcript(['en'])
-                except:
-                    transcript = transcript_list.find_generated_transcript(['en'])
-
-                text = " ".join([t['text'] for t in transcript.fetch()])
-                if not text.strip():
-                    raise ValueError("Transcript is empty")
-
-                # Clean up the text for sentiment analysis
-                cleaned_text = clean_text_for_analysis(text)
-
-                # Sentiment analysis
-                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
-                summary = f"Summary: {cleaned_text[:400]}..."
-
-                # Extract subtitle information
-                subtitle_info = extract_subtitle_info(cleaned_text)
-
-            except TranscriptsDisabled:
-                metadata = get_video_metadata(video_id)
-                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", "\u26a0\ufe0f No English transcript available.")
-                sentiment_label = "N/A"
-                subtitle_info = "No subtitles available for analysis."
-            except Exception as e:
-                return thumbnail, f"\u26a0\ufe0f Error processing transcript: {str(e)}", "N/A", "", ""
-
-        # If keywords are provided, get recommendations
-        if keywords.strip():
-            recommendations = get_recommendations(keywords)
-
-        return thumbnail, summary, sentiment_label, subtitle_info, recommendations
-
+                transcript = transcript_list.find_transcript(['en'])
+            except:
+                transcript = transcript_list.find_generated_transcript(['en'])
+
+            text = " ".join([t['text'] for t in transcript.fetch()])
+            cleaned_text = clean_text_for_analysis(text)
+
+            sentiment = TextBlob(cleaned_text).sentiment
+            sentiment_label = f"{'Positive' if sentiment.polarity > 0 else 'Negative' if sentiment.polarity < 0 else 'Neutral'} ({sentiment.polarity:.2f})"
+            summary = f"Summary: {cleaned_text[:400]}..."
+            subtitle_info = extract_subtitle_info(cleaned_text)
+
+            return thumbnail, summary, sentiment_label, subtitle_info
+        except TranscriptsDisabled:
+            metadata = get_video_metadata(video_id)
+            return thumbnail, metadata.get("description", "No transcript available"), "N/A", "No subtitles available"
+        except NoTranscriptFound:
+            metadata = get_video_metadata(video_id)
+            return thumbnail, metadata.get("description", "No transcript available"), "N/A", "No subtitles available"
     except Exception as e:
-        return None, f"Error: {str(e)}", "N/A", "", ""
+        return None, f"Error: {str(e)}", "N/A", ""
+
+def fetch_recommendations(keywords):
+    return get_recommendations(keywords)
 
-# User credentials (replace with a secure method in production)
+# User credentials
 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("# \ud83c\udf93 Educational Learning Management System")
+        gr.Markdown("# 🎓 Educational Learning Management System")
         username = gr.Textbox(label="Username")
         password = gr.Textbox(label="Password", type="password")
         login_btn = gr.Button("Login", variant="primary")
         login_msg = gr.Markdown()
 
-    # Main Interface
     with gr.Group(visible=False) as main_page:
         with gr.Row():
             with gr.Column(scale=1):
-                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")
+                gr.Markdown("### 📋 Navigation")
+                nav_youtube = gr.Button("🎥 YouTube Tool")
+                logout_btn = gr.Button("🚪 Logout", variant="stop")
 
             with gr.Column(scale=3):
-                # YouTube Tool Content
                 youtube_page = gr.Group(visible=False)
                 with youtube_page:
                     gr.Markdown("## Agent for YouTube Content Exploration")
-                    with gr.Row():
-                        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():
-                        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)
+                    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")
+                    analyze_btn = gr.Button("🔍 Analyze Video", variant="primary")
+                    recommend_btn = gr.Button("🔍 Get Recommendations", variant="secondary")
+                    video_thumbnail = gr.Image(label="Video Preview")
+                    summary = gr.Textbox(label="📝 Summary", lines=8)
+                    sentiment = gr.Textbox(label="😊 Content Sentiment")
+                    subtitle_info = gr.Textbox(label="📜 Subtitle Info")
+                    recommendations = gr.Textbox(label="🎯 Related Videos", lines=10)
 
     def login_check(user, pwd):
         if USER_CREDENTIALS.get(user) == pwd:
-            return {
-                login_page: gr.update(visible=False),
-                main_page: gr.update(visible=True),
-                login_msg: ""
-            }
-        return {
-            login_page: gr.update(visible=True),
-            main_page: gr.update(visible=False),
-            login_msg: "\u274c Invalid credentials"
-        }
-
-    def show_page(page_name):
-        updates = {
-            youtube_page: gr.update(visible=False)
-        }
-        updates[page_name] = gr.update(visible=True)
-        return updates
-
-    # Event Handlers
-    login_btn.click(
-        login_check,
-        inputs=[username, password],
-        outputs=[login_page, main_page, login_msg]
-    )
-
-    nav_youtube.click(lambda: show_page(youtube_page), outputs=list(show_page(youtube_page).keys()))
-
-    analyze_btn.click(
-        process_youtube_video,
-        inputs=[video_url, ""],
-        outputs=[video_thumbnail, summary, sentiment, recommendations]
-    )
-
-    recommend_btn.click(
-        lambda keywords: process_youtube_video("", keywords)[-1],
-        inputs=[keywords],
-        outputs=[recommendations]
-    )
-
-    logout_btn.click(
-        lambda: {
-            login_page: gr.update(visible=True),
-            main_page: gr.update(visible=False)
-        },
-        outputs=[login_page, main_page]
-    )
+            return {login_page: gr.update(visible=False), main_page: gr.update(visible=True), login_msg: ""}
+        return {login_page: gr.update(visible=True), main_page: gr.update(visible=False), login_msg: "❌ Invalid credentials"}
+
+    def show_youtube_page():
+        return {youtube_page: gr.update(visible=True)}
+
+    login_btn.click(login_check, inputs=[username, password], outputs=[login_page, main_page, login_msg])
+    nav_youtube.click(show_youtube_page, outputs=[youtube_page])
+    analyze_btn.click(process_youtube_video, inputs=[video_url], outputs=[video_thumbnail, summary, sentiment, subtitle_info])
+    recommend_btn.click(fetch_recommendations, inputs=[keywords], outputs=[recommendations])
+    logout_btn.click(lambda: {login_page: gr.update(visible=True), main_page: gr.update(visible=False)}, outputs=[login_page, main_page])
 
 if __name__ == "__main__":
-    app.launch()
+    app.launch()