init!

app.py

@@ -141,72 +141,88 @@ def create_resume_ranking_model(job_description, resume_directory):
     return ranked_resumes
 
 
+
+#Streamlit App
+
+import streamlit as st
+import tempfile
+import os
+
 # Streamlit app
 st.title('Resume Ranking System')
 
 st.write("""
 This app ranks resumes based on their similarity to a given job description.
-Upload a directory containing resumes (PDF and DOCX formats) and enter a job description to get started.
+Upload resume files (PDF and DOCX formats) and enter a job description to get started.
 """)
 
 # Job description input
 job_description = st.text_area("Enter the job description:", height=200)
 
-# Resume directory input
-resume_directory = st.text_input("Enter the path to the resume directory:")
+# File uploader for resumes
+uploaded_files = st.file_uploader("Upload resume files", accept_multiple_files=True, type=['pdf', 'docx'])
 
 if st.button('Rank Resumes'):
-    if job_description and resume_directory:
+    if job_description and uploaded_files:
         try:
-            with st.spinner('Processing resumes...'):
-                ranked_resumes = create_resume_ranking_model(job_description, resume_directory)
-            
-            st.success('Resumes ranked successfully!')
-            
-            # Display results
-            st.write("Top 5 Ranked Resumes:")
-            st.dataframe(ranked_resumes.head())
-            
-            # Create a folder with ranked resumes
-            output_folder = "ranked_resumes"
-            if os.path.exists(output_folder):
-                shutil.rmtree(output_folder)
-            os.makedirs(output_folder)
-            
-            for index, row in ranked_resumes.iterrows():
-                src_file = os.path.join(resume_directory, row['filename'])
-                dst_file = os.path.join(output_folder, f"{index+1:03d}_{row['filename']}")
-                shutil.copy2(src_file, dst_file)
-            
-            # Create a zip file of the ranked resumes
-            shutil.make_archive(output_folder, 'zip', output_folder)
-            
-            # Offer the zip file for download
-            with open(f"{output_folder}.zip", "rb") as file:
+            # Create a temporary directory to store uploaded files
+            with tempfile.TemporaryDirectory() as temp_dir:
+                # Save uploaded files to the temporary directory
+                for uploaded_file in uploaded_files:
+                    file_path = os.path.join(temp_dir, uploaded_file.name)
+                    with open(file_path, "wb") as f:
+                        f.write(uploaded_file.getbuffer())
+                
+                # Process resumes
+                with st.spinner('Processing resumes...'):
+                    ranked_resumes = create_resume_ranking_model(job_description, temp_dir)
+                
+                st.success('Resumes ranked successfully!')
+                
+                # Display results
+                st.write("Top 5 Ranked Resumes:")
+                st.dataframe(ranked_resumes.head())
+                
+                # Create a folder with ranked resumes
+                output_folder = "ranked_resumes"
+                if os.path.exists(output_folder):
+                    shutil.rmtree(output_folder)
+                os.makedirs(output_folder)
+                
+                for index, row in ranked_resumes.iterrows():
+                    src_file = os.path.join(temp_dir, row['filename'])
+                    dst_file = os.path.join(output_folder, f"{index+1:03d}_{row['filename']}")
+                    shutil.copy2(src_file, dst_file)
+                
+                # Create a zip file of the ranked resumes
+                shutil.make_archive(output_folder, 'zip', output_folder)
+                
+                # Offer the zip file for download
+                with open(f"{output_folder}.zip", "rb") as file:
+                    st.download_button(
+                        label="Download ranked resumes as ZIP",
+                        data=file,
+                        file_name="ranked_resumes.zip",
+                        mime="application/zip"
+                    )
+                
+                # Option to download full results as CSV
+                csv = ranked_resumes.to_csv(index=False)
                 st.download_button(
-                    label="Download ranked resumes as ZIP",
-                    data=file,
-                    file_name="ranked_resumes.zip",
-                    mime="application/zip"
+                    label="Download full results as CSV",
+                    data=csv,
+                    file_name="ranked_resumes.csv",
+                    mime="text/csv",
                 )
-            
-            # Option to download full results as CSV
-            csv = ranked_resumes.to_csv(index=False)
-            st.download_button(
-                label="Download full results as CSV",
-                data=csv,
-                file_name="ranked_resumes.csv",
-                mime="text/csv",
-            )
         except Exception as e:
             st.error(f"An error occurred: {str(e)}")
     else:
-        st.warning("Please enter both a job description and a resume directory path.")
+        st.warning("Please enter a job description and upload at least one resume file.")
 
 st.write("""
 ### How to use this app:
 1. Enter the job description in the text area above.
-2. Provide the full path to the directory containing the resumes.
+2. Upload resume files (PDF and DOCX formats) using the file uploader.
 3. Click the 'Rank Resumes' button.
 4. View the top 5 ranked resumes in the table.
 5. Download the ranked resumes as a ZIP file.