Update app.py

app.py

@@ -1,85 +1,52 @@
-
-import streamlit as st
-import http.client
-import json
-import google.generativeai as genai
-
-# Configure the Generative AI API
-genai.configure(api_key="AIzaSyDiGhxeTJVQoK8jyRge6aaZGg7CrLq0zGk")
-
-import urllib.parse
-
-
-# Function to fetch subtitles using RapidAPI
-def fetch_youtube_subtitles(video_url):
-    try:
-        # Ensure the URL is properly encoded
-        encoded_url = urllib.parse.quote(video_url, safe="")
-        
-        # Establish connection to RapidAPI service
-        conn = http.client.HTTPSConnection("youtube-subtitles-captions-downloader.p.rapidapi.com")
-        headers = {
-            'x-rapidapi-key': "ef4899abb4msh72ad9a5b860e697p191aa4jsn14fbc3ce6661",
-            'x-rapidapi-host': "youtube-subtitles-captions-downloader.p.rapidapi.com"
-        }
-
-        # Request subtitles
-        conn.request("GET", f"/subtitles/srt/?url={encoded_url}&lang=en", headers=headers)
-        res = conn.getresponse()
-
-        # Handle response
-        if res.status != 200:
-            raise Exception(f"Failed to fetch subtitles. Status code: {res.status}")
-        
-        data = res.read()
-        transcript = data.decode("utf-8")
-
-        # Check if subtitle data is empty
-        if not transcript:
-            raise Exception("No subtitles available for this video.")
-        
-        return transcript
-
-    except Exception as e:
-        st.error(f"Error fetching subtitles: {str(e)}")
-        return None
-
-# Function to generate summary using Gemini
-def get_gemini_response(prompt):
-    model = genai.GenerativeModel('gemini-pro')
-    response = model.generate_content(prompt)
-    if hasattr(response.candidates[0], "content") and hasattr(response.candidates[0].content, "parts"):
-        return response.candidates[0].content.parts[0].text
-    else:
-        return "Failed to generate summary."
-
-# Streamlit UI
-st.title("YouTube Video Summarizer")
-
-video_url = st.text_input("Enter YouTube Video URL:")
-
-if st.button("Summarize"):
-    if video_url:
-        try:
-            # Fetch subtitles
-            st.info("Fetching subtitles...")
-            transcript = fetch_youtube_subtitles(video_url)
-            
-            if transcript:
-                # Display fetched subtitles
-                st.subheader("Fetched Transcript:")
-                st.text_area("Transcript", transcript, height=200)
-
-                # Summarize the transcript using Gemini
-                st.info("Summarizing transcript...")
-                summary = get_gemini_response(f"Summarize the following transcript: {transcript}")
-                
-                # Display the summary
-                st.subheader("Summary:")
-                st.write(summary)
-            else:
-                st.warning("No subtitles found for the video.")
-        except Exception as e:
-            st.error(f"An error occurred: {e}")
-    else:
-        st.error("Please enter a valid YouTube URL.")
+
+import streamlit as st
+import http.client
+import json
+import google.generativeai as genai
+
+# Configure the Generative AI API
+genai.configure(api_key="AIzaSyDiGhxeTJVQoK8jyRge6aaZGg7CrLq0zGk")
+
+
+
+# Function to generate summary using Generative AI
+def get_gemini_response(prompt):
+    model = genai.GenerativeModel('gemini-pro')
+    response = model.generate_content(prompt)
+    if hasattr(response.candidates[0], "content"):
+        return response.candidates[0].content
+    else:
+        return "Failed to generate summary."
+
+# Streamlit App
+st.title("YouTube Video Summarizer")
+
+# Input for YouTube video URL
+video_url = st.text_input("Enter YouTube Video URL:")
+
+if st.button("Generate Summary"):
+    if "youtube.com" in video_url or "youtu.be" in video_url:
+        # Extract video ID
+        video_id = video_url.split("v=")[-1].split("&")[0] if "v=" in video_url else video_url.split("/")[-1]
+        
+        # Fetch transcript
+        with st.spinner("Fetching transcript..."):
+            api_url = "https://resumegenie-c0pc.onrender.com/fetch_captions"
+            try:
+                response = requests.post(api_url, json={"video_id": video_id})
+                if response.status_code == 200:
+                    transcript_data = response.json().get("transcript")
+                    transcript = " ".join([item["text"] for item in transcript_data])
+                    
+                    # Generate summary using Gemini
+                    with st.spinner("Generating summary..."):
+                        summary = get_gemini_response(f"Summarize the following text: {transcript}")
+                        st.subheader("Video Summary")
+                        st.write(summary)
+                else:
+                    st.error(f"Error: {response.json().get('detail', 'Unknown error')}")
+            except Exception as e:
+                st.error(f"Error: {str(e)}")
+    else:
+        st.error("Invalid YouTube URL.")
+