Update app.py

app.py

@@ -1,65 +1,60 @@
 import streamlit as st
-from youtube_transcript_api import YouTubeTranscriptApi
 import openai
 import pandas as pd
+from youtube_transcript_api import YouTubeTranscriptApi
 import io
 
-# Streamlit UI
+# Streamlit interface
 st.title("YouTube Video to Shorts Script Generator")
+st.write("Convert YouTube video transcripts to a 1-minute script using ChatGPT.")
 
-# 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
+# API key inputs
+openai_api_key = st.text_input("Enter your OpenAI API key:", type="password")
+youtube_video_id = st.text_input("Enter YouTube Video ID:")
 
-# 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
+# Generate 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()
+    if openai_api_key and youtube_video_id:
+        # Loading screen
+        with st.spinner("Generating script..."):
+            # Get the transcript
+            transcript = YouTubeTranscriptApi.get_transcript(youtube_video_id)
+            full_text = " ".join([entry['text'] for entry in transcript])
+            
+            # Call OpenAI API
+            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: {full_text}"}
+                ]
+            )
+            shorts_script = response['choices'][0]['message']['content']
+
+            # Display the result
+            st.subheader("Generated Script")
+            st.write(shorts_script)
+
+            # Options to download the script
+            download_format = st.selectbox("Choose download format:", ["CSV", "Text", "Word"])
+
+            if download_format == "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":
+                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
                 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}")
+                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")
     else:
-        st.warning("Please enter all required fields (API keys and Video ID).")
+        st.error("Please provide both API key and YouTube video ID.")