Update app.py

app.py

@@ -134,6 +134,8 @@ from collections import Counter
 from googleapiclient.discovery import build
 
 
+YOUTUBE_API_KEY = "AIzaSyD_SDF4lC3vpHVAMnBOcN2ZCTz7dRjUc98"  # Replace with your actual API key
+
 def extract_video_id(url):
     match = re.search(r"(?:v=|\/)([0-9A-Za-z_-]{11})", url)
     return match.group(1) if match else None
@@ -164,10 +166,14 @@ def extract_subtitle_info(text):
         words = text.split()
         common_words = Counter(words).most_common(10)
         key_topics = ", ".join([word for word, count in common_words])
-        info = f"Key topics discussed: {key_topics}. \nNumber of sentences: {len(sentences)}. \nTotal words: {len(words)}."
+        info = {
+            "key_topics": key_topics,
+            "sentence_count": len(sentences),
+            "word_count": len(words),
+        }
         return info
     except Exception as e:
-        return f"Error extracting subtitle information: {str(e)}"
+        return {"error": str(e)}
 
 def get_recommendations(keywords, max_results=5):
     if not keywords:
@@ -201,11 +207,12 @@ def process_youtube_video(url, keywords):
         thumbnail = None
         summary = "No transcript available"
         sentiment_label = "N/A"
+        subtitle_info = {}
         recommendations = ""
 
         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"
 
@@ -222,6 +229,7 @@ def process_youtube_video(url, keywords):
                 raise ValueError("Transcript is empty")
 
             cleaned_text = clean_text_for_analysis(text)
+            subtitle_info = extract_subtitle_info(cleaned_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})"
@@ -236,38 +244,13 @@ def process_youtube_video(url, keywords):
         if keywords.strip():
             recommendations = get_recommendations(keywords)
 
-        return thumbnail, summary, sentiment_label, recommendations
+        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):
-#     if not keywords:
-#         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"📺 {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)}"
+        return None, f"Error: {str(e)}", "N/A", {}, ""
+
+
+
 
 # Gradio Interface
 with gr.Blocks(theme=gr.themes.Soft()) as app: