Update app.py

app.py

@@ -3,6 +3,7 @@ import openai
 import pandas as pd
 from youtube_transcript_api import YouTubeTranscriptApi
 import io
+from docx import Document
 
 # Streamlit interface
 st.title("YouTube Video to Shorts Script Generator")
@@ -10,15 +11,20 @@ st.write("Convert YouTube video transcripts to a 1-minute script using ChatGPT."
 
 # API key inputs
 openai_api_key = st.text_input("Enter your OpenAI API key:", type="password")
+youtube_transcript_api_key = st.text_input("Enter your YouTube Transcript API key:", type="password")
+
+# Video ID and content inputs
 youtube_video_id = st.text_input("Enter YouTube Video ID:")
+ai_agent_system_content = st.text_area("Enter AI agent system content:")
+user_content = st.text_area("Enter user content:")
 
 # Generate button
 if st.button("Generate Script"):
-    if openai_api_key and youtube_video_id:
+    if openai_api_key and youtube_transcript_api_key and youtube_video_id and ai_agent_system_content and user_content:
         # Loading screen
         with st.spinner("Generating script..."):
             # Get the transcript
-            transcript = YouTubeTranscriptApi.get_transcript(youtube_video_id)
+            transcript = YouTubeTranscriptApi.get_transcript(youtube_video_id, api_key=youtube_transcript_api_key)
             full_text = " ".join([entry['text'] for entry in transcript])
             
             # Call OpenAI API
@@ -26,30 +32,51 @@ if st.button("Generate Script"):
             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: {full_text}"}
+                    {"role": "system", "content": ai_agent_system_content},
+                    {"role": "user", "content": f"{user_content}: {full_text}"}
                 ]
             )
-            shorts_script = response['choices'][0]['message']['content']
+            shorts_script = response.choices[0].message['content']
 
             # Display the result
             st.subheader("Generated Script")
             st.write(shorts_script)
 
+            # Options to download the transcript
+            st.subheader("Download Transcript")
+            download_transcript = st.selectbox("Choose download format for transcript:", ["CSV", "Text", "Word"])
+
+            if download_transcript == "CSV":
+                df = pd.DataFrame([full_text], columns=["Transcript"])
+                csv_buffer = io.StringIO()
+                df.to_csv(csv_buffer, index=False)
+                st.download_button(label="Download CSV", data=csv_buffer.getvalue(), file_name="transcript.csv", mime="text/csv")
+
+            elif download_transcript == "Text":
+                st.download_button(label="Download Text", data=full_text, file_name="transcript.txt", mime="text/plain")
+
+            elif download_transcript == "Word":
+                doc = Document()
+                doc.add_paragraph(full_text)
+                doc_buffer = io.BytesIO()
+                doc.save(doc_buffer)
+                doc_buffer.seek(0)
+                st.download_button(label="Download Word", data=doc_buffer, file_name="transcript.docx", mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document")
+
             # Options to download the script
-            download_format = st.selectbox("Choose download format:", ["CSV", "Text", "Word"])
+            st.subheader("Download Script")
+            download_script = st.selectbox("Choose download format for script:", ["CSV", "Text", "Word"], index=1)
 
-            if download_format == "CSV":
+            if download_script == "CSV":
                 df = pd.DataFrame([shorts_script], columns=["Script"])
                 csv_buffer = io.StringIO()
                 df.to_csv(csv_buffer, index=False)
                 st.download_button(label="Download CSV", data=csv_buffer.getvalue(), file_name="shorts_script.csv", mime="text/csv")
 
-            elif download_format == "Text":
+            elif download_script == "Text":
                 st.download_button(label="Download Text", data=shorts_script, file_name="shorts_script.txt", mime="text/plain")
 
-            elif download_format == "Word":
-                from docx import Document
+            elif download_script == "Word":
                 doc = Document()
                 doc.add_paragraph(shorts_script)
                 doc_buffer = io.BytesIO()
@@ -57,4 +84,4 @@ if st.button("Generate Script"):
                 doc_buffer.seek(0)
                 st.download_button(label="Download Word", data=doc_buffer, file_name="shorts_script.docx", mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document")
     else:
-        st.error("Please provide both API key and YouTube video ID.")
+        st.error("Please provide all required inputs.")