Updating app.py

.gitignore

@@ -1,3 +1,4 @@
 /.streamlit
 script_history.json
-/__pycache__
+/__pycache__\
+.env

DOCS.md

@@ -12,7 +12,6 @@ Instead of DuckDuckGo due to unpredictable API limits, we will be using Serpo AP
 
 Storing scripts using **json** provides an organized access to scripts and files which can be stored an prepared later for future use.
 
-To store secrets, we will use Streamlit instead as being modular and quick to set-up rather than dotenv.
 ### 3. Every Python File Explained
 
 This project utilizes multiple python files. The **app.py** is where the main application resides. Following that includes the **scriptgen.py (Script Generation)**, **imggen.py (Image Generation)**, **tweaker.py (Creative Consultant)**, and the **secretary.py (Script Secretary)**.

app.py

@@ -1,38 +1,43 @@
 import streamlit as st
+from dotenv import load_dotenv, set_key
+import os
+
+# Load environment variables from .env file
+load_dotenv()
 
 st.set_page_config(layout='wide', page_icon='✨')
 
 scriptgen = st.Page("scriptgen.py", title="Script Generator", icon='📃')
 imggen = st.Page("imggen.py", title="Thumbnail Generator", icon='🖼️')
 tweaker = st.Page("tweaker.py", title="Creative Consultant", icon='🎬')
-jsonbase = st.Page("secretary.py", title = "Script Secretary", icon='📚')
-manual = st.Page("manual.py", title="Manual", icon="👓", default= True)
-docs = st.Page("docs.py", title="Documentation", icon= '📑')
+jsonbase = st.Page("secretary.py", title="Script Secretary", icon='📚')
+manual = st.Page("manual.py", title="Manual", icon="👓", default=True)
+docs = st.Page("docs.py", title="Documentation", icon='📑')
 
-api_key = st.secrets['OPENAI_API_KEY']
+# Check if OPENAI_API_KEY exists in the environment
+api_key = os.getenv('OPENAI_API_KEY', None)  # Fallback to None if not found
 
 # Sidebar for API Key input and options
 with st.sidebar:
-    # Retrieve API Key from Streamlit secrets or prompt for input
-    if 'OPENAI_API_KEY' in st.secrets:
-        api_key = st.secrets['OPENAI_API_KEY']  # Retrieve the API key from secrets
-        if not api_key.startswith('sk-proj'):  # Validate the API key format
-            api_key = st.text_input('Enter your OpenAI API token:', type='password')
+    api_key_input = st.text_input('Enter your OpenAI API token:', type='password')
+
+    if st.button("Save API Key"):
+        if api_key_input.startswith("sk-"):  # Basic validation
+            env_file = ".env"
+            set_key(env_file, "OPENAI_API_KEY", api_key_input)  # Save to .env
+            st.success("API Key saved successfully!")
         else:
-            st.success('API Key Recognized!')
-    else:
-        api_key = st.text_input('Enter your OpenAI API token:', type='password')
+            st.error("Invalid API Key format!")
 
-# Optional: Display feedback based on the provided or entered API key
-if api_key:
-    if api_key.startswith('sk-proj'):
-        st.sidebar.success('API Key is Valid!')
+    # Check and display the status of the API key
+    if api_key:
+        st.success("API Key loaded from environment!")
     else:
-        st.sidebar.warning('The provided API Key format is incorrect.')
+        st.warning("No API Key found. Please enter one.")
 
 pg = st.navigation(
     {
-        "Others":[manual, docs],
+        "Others": [manual, docs],
         "Tools": [scriptgen, imggen, tweaker, jsonbase]
     }
 )

imggen.py

@@ -28,7 +28,7 @@ def load_script_history():
 script_history = load_script_history()
 
 # Retrieve the API key
-api_key = st.secrets.get('OPENAI_API_KEY') or os.getenv("OPENAI_API_KEY")
+api_key = os.getenv("OPENAI_API_KEY")
 
 if not api_key:
     st.error("API Key not provided! Please add it to Streamlit secrets or the .env file.")

scriptgen.py

@@ -14,8 +14,8 @@ st.title("Script Generator 📃")
 st.subheader("Create your script with just a single prompt!")
 
 # Retrieve API key from secrets or environment variables
-api_key = st.secrets.get("OPENAI_API_KEY") or os.getenv("OPENAI_API_KEY")
-serper_api_key = st.secrets.get("SERPER_API_KEY") or os.getenv("SERPER_API_KEY")
+api_key = os.getenv("OPENAI_API_KEY")
+serper_api_key = os.getenv("SERPER_API_KEY")
 
 if not api_key:
     st.error("Please provide a valid API Key in Streamlit secrets or .env file.")

tweaker.py

@@ -39,15 +39,39 @@ script_history = load_script_history()
 # Sidebar for script selection
 if script_history:
     st.sidebar.header("Previously Generated Scripts")
+    
+    # Add "Other" option to the list of scripts
     script_titles = [f"Script {idx + 1}: {item['title']}" for idx, item in enumerate(script_history)]
-    selected_script_idx = st.sidebar.selectbox("Choose a script to tweak:", range(len(script_titles)), format_func=lambda x: script_titles[x])
-    selected_script = script_history[selected_script_idx]
-
-    # Display the selected script
-    st.sidebar.text_area("Script Content", value=selected_script["script"], height=200, disabled=True)
+    script_titles.append("Other")  # Add "Other" as an option
+    
+    # Let the user select a script or "Other"
+    selected_script_idx = st.sidebar.selectbox(
+        "Choose a script to tweak:",
+        range(len(script_titles)),
+        format_func=lambda x: script_titles[x]
+    )
+
+    # If "Other" is chosen
+    if selected_script_idx == len(script_titles) - 1:  
+        st.sidebar.text_area(
+            "Free Conversation or Custom Tweaking",
+            placeholder="Write your thoughts or tweaks here...",
+            height=200,
+            key="free_conversation"
+        )
+    else:
+        # Display the selected script
+        selected_script = script_history[selected_script_idx]
+        st.sidebar.text_area(
+            "Script Content",
+            value=selected_script["script"],
+            height=200,
+            disabled=True
+        )
 else:
     st.sidebar.info("No scripts available. Please generate scripts using the Script Generator.")
 
+
 st.sidebar.button('Clear Chat History', on_click=clear_chat_history)
 
 # Prompt template for the Creative Consultant