Update app.py

app.py

@@ -6,64 +6,59 @@ import os
 openai.api_key = os.getenv("OPENAI_API_KEY")
 
 def generate_video_script(title, formality_level, emotional_tone, engagement_level):
+    # Construct the prompt with the provided details
     prompt = (
         f"[Use this video framework below]\n"
         f"First we need to have a strong hook.\n"
         f"A short intro that stops people from scrolling and holds their attention\n"
-        f"Example. \"{title}\"\n"
+        f"Example: \"{title}\"\n"
         f"Now we add in “The Setup\"\n"
         f"* This is used to Setup the scene and provide context to the Viewer\n"
-        f"* ex. \"I've been creating videos recently and I guess something finally clicked\"\n"
-        f"* This lets the viewer know that she’s been uploading more videos recently, and that could have led to an increase in views and followers.\n"
+        f"* Example: \"I've been creating videos recently and I guess something finally clicked\"\n"
         f"ACT 2: \"The Conflict\"\n"
         f"* Introduce a point of conflict or problem that requires a solution\n"
-        f"* ex. \"Some of you have been asking about scripting and storytelling and there's a lot to talk about, but I gotchu\"\n"
-        f"* Setting up the foundation to lead to the topic of the video and posing the concern that 'there's a lot to talk about'\n"
-        f"* Which opens the question of how we're going to tackle the topic in a limited amount of time\n"
+        f"* Example: \"Some of you have been asking about scripting and storytelling and there's a lot to talk about, but I gotchu\"\n"
         f"ACT 3: \"The Resolution\"\n"
         f"* Addressing the conflict we introduced and resolving it\n"
-        f"* ex. \"Here's how I tell my stories. All you need are these 3: ACT I, ACT II, and ACT III.\"\n"
-        f"* She proceeds to explain how we structure and tell our stories, which answers and resolves the initial conflict\n"
-        f"A call to action or loop.\n"
+        f"* Example: \"Here's how I tell my stories. All you need are these 3: ACT I, ACT II, and ACT III.\"\n"
         f"Call-To-Action\n"
-        f"* After giving valuable information to your viewer, you have to be able to lead them somewhere or encourage them to engage\n"
-        f"* ex. \"And if you wanna know how that looks like, then watch this video back and follow for more.\"\n"
-        f"* This encourages the audience to rewatch the video to see the video is an actual example of what they just learned and it encourages them to follow and share the content if they found it valuable!"
-        f"Follow these instructions above in creating a Viral Video script for me."
+        f"* Example: \"And if you wanna know how that looks like, then watch this video back and follow for more.\"\n"
+        f"Follow these instructions above in creating a Viral Video script for me.\n"
         f"Topic: {title}\n"
         f"Tone: {emotional_tone}\n"
         f"Formality Level: {formality_level}\n"
         f"Engagement Level: {engagement_level}"
     )
 
-    response = openai.ChatCompletion.create(
-        engine="gpt-3.5-turbo",
-        prompt=prompt,
-        max_tokens=300,
-        stop="\n"
-    )
-
-    script = response.choices[0].text.strip()
-    return script
+    try:
+        response = openai.ChatCompletion.create(
+            engine="gpt-3.5-turbo",
+            prompt=prompt,
+            max_tokens=300,
+            temperature=0.7,  # Adjust for creativity
+            stop="\n"
+        )
+        script = response.choices[0].text.strip()
+        return script
+    except Exception as e:
+        st.error(f"Failed to generate script: {str(e)}")
+        return "Error in generating script."
 
-# Streamlit UI for collecting inputs
+# Streamlit UI setup
 st.set_page_config(layout="wide")
 st.markdown("<h1 style='text-align: center; color: black;'>Viral Video Script Generator</h1>", unsafe_allow_html=True)
 
 # Create columns for input and output
 col1, col2 = st.columns([3, 1])
 
-# User input sections
 with col1:
     st.markdown("<h2 style='text-align: center; color: black;'>Video Information</h2>", unsafe_allow_html=True)
     video_title = st.text_input("Title of Video", placeholder="Enter the title of your video")
-    intended_audience = st.text_input("Intended Audience", placeholder="Who is your intended audience?")
     formality_level = st.selectbox("Formality Level", ["Casual", "Neutral", "Formal"])
     emotional_tone = st.selectbox("Emotional Tone", ["Positive", "Neutral", "Humorous"])
     engagement_level = st.selectbox("Engagement Level", ["Interactive", "Informative", "Persuasive"])
 
-# If the button is pressed, generate video script and display in the second column (right)
-generate_button = col1.button('Generate Video Script')  # Place the button in the first column
+generate_button = col1.button('Generate Video Script')
 
 if generate_button:
     if not video_title: