Update app.py

app.py

@@ -1,53 +1,52 @@
-from dotenv import load_dotenv
-load_dotenv()
-import streamlit as st
-import os
-import google.generativeai as genai
-from PIL import Image
-import pdf2image
-import io
-import base64
-from youtube_transcript_api import YouTubeTranscriptApi
-
-genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
-# load gemini model
-
-prompt="""You are Yotube video summarizer. You will be taking the transcript text
-and summarizing the entire video and providing the important summary in points
-within 250 words. Please provide the summary of the text given here:  """
-
-def get_youtube_transcript(video_url):
-    try:
-        video_id = video_url.split("=")[1]
-        # video_id = video_id.split("&")[0]
-        print(video_id)
-        transcript = YouTubeTranscriptApi.get_transcript(video_id)
-        transcript_text = ""
-        for i in transcript:
-            transcript_text +=  " " + i['text']
-        return transcript_text
-    except Exception as e:
-        raise e
-
-
-def get_gemini_response(transcript_text,prompt):
-    model=genai.GenerativeModel("gemini-pro")
-    response=model.generate_content(prompt+transcript_text)
-    return response.text
-
-
-
-st.title("Youtube Transcript to detailed summary")
-st.markdown("This app uses gemini pro to generate a detailed summary of the transcript of a youtube video")
-video_url = st.text_input("Enter the youtube video url")
-
-if video_url:
-    video_id=video_url.split("=")[1]
-    print(video_id)
-    st.image(f"https://img.youtube.com/vi/{video_id}/0.jpg",use_column_width=True)
-if st.button("Generate Summary"):
-        transcript_text = get_youtube_transcript(video_url)
-        if transcript_text:
-            response = get_gemini_response(transcript_text,prompt)
-            st.markdown("## Summary:")
+from dotenv import load_dotenv
+load_dotenv()
+import streamlit as st
+import os
+import google.generativeai as genai
+from PIL import Image
+import io
+import base64
+from youtube_transcript_api import YouTubeTranscriptApi
+
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+# load gemini model
+
+prompt="""You are Yotube video summarizer. You will be taking the transcript text
+and summarizing the entire video and providing the important summary in points
+within 250 words. Please provide the summary of the text given here:  """
+
+def get_youtube_transcript(video_url):
+    try:
+        video_id = video_url.split("=")[1]
+        # video_id = video_id.split("&")[0]
+        print(video_id)
+        transcript = YouTubeTranscriptApi.get_transcript(video_id)
+        transcript_text = ""
+        for i in transcript:
+            transcript_text +=  " " + i['text']
+        return transcript_text
+    except Exception as e:
+        raise e
+
+
+def get_gemini_response(transcript_text,prompt):
+    model=genai.GenerativeModel("gemini-pro")
+    response=model.generate_content(prompt+transcript_text)
+    return response.text
+
+
+
+st.title("Youtube Transcript to detailed summary")
+st.markdown("This app uses gemini pro to generate a detailed summary of the transcript of a youtube video")
+video_url = st.text_input("Enter the youtube video url")
+
+if video_url:
+    video_id=video_url.split("=")[1]
+    print(video_id)
+    st.image(f"https://img.youtube.com/vi/{video_id}/0.jpg",use_column_width=True)
+if st.button("Generate Summary"):
+        transcript_text = get_youtube_transcript(video_url)
+        if transcript_text:
+            response = get_gemini_response(transcript_text,prompt)
+            st.markdown("## Summary:")
             st.write(response)