Upload 2 files

app.py

@@ -1,60 +1,83 @@
 import streamlit as st
 import google.generativeai as genai
 from youtube_transcript_api import YouTubeTranscriptApi
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
+import io
 
-# Define the prompt
-prompt = """You are a YouTube video summarizer. You will be taking the transcript text
-and summarizing the entire video and providing the important summary in points
-within 1500 words. Please provide the summary of the text given here:  """
+# Set Streamlit page configuration
+st.set_page_config(page_title="YouTube Video Summarizer", layout="wide")
 
-## getting the transcript data from yt videos
+# Sidebar for user inputs
+google_api_key = st.sidebar.text_input("Enter your Google API Key:", type="password")
+youtube_link = st.sidebar.text_input("Enter YouTube Video Link:")
+
+# Summary length customization
+summary_length = st.sidebar.select_slider(
+    "Select Summary Length:", options=['Short', 'Medium', 'Long'], value='Medium'
+)
+
+# Define functions
 def extract_transcript_details(youtube_video_url):
     try:
         video_id = youtube_video_url.split("=")[1]
-        transcript_text = YouTubeTranscriptApi.get_transcript(video_id)
-        transcript = ""
-        for i in transcript_text:
-            transcript += " " + i["text"]
-        return transcript
+        transcript = YouTubeTranscriptApi.get_transcript(video_id)
+        return " ".join(segment["text"] for segment in transcript)
     except Exception as e:
-        raise e
+        st.sidebar.error(f"An error occurred: {e}")
+        return None
 
-## getting the summary based on Prompt from Google Gemini Pro
 def generate_gemini_content(transcript_text, prompt, api_key):
-    genai.configure(api_key=api_key)
-    model = genai.GenerativeModel("gemini-pro")
-    response = model.generate_content(prompt + transcript_text)
-    return response.text
+    try:
+        genai.configure(api_key=api_key)
+        model = genai.GenerativeModel("gemini-pro")
+        response = model.generate_content(prompt + transcript_text)
+        return response.text
+    except Exception as e:
+        st.error(f"An error occurred: {e}")
+        return None
+
+def create_pdf(summary_text):
+    buffer = io.BytesIO()
+    c = canvas.Canvas(buffer, pagesize=letter)
+    c.drawString(72, 800, "Summary")
+    text = c.beginText(40, 780)
+    text.setFont("Helvetica", 12)
+    for line in summary_text.split('\n'):
+        text.textLine(line)
+    c.drawText(text)
+    c.showPage()
+    c.save()
+    buffer.seek(0)
+    return buffer.getvalue()
 
+# UI elements
 st.title("YouTube Video Summarizer")
 
-# Display the link to get the Google API key
-st.markdown(
-    "If you don't have a Google API Key, you can get one [here](https://makersuite.google.com/app/apikey).",
-    unsafe_allow_html=True
-)
+# Display video thumbnail
+if youtube_link:
+    video_id = youtube_link.split("=")[1]
+    video_thumbnail = f"http://img.youtube.com/vi/{video_id}/0.jpg"
+    st.image(video_thumbnail, caption="Video Thumbnail", use_column_width=True)
 
-# Get the API key from user input
-google_api_key = st.text_input("Enter your Google API Key:", type="password")
-
-if google_api_key:
-    youtube_link = st.text_input("Enter YouTube Video Link:")
-
-    if youtube_link:
-        video_id = youtube_link.split("=")[1]
-        st.image(f"http://img.youtube.com/vi/{video_id}/0.jpg", use_column_width=True)
-        
-    if st.button("Get Detailed Notes"):
-        # Call function to extract transcript
-        try:
-            transcript_text = extract_transcript_details(youtube_link)
-            if transcript_text:
-                st.success("Transcript extracted successfully!")
-                # Generate detailed notes
-                summary = generate_gemini_content(transcript_text, prompt, google_api_key)
-                st.markdown("## Detailed Notes:")
-                st.write(summary)
-            else:
-                st.error("Failed to extract transcript.")
-        except Exception as e:
-            st.error(f"An error occurred: {e}")
+# Process and display summary
+if google_api_key and youtube_link and st.button("Get Detailed Notes"):
+    transcript_text = extract_transcript_details(youtube_link)
+    if transcript_text:
+        prompt = """You are a YouTube video summarizer. Summarize the video content into key points within 1500 words."""
+        customized_prompt = f"{prompt} Please generate a {summary_length.lower()} summary."
+        summary = generate_gemini_content(transcript_text, customized_prompt, google_api_key)
+        if summary:
+            st.success("Transcript extracted and summary generated successfully!")
+            st.subheader("Detailed Notes:")
+            st.write(summary)
+            # PDF download
+            pdf_bytes = create_pdf(summary)
+            st.download_button(label="Download Summary as PDF",
+                               data=pdf_bytes,
+                               file_name="YouTube_Summary.pdf",
+                               mime="application/pdf")
+        else:
+            st.error("Failed to generate summary.")
+    else:
+        st.error("Failed to extract transcript.")

requirements.txt

@@ -2,4 +2,5 @@ youtube_transcript_api
 streamlit
 google-generativeai
 python-dotenv
-pathlib
+pathlib
+reportlab