Update resume_screener.py

resume_screener.py

@@ -1,72 +1,126 @@
-import streamlit as st
-
-#importing the pages
-from resume_screener import ResumeScreener
-from user_documentation import UserDocumentation
-from home import Home
-
-UserDocPage = st.Page(UserDocumentation, title="User Documentation", icon=":material/help:")
-ResumeScreenerPage = st.Page(ResumeScreener, title="Resume Screener", icon=":material/recent_actors:")
-HomePage = st.Page(Home, title="Home", icon=":material/home:")
-
-
-if "LOGGED_IN" not in st.session_state:
-    st.session_state.LOGGED_IN = False
-
-if st.session_state.LOGGED_IN:
-    pages = {
-        "Dashboard": [
-            HomePage,
-            ResumeScreenerPage,
-        ],
-        "Resources": [
-            UserDocPage,
-        ]
-    }
-
-    pg = st.navigation(pages)
-    pg.run()
-
-else:
-    col1, col2, col3 = st.columns(spec=[1, 3, 1])
-
-    with col2:
-        st.title("Login")
-        username = st.text_input("Email Address", value="johndoe@hr.company.com")
-        password = st.text_input("Password", type="password")
-
-        if st.button("Login"):
-
-            class UnregisteredUser(Exception):
-                """
-                Represents an exception when the user is not registered.
-                """
-
-                def __init__(self, message="User not registered"):
-                    self.message = message
-                    super().__init__(self.message)
-
-            class IncompleteDetails(Exception):
-                """
-                Represents an exception when the user's details are incomplete.
-                """
-
-                def __init__(self, message="Incomplete details"):
-                    self.message = message
-                    super().__init__(self.message)
-
-            #imaginary db calls for existing user, if user is in imaginary db, user is logged in
-            try:
-                if username == "johndoe@hr.company.com" and password == "password":
-                    st.session_state.LOGGED_IN = True
-                    st.rerun()
-
-                if username == "" or password == "":
-                    raise IncompleteDetails
-
-                raise UnregisteredUser
-
-            except UnregisteredUser as e:
-                st.error(e)
-            except IncompleteDetails as e:
-                st.error(e)
+from helpers import *
+
+# uuid generator
+from uuid import uuid4
+
+load_dotenv()
+
+def ResumeScreener():
+    
+    class EmptyText(Exception):
+        """
+        A custom exception for when an input component
+        like st.text_area is empty.
+        """
+
+        def __init__(self, message = 'Text area cannot be empty'):
+            self.message = message
+            super().__init__(self.message)
+
+    st.set_page_config(
+        page_title="Screening Tool",
+        page_icon="🔎",
+        layout="wide",
+        initial_sidebar_state="auto"
+    )
+
+    try:
+        # raises an error if keys are missing
+        check_missing_keys()
+        
+        # Initialize the vector store -> the pinecone vector store
+        vector_store = pinecone_vector_store(
+            embedding="text-embedding-3-small",
+            index='resumedb'
+            ) 
+
+        # File Uploader component -> list of Documents()
+        documents = file_uploader() 
+
+        # 'Upload the documents' button to the vector store
+        upload_button(
+            documents=documents, 
+            vector_store=vector_store
+            )
+
+        col1, col2 = st.columns(
+            spec=[2, 3],
+            gap="medium"
+            )
+
+
+        # Job Description text area
+        with col1:
+
+            # Dynamic text area height
+            st.markdown("""
+            <style>
+            textarea {
+                height: calc(100vh - 350px) !important; /* Adjust based on screen height */
+                min-height: 200px; /* Optional: Set a minimum height */
+            }
+            </style>
+            """, unsafe_allow_html=True)
+
+            txt = st.text_area(
+            "Describe the job description to match.",
+            placeholder = "Position: Data Scientist ...",
+            height = 600,
+            )
+
+            # Documents to retrieve
+            docs_num = st.number_input(
+                label="Number of resumes to retrieve",
+                step=1,
+                min_value=1,
+                max_value=100,
+                value=10
+                )
+
+            if 'DISPLAY_RESULTS' not in st.session_state:
+                st.session_state.DISPLAY_RESULTS = False
+
+            # Match resumes button
+            if st.button(
+                label= 'Match resumes', 
+                disabled = st.session_state.KEYS_ARE_MISSING,
+                use_container_width = True,
+                type = 'primary'
+                ):
+            
+                st.session_state.DISPLAY_RESULTS = True
+
+        # Display the results here
+        with col2:
+            if st.session_state.DISPLAY_RESULTS:
+
+                # Dynamic UI height
+                st.markdown("""
+                <style>
+                textarea {
+                    height: calc(100vh - 350px) !important; /* Adjust based on screen height */
+                    min-height: 200px; /* Optional: Set a minimum height */
+                }
+                </style>
+                """, unsafe_allow_html=True)
+                with st.container(
+                        height=600,
+                        border=False
+                        ):
+
+                    # The LLM chain for summarization
+                    if 'SUMMARIZATION_CHAIN' not in st.session_state:
+                        st.session_state.SUMMARIZATION_CHAIN = summarize_chain()
+
+                    # The results component
+                    match_resumes(
+                        job_description=txt,
+                        k=docs_num,
+                        vector_store=vector_store,
+                        summarization_chain=st.session_state.SUMMARIZATION_CHAIN
+                    )
+
+                    st.session_state.DISPLAY_RESULTS = False
+
+    except Exception as e:
+        st.error(f"An error in function ResumeScreener has occurred: {e}")