Update app.py

app.py

@@ -156,31 +156,52 @@ def analyze_sentiment(text):
     sentiment_scores = sid.polarity_scores(text)
     return sentiment_scores
 
-def main():
-    # Check if a URL is passed as a command-line argument
-    if len(sys.argv) > 1:
-        video_url = sys.argv[1]
-    else:
-        # Use a hardcoded URL for testing if no input is provided
-        video_url = "https://www.youtube.com/watch?v=dQw4w9WgXcQ"  # Replace with a valid URL
+# def main():
+#     # Check if a URL is passed as a command-line argument
+#     if len(sys.argv) > 1:
+#         video_url = sys.argv[1]
+#     else:
+#         # Use a hardcoded URL for testing if no input is provided
+#         video_url = "https://www.youtube.com/watch?v=dQw4w9WgXcQ"  # Replace with a valid URL
 
-    print("Fetching transcript...")
+#     print("Fetching transcript...")
 
+#     try:
+#         transcript = get_youtube_transcript(video_url)
+#         print("Transcript fetched successfully.")
+
+#         print("Summarizing transcript...")
+#         summary = summarize_text(transcript)
+#         print("Summary:\n", summary)
+
+#         print("Analyzing sentiment...")
+#         sentiment = analyze_sentiment(transcript)
+#         print("Sentiment Analysis:\n", sentiment)
+
+#     except Exception as e:
+#         print("An error occurred:", str(e))
+
+def process_youtube_video(video_url, keywords):
+    """Process YouTube video URL and keywords for analysis."""
     try:
+        # Fetch transcript
         transcript = get_youtube_transcript(video_url)
-        print("Transcript fetched successfully.")
-
-        print("Summarizing transcript...")
+        
+        # Summarize transcript
         summary = summarize_text(transcript)
-        print("Summary:\n", summary)
-
-        print("Analyzing sentiment...")
+        
+        # Perform sentiment analysis
         sentiment = analyze_sentiment(transcript)
-        print("Sentiment Analysis:\n", sentiment)
-
+        
+        # Generate recommendations
+        recommendations = get_recommendations(keywords)
+        
+        # Return results
+        return None, summary, sentiment, recommendations
     except Exception as e:
-        print("An error occurred:", str(e))
-
+        # Handle errors and return them in the output fields
+        error_message = f"Error: {str(e)}"
+        return None, error_message, error_message, error_message
 
 
 def get_recommendations(keywords, max_results=5):
@@ -340,10 +361,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     nav_youtube.click(lambda: show_page(youtube_page), outputs=list(show_page(youtube_page).keys()))
     
     analyze_btn.click(
-        process_youtube_video,
-        inputs=[video_url, keywords],
-        outputs=[video_thumbnail, summary, sentiment, recommendations]
-    )
+    process_youtube_video,
+    inputs=[video_url, keywords],
+    outputs=[video_thumbnail, summary, sentiment, recommendations]
+)
     
     logout_btn.click(
         lambda: {