Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from dotenv import load_dotenv
 from utils import generate_script
+import time
 
 # Load environment variables (for local development)
 load_dotenv()
@@ -21,17 +22,34 @@ creativity = st.slider("Set creativity level:", min_value=0.0, max_value=1.0, va
 # Button to generate the script
 generate_script_button = st.button("Generate Script")
 
-# When the button is clicked, generate the script
+# Placeholder for generated script
+script_placeholder = st.empty()
+
 if generate_script_button:
     if not api_key:
         st.error("Please provide a valid Google API key.")
     else:
-        try:
-            title, script, search_data = generate_script(prompt, video_length, creativity, api_key)
-            st.success("Script generated successfully!")
-            st.subheader(f"Title: {title}")
-            st.write(f"Script:\n\n{script}")
-            with st.expander("Search Data Used for the Script"):
-                st.write(search_data)
-        except Exception as e:
-            st.error(f"An error occurred: {e}")
+        # Show loading spinner while generating the script
+        with st.spinner("Generating script... Please wait."):
+            try:
+                title, script, search_data = generate_script(prompt, video_length, creativity, api_key)
+                time.sleep(1)  # Simulate processing time for a better user experience
+
+                # Display generated script
+                st.success("Script generated successfully!")
+                st.subheader(f"Title: {title}")
+                st.write(f"Script:\n\n{script}")
+                with st.expander("Search Data Used for the Script"):
+                    st.write(search_data)
+
+                # Add a download button for the script
+                script_filename = "generated_script.txt"
+                script_content = f"Title: {title}\n\nScript:\n{script}\n\nSearch Data:\n{search_data}"
+                st.download_button(
+                    label="Download Script",
+                    data=script_content,
+                    file_name=script_filename,
+                    mime="text/plain"
+                )
+            except Exception as e:
+                st.error(f"An error occurred: {e}")