Create app.py

app.py

@@ -0,0 +1,65 @@
+import streamlit as st
+from youtube_transcript_api import YouTubeTranscriptApi
+import openai
+import pandas as pd
+import io
+
+# Streamlit UI
+st.title("YouTube Video to Shorts Script Generator")
+
+# User inputs for API keys and video ID
+youtube_api_key = st.text_input("Enter your YouTube API Key", type="password")
+openai_api_key = st.text_input("Enter your OpenAI API Key", type="password")
+video_id = st.text_input("Enter the Video ID")
+
+# Function to get transcript from a YouTube video
+def get_transcript(video_id):
+    transcript = YouTubeTranscriptApi.get_transcript(video_id)
+    full_text = " ".join([entry['text'] for entry in transcript])
+    return full_text
+
+# Function to generate a YouTube Shorts script using OpenAI
+def generate_shorts_script(transcript, openai_api_key):
+    openai.api_key = openai_api_key
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[
+            {"role": "system", "content": "You are an expert short form video script writer."},
+            {"role": "user", "content": f"Write a compelling 1 minute YouTube Shorts script based on this transcript: {transcript}"}
+        ]
+    )
+    return response.choices[0].message['content']
+
+# Action button
+if st.button("Generate Script"):
+    if youtube_api_key and openai_api_key and video_id:
+        with st.spinner("Extracting transcript and generating script..."):
+            try:
+                transcript = get_transcript(video_id)
+                shorts_script = generate_shorts_script(transcript, openai_api_key)
+
+                # Display output
+                st.subheader("Generated YouTube Shorts Script")
+                st.write(shorts_script)
+
+                # Prepare download options
+                buffer = io.StringIO()
+                buffer.write(shorts_script)
+                st.download_button(label="Download as Text", data=buffer.getvalue(), file_name=f"{video_id}_shorts_script.txt", mime="text/plain")
+
+                buffer = io.StringIO()
+                df = pd.DataFrame([{'Script': shorts_script}])
+                df.to_csv(buffer, index=False)
+                st.download_button(label="Download as CSV", data=buffer.getvalue(), file_name=f"{video_id}_shorts_script.csv", mime="text/csv")
+
+                buffer = io.BytesIO()
+                from docx import Document
+                doc = Document()
+                doc.add_paragraph(shorts_script)
+                doc.save(buffer)
+                st.download_button(label="Download as Word Document", data=buffer.getvalue(), file_name=f"{video_id}_shorts_script.docx", mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document")
+
+            except Exception as e:
+                st.error(f"An error occurred: {e}")
+    else:
+        st.warning("Please enter all required fields (API keys and Video ID).")