update

app.py

@@ -49,6 +49,15 @@ class GPTAgent:
 
 
 
+def authenticate_user():
+    user_key = st.text_input("Enter your authentication key", type="password")
+
+    if user_key == os.getenv("AUTH_KEY"):
+        st.success("Authentication successful!")
+        return True
+    else:
+        st.error("Invalid authentication key. Please try again.")
+        return False
 
 def generate_potshot_prompt(batch, role, tone, audience, values):
     prompt = f"""You are a recruitment specialist tasked with generating witty and engaging recruitment potshots.
@@ -99,53 +108,55 @@ def get_potshots(n_repeat=1, batch=20, role="Developer", tone="humorous", audien
 # Streamlit App Interface
 st.title("Recruiting Potshots Generator")
 
-# Input Fields for Customization
-# Input Fields for Full Customization with Default Values
-role = st.text_input("Job Role", value="Developer", help="Enter the job role you are recruiting for. Default is 'Developer'.")
-tone = st.text_input("Tone of the Potshots", value="humorous", help="Enter any tone for the potshots (e.g., humorous, serious, edgy). Default is 'humorous'.")
-audience = st.text_input("Target Audience", value="tech-savvy candidates", help="Define the target audience for the potshots. Default is 'tech-savvy candidates'.")
-values = st.text_area("Company Values", value="innovation, teamwork, transparency", help="Enter your company values that should be reflected in the potshots. Default is 'innovation, teamwork, transparency'.")
-batch_size = st.number_input("Batch Size", min_value=1, max_value=100, value=10)
-repeat_times = st.number_input("Number of Batches", min_value=1, max_value=10, value=1)
+if authenticate_user():
+
+    # Input Fields for Customization
+    # Input Fields for Full Customization with Default Values
+    role = st.text_input("Job Role", value="Developer", help="Enter the job role you are recruiting for. Default is 'Developer'.")
+    tone = st.text_input("Tone of the Potshots", value="humorous", help="Enter any tone for the potshots (e.g., humorous, serious, edgy). Default is 'humorous'.")
+    audience = st.text_input("Target Audience", value="tech-savvy candidates", help="Define the target audience for the potshots. Default is 'tech-savvy candidates'.")
+    values = st.text_area("Company Values", value="innovation, teamwork, transparency", help="Enter your company values that should be reflected in the potshots. Default is 'innovation, teamwork, transparency'.")
+    batch_size = st.number_input("Batch Size", min_value=1, max_value=100, value=10)
+    repeat_times = st.number_input("Number of Batches", min_value=1, max_value=10, value=1)
 
 
-def to_excel(df):
-    output = BytesIO()
+    def to_excel(df):
+        output = BytesIO()
 
-    # Use the 'with' statement to handle the file writer and ensure proper closure
-    with pd.ExcelWriter(output, engine='xlsxwriter') as writer:
-        df.to_excel(writer, index=False, sheet_name='Potshots')
+        # Use the 'with' statement to handle the file writer and ensure proper closure
+        with pd.ExcelWriter(output, engine='xlsxwriter') as writer:
+            df.to_excel(writer, index=False, sheet_name='Potshots')
 
-    # Get the processed data from the buffer
-    processed_data = output.getvalue()
+        # Get the processed data from the buffer
+        processed_data = output.getvalue()
 
-    # Reset the buffer for safety (useful if this is reused)
-    output.seek(0)
+        # Reset the buffer for safety (useful if this is reused)
+        output.seek(0)
 
-    return processed_data
+        return processed_data
 
 
-if st.button("Generate Potshots"):
-    with st.spinner("Generating customized potshots..."):
-        df = get_potshots(n_repeat=repeat_times, batch=batch_size, role=role, tone=tone, audience=audience,
-                          values=values)
+    if st.button("Generate Potshots"):
+        with st.spinner("Generating customized potshots..."):
+            df = get_potshots(n_repeat=repeat_times, batch=batch_size, role=role, tone=tone, audience=audience,
+                              values=values)
 
-        # Display the generated potshots as a table
-        st.write("### Generated Potshots")
-        st.dataframe(df)
+            # Display the generated potshots as a table
+            st.write("### Generated Potshots")
+            st.dataframe(df)
 
-        excel_data = to_excel(df)
-        file_name = "potshots.xlsx"
+            excel_data = to_excel(df)
+            file_name = "potshots.xlsx"
 
-        # Open and read the file to serve for download
-        with open(file_name, "wb") as f:
-            f.write(excel_data)
+            # Open and read the file to serve for download
+            with open(file_name, "wb") as f:
+                f.write(excel_data)
 
-        # Provide a download button for the Excel file
-        with open(file_name, "rb") as template_file:
-            template_byte = template_file.read()
+            # Provide a download button for the Excel file
+            with open(file_name, "rb") as template_file:
+                template_byte = template_file.read()
 
-        st.download_button(label="Click to Download Potshots File",
-                           data=template_byte,
-                           file_name="potshots.xlsx",
-                           mime='application/octet-stream')
+            st.download_button(label="Click to Download Potshots File",
+                               data=template_byte,
+                               file_name="potshots.xlsx",
+                               mime='application/octet-stream')