Update app.py

app.py

@@ -2,86 +2,69 @@ import streamlit as st
 import openai
 import pandas as pd
 from youtube_transcript_api import YouTubeTranscriptApi
-import io
+from io import StringIO
 from docx import Document
 
-# Streamlit interface
-st.title("YouTube Video to Shorts Script Generator")
-st.write("Convert YouTube video transcripts to a 1-minute script using ChatGPT.")
+# Function to get YouTube transcript
+def get_transcript(video_id):
+    transcript = YouTubeTranscriptApi.get_transcript(video_id)
+    full_text = " ".join([entry['text'] for entry in transcript])
+    return full_text
 
-# 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")
+# Function to generate a script using OpenAI
+def generate_shorts_script(transcript, ai_system_content, user_content):
+    openai.api_key = st.secrets["OPENAI_API_KEY"]  # API Key from Streamlit secrets
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[
+            {"role": "system", "content": ai_system_content},
+            {"role": "user", "content": f"{user_content}: {transcript}"}
+        ]
+    )
+    return response.choices[0].message["content"]
 
-# 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:")
+# Streamlit UI
+st.title("YouTube Shorts Script Generator")
+
+# Input fields
+openai_api_key = st.text_input("OpenAI API Key", type="password")
+video_id = st.text_input("YouTube Video ID")
+ai_system_content = st.text_area("AI System Content")
+user_content = st.text_area("User Content")
 
-# Generate button
 if st.button("Generate Script"):
-    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, api_key=youtube_transcript_api_key)
-            full_text = " ".join([entry['text'] for entry in transcript])
-            
-            # Call OpenAI API
+    if openai_api_key and video_id and ai_system_content and user_content:
+        try:
+            # Set OpenAI API key
             openai.api_key = openai_api_key
-            response = openai.ChatCompletion.create(
-                model="gpt-3.5-turbo",
-                messages=[
-                    {"role": "system", "content": ai_agent_system_content},
-                    {"role": "user", "content": f"{user_content}: {full_text}"}
-                ]
-            )
-            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")
+            # Get transcript
+            transcript = get_transcript(video_id)
 
-            elif download_transcript == "Text":
-                st.download_button(label="Download Text", data=full_text, file_name="transcript.txt", mime="text/plain")
+            # Generate script
+            shorts_script = generate_shorts_script(transcript, ai_system_content, user_content)
 
-            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
-            st.subheader("Download Script")
-            download_script = st.selectbox("Choose download format for script:", ["CSV", "Text", "Word"], index=1)
+            # Display script
+            st.write("Generated Script:")
+            st.write(shorts_script)
 
-            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")
+            # Options to download
+            if st.button("Download as CSV"):
+                df = pd.DataFrame({"Script": [shorts_script]})
+                csv = df.to_csv(index=False)
+                st.download_button("Download CSV", csv, "script.csv", "text/csv")
 
-            elif download_script == "Text":
-                st.download_button(label="Download Text", data=shorts_script, file_name="shorts_script.txt", mime="text/plain")
+            if st.button("Download as Text"):
+                st.download_button("Download Text", shorts_script, "script.txt", "text/plain")
 
-            elif download_script == "Word":
+            if st.button("Download as Word"):
                 doc = Document()
                 doc.add_paragraph(shorts_script)
-                doc_buffer = io.BytesIO()
-                doc.save(doc_buffer)
-                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")
+                buffer = StringIO()
+                doc.save(buffer)
+                docx = buffer.getvalue()
+                st.download_button("Download Word", docx, "script.docx", "application/vnd.openxmlformats-officedocument.wordprocessingml.document")
+        except Exception as e:
+            st.error(f"An error occurred: {e}")
     else:
-        st.error("Please provide all required inputs.")
+        st.error("Please fill all the input fields.")