Update app.py

app.py

@@ -1,3 +1,5 @@
+# app.py
+
 import streamlit as st
 from streamlit_option_menu import option_menu
 from langchain_groq import ChatGroq
@@ -15,6 +17,11 @@ import yaml
 import json
 import os
 
+# -------------------------------
+# Initialize the LLM with Groq API Key
+# -------------------------------
+
+# Ensure API keys are stored securely using secrets.toml
 GROQ_API_KEY = st.secrets["GROQ_API_KEY"]
 RAPIDAPI_KEY = st.secrets["RAPIDAPI_KEY"]
 
@@ -24,6 +31,10 @@ llm = ChatGroq(
     model_name="llama-3.1-70b-versatile"
 )
 
+# -------------------------------
+# Authentication Setup
+# -------------------------------
+
 def load_authentication():
     """
     Loads and parses authentication configurations from Streamlit secrets.
@@ -31,24 +42,25 @@ def load_authentication():
     Returns:
         dict: Parsed authentication configurations with 'credentials' and 'cookie' as dictionaries.
     """
-    # Load authentication config from secrets.toml
-    config = st.secrets["auth"]
+    # Load raw authentication config from secrets.toml
+    raw_credentials = st.secrets["auth"]["credentials"]
+    raw_cookie = st.secrets["auth"]["cookie"]
     
     # Parse credentials from YAML string to dict
     try:
-        config['credentials'] = yaml.safe_load(config['credentials'])
+        credentials = yaml.safe_load(raw_credentials)
     except yaml.YAMLError as e:
         st.error(f"Error parsing credentials: {e}")
         return {}
     
     # Parse cookie from JSON string to dict
     try:
-        config['cookie'] = json.loads(config['cookie'])
+        cookie = json.loads(raw_cookie)
     except json.JSONDecodeError as e:
         st.error(f"Error parsing cookie configuration: {e}")
         return {}
     
-    return config
+    return {"credentials": credentials, "cookie": cookie}
 
 config = load_authentication()
 
@@ -70,7 +82,7 @@ if config:
         # -------------------------------
         # Define Helper Functions with Caching
         # -------------------------------
-        
+    
         @st.cache_data(ttl=3600)
         def extract_text_from_pdf(pdf_file):
             """
@@ -85,7 +97,7 @@ if config:
             except Exception as e:
                 st.error(f"Error extracting text from PDF: {e}")
                 return ""
-        
+    
         @st.cache_data(ttl=3600)
         def extract_job_description(job_link):
             """
@@ -104,7 +116,7 @@ if config:
             except Exception as e:
                 st.error(f"Error fetching job description: {e}")
                 return ""
-        
+    
         @st.cache_data(ttl=3600)
         def extract_requirements(job_description):
             """
@@ -127,7 +139,7 @@ if config:
             except Exception as e:
                 st.error(f"Error extracting requirements: {e}")
                 return ""
-        
+    
         @st.cache_data(ttl=3600)
         def generate_email(job_description, requirements, resume_text):
             """
@@ -161,7 +173,7 @@ if config:
             except Exception as e:
                 st.error(f"Error generating email: {e}")
                 return ""
-        
+    
         @st.cache_data(ttl=3600)
         def generate_cover_letter(job_description, requirements, resume_text):
             """
@@ -199,7 +211,7 @@ if config:
             except Exception as e:
                 st.error(f"Error generating cover letter: {e}")
                 return ""
-        
+    
         @st.cache_data(ttl=3600)
         def extract_skills(text):
             """
@@ -223,7 +235,7 @@ if config:
             except Exception as e:
                 st.error(f"Error extracting skills: {e}")
                 return []
-        
+    
         @st.cache_data(ttl=3600)
         def suggest_keywords(resume_text, job_description=None):
             """
@@ -249,7 +261,7 @@ if config:
             except Exception as e:
                 st.error(f"Error suggesting keywords: {e}")
                 return []
-        
+    
         @st.cache_data(ttl=3600)
         def get_job_recommendations(job_title, location="India"):
             """
@@ -292,7 +304,7 @@ if config:
             except Exception as err:
                 st.error(f"An error occurred: {err}")
                 return {}
-        
+    
         def create_skill_distribution_chart(skills):
             """
             Creates a bar chart showing the distribution of skills.
@@ -303,7 +315,7 @@ if config:
             df = pd.DataFrame(list(skill_counts.items()), columns=['Skill', 'Count'])
             fig = px.bar(df, x='Skill', y='Count', title='Skill Distribution')
             return fig
-        
+    
         def create_experience_timeline(resume_text):
             """
             Creates an experience timeline from the resume text.
@@ -346,7 +358,7 @@ if config:
             except Exception as e:
                 st.error(f"Error creating experience timeline: {e}")
                 return None
-        
+    
         def parse_duration(duration_str):
             """
             Parses duration strings like '2 years' or '6 months' into float years.
@@ -362,7 +374,7 @@ if config:
                     return 0
             except:
                 return 0
-        
+    
         # -------------------------------
         # Database Functions
         # -------------------------------
@@ -471,7 +483,7 @@ if config:
             except Exception as e:
                 st.error(f"Error generating learning path: {e}")
                 return ""
-        
+    
         # -------------------------------
         # Page Functions
         # -------------------------------
@@ -1193,7 +1205,6 @@ if config:
             elif selected == "Help":
                 help_page()
     
-
     
         if __name__ == "__main__":
             main_app()
@@ -1201,4 +1212,4 @@ if config:
     elif authentication_status == False:
         st.error('Username/password is incorrect')
     elif authentication_status == None:
-        st.warning('Please enter your username and password')
+        st.warning('Please enter your username and password')