Update app.py

app.py

@@ -31,6 +31,10 @@ def clean_script(script):
     cleaned_script = re.sub(r"([a-zA-Z])(\d+)", r"\1 \2", cleaned_script)
     return cleaned_script
 
+# Initialize session state for script to avoid disappearing after download
+if 'script' not in st.session_state:
+    st.session_state.script = ""
+
 # Streamlit UI for input
 business_description = st.text_area(
     "Describe your business or video topic/idea:",
@@ -118,15 +122,17 @@ if generate_button:
         script = call_openai_api(prompt)
         if script:
             script = clean_script(script)  # Clean the output
-            st.markdown("### Generated Video Script")
-            st.write(script)
+            st.session_state.script = script  # Store the generated script in session state
+
+# Display the script if it exists
+if st.session_state.script:
+    st.markdown("### Generated Video Script")
+    st.write(st.session_state.script)
 
-            # Add a download button
-            st.download_button(
-                label="📥 Download Script as Text File",
-                data=script,
-                file_name='video_script.txt',
-                mime='text/plain'
-            )
-        else:
-            st.write("An error occurred while generating the script. Please try again.")
+    # Add a download button
+    st.download_button(
+        label="📥 Download Script as Text File",
+        data=st.session_state.script,
+        file_name='video_script.txt',
+        mime='text/plain'
+    )