Create app.py

app.py

@@ -0,0 +1,80 @@
+import streamlit as st
+import openai
+from youtube_transcript_api import YouTubeTranscriptApi
+
+# Function to get YouTube transcript
+def get_transcript(video_id):
+    """Fetches the transcript for a given YouTube video ID."""
+    try:
+        transcript_list = YouTubeTranscriptApi.get_transcript(video_id)
+        transcript = " ".join([entry['text'] for entry in transcript_list])
+        return transcript
+    except Exception as e:
+        st.error(f"Error fetching transcript: {e}")
+        return None
+
+# Function to summarize the transcript using OpenAI's GPT API
+def summarize_transcript(transcript):
+    """Summarizes the given transcript using OpenAI's GPT API."""
+    if transcript is None:
+        return None
+
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[
+            {"role": "system", "content": "You are a helpful AI assistant that summarizes YouTube videos."},
+            {"role": "user", "content": f"Please provide a concise summary of the following YouTube video transcript:\n\n{transcript}"}
+        ]
+    )
+    summary = response.choices[0].message["content"].strip()
+    return summary
+
+# Function to generate a 1-minute YouTube Shorts script based on the summary
+def generate_shorts_script(summary):
+    """Generates a 1-minute shorts script based on the summary."""
+    if summary is None:
+        return None
+
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[
+            {"role": "system", "content": "You are a creative script writer for YouTube Shorts."},
+            {"role": "user", "content": f"Write a captivating and engaging 1-minute script for a YouTube Short based on this summary: {summary}. Focus on impactful visuals and concise storytelling."}
+        ]
+    )
+    script = response.choices[0].message["content"].strip()
+    return script
+
+# Streamlit UI
+st.title("YouTube Video to Shorts Script Converter")
+
+# Input fields
+openai_api_key = st.text_input("OpenAI API Key", type="password")
+video_id = st.text_input("YouTube Video ID")
+
+if st.button("Generate Shorts Script"):
+    if openai_api_key and video_id:
+        try:
+            # Set OpenAI API key
+            openai.api_key = openai_api_key
+
+            # Get transcript
+            transcript = get_transcript(video_id)
+
+            # Summarize the transcript
+            summary = summarize_transcript(transcript)
+
+            # Generate shorts script
+            if summary:
+                shorts_script = generate_shorts_script(summary)
+
+                # Display the generated shorts script
+                if shorts_script:
+                    st.write("Generated YouTube Shorts Script:")
+                    st.write(shorts_script)
+
+        except Exception as e:
+            st.error(f"An error occurred: {e}")
+    else:
+        st.error("Please fill in both the OpenAI API Key and YouTube Video ID.")
+