Update app.py

app.py

@@ -1,93 +1,56 @@
+
 import streamlit as st
-from pytube import YouTube
 from youtube_transcript_api import YouTubeTranscriptApi
 from crewai import Agent, Task, Crew
 from transformers import pipeline
-import torch  # Ensure PyTorch is imported to avoid errors
+import torch
 
-# Initialize Hugging Face summarization model
+# Hugging Face summarizer setup
 summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
-# Streamlit App Setup
-st.title("AI-Powered Blog Writer with CrewAI")
-st.write("Input a YouTube video URL, and AI agents will collaboratively create a blog.")
-
-# Input for YouTube URL
-video_url = st.text_input("Enter YouTube Video URL")
-
-# Define CrewAI Agents and Tasks
-def create_crew_agents(transcript_text):
-    # Research Agent - Extracts transcript
-    research_agent = Agent(
-        name="Research Agent",
-        role="Transcript Extractor",
-        goal="Extract transcript from YouTube videos.",
-        backstory="Expert in video transcription and data extraction."
-    )
-
-    # Blog Writer Agent - Generates a blog
-    blog_writer_agent = Agent(
-        name="Blog Writer Agent",
-        role="Blog Creator",
-        goal="Write an engaging blog post from the transcript.",
-        backstory="Creative writer specializing in transforming transcripts into engaging blogs."
-    )
+st.title("AI Blog Writer from YouTube")
 
-    # Editor Agent - Reviews and refines content
-    editor_agent = Agent(
-        name="Editor Agent",
-        role="Content Editor",
-        goal="Ensure blog quality by refining the text.",
-        backstory="Experienced editor ensuring clarity and quality in written content."
-    )
+# YouTube URL Input
+video_url = st.text_input("Enter YouTube Video URL:")
 
-    # Tasks for Each Agent
-    research_task = Task(
-        description="Extract transcript from the YouTube video.",
-        agent=research_agent
-    )
+def get_video_id(url):
+    """Extract YouTube video ID from the URL."""
+    import re
+    match = re.search(r"(?<=v=)[^&]+", url)
+    if match:
+        return match.group(0)
+    else:
+        st.error("Invalid YouTube URL. Please try again.")
+        return None
 
-    blog_task = Task(
-        description="Write a blog using the provided transcript.",
-        agent=blog_writer_agent,
-        context=transcript_text
-    )
+# CrewAI Agent Setup
+def create_crew_agents(transcript_text):
+    research_agent = Agent(name="Research Agent", goal="Extract transcript from videos.")
+    blog_writer_agent = Agent(name="Blog Writer", goal="Generate blog content.")
+    editor_agent = Agent(name="Editor", goal="Refine blog for clarity.")
 
-    editor_task = Task(
-        description="Edit and refine the generated blog for better clarity and quality.",
-        agent=editor_agent
-    )
+    research_task = Task(description="Fetch transcript.", agent=research_agent)
+    blog_task = Task(description="Write blog using transcript.", agent=blog_writer_agent, context=transcript_text)
+    editor_task = Task(description="Edit the blog.", agent=editor_agent)
 
-    # Define the Crew with Task Sequence
-    crew = Crew(
-        agents=[research_agent, blog_writer_agent, editor_agent],
-        tasks=[research_task, blog_task, editor_task]
-    )
-    
+    crew = Crew(agents=[research_agent, blog_writer_agent, editor_agent], tasks=[research_task, blog_task, editor_task])
     return crew
 
-# Button to Trigger Blog Generation
-if st.button("Generate Blog with AI"):
-    if video_url:
+# Button Trigger
+if st.button("Generate Blog"):
+    video_id = get_video_id(video_url)
+    if video_id:
         try:
-            yt = YouTube(video_url)
-            st.write(f"**Video Title:** {yt.title}")
-            
-            # Extract Transcript
-            video_id = yt.video_id
+            # Fetch Transcript using YouTube API
             transcript = YouTubeTranscriptApi.get_transcript(video_id)
             transcript_text = " ".join([item['text'] for item in transcript])
-            st.text_area("Transcript Extracted", transcript_text, height=200)
-            
-            # Initialize Crew and Execute
+            st.text_area("Transcript Extracted", transcript_text)
+
+            # CrewAI Agents Collaboration
             crew = create_crew_agents(transcript_text)
-            results = crew.kickoff()  # Trigger the sequence of tasks
-            
-            # Display the Blog Result
-            st.subheader("Final Blog Generated by the Crew")
+            results = crew.kickoff()
+            st.subheader("Generated Blog")
             st.write(results)
+
         except Exception as e:
             st.error(f"Error: {e}")
-    else:
-        st.warning("Please enter a valid YouTube video URL.")
-