Create app.py

app.py

@@ -0,0 +1,113 @@
+import streamlit as st
+import numpy as np
+import pickle
+import streamlit.components.v1 as components
+
+# Load the pickled model
+def load_model():
+    return pickle.load(open('Employee_Attrition_ra.pkl', 'rb'))
+
+# Function for model prediction
+def model_prediction(model, features):
+    predicted = str(model.predict(features)[0])
+    return predicted
+
+def app_design():
+    # Add input fields for High, Open, and Low values
+    image = '13.png'
+    st.image(image, use_column_width=True)
+    
+    st.subheader("Enter the following values:")
+
+    Age = st.number_input("Age")
+    BusinessTravel = st.number_input("Business Travel")
+    DailyRate = st.number_input("Daily Rate")
+    Department = st.number_input("Department")
+    DistanceFromHome = st.number_input("Distance From Home")
+    Education = st.number_input("Education")
+    EducationField = st.number_input("Education Field")
+    EmployeeCount = st.number_input("Employee Count")
+    EmployeeNumber = st.number_input("Employee Number")
+    EnvironmentSatisfaction = st.number_input("Environment Satisfaction")
+    Gender = st.selectbox('Gender',('Male','Female'))
+    if Gender == 'Male':
+          Gender = 0
+    elif Gender == 'Female':
+          Gender = 1      
+    HourlyRate = st.number_input("Hourly Rate")
+    JobInvolvement = st.number_input("Job Involvement")
+    JobLevel = st.number_input("Job Level")
+    JobRole = st.number_input("Job Role")
+    JobSatisfaction = st.number_input("JobSatisfaction")
+    MaritalStatus = st.number_input("Marital Status")
+    MonthlyIncome = st.number_input("Monthly Income")
+    MonthlyRate = st.number_input("Monthly Rate")
+    NumCompaniesWorked = st.number_input("Number of Companies in you worked")
+    Over18 = st.number_input("Over 18 age")
+    OverTime = st.number_input("Overtime hours")
+    PercentSalaryHike = st.number_input("Percent Salary Hike")
+    PerformanceRating = st.number_input("Performance Rating")
+    RelationshipSatisfaction = st.number_input("Relationship Satisfaction")
+    StandardHours = st.number_input("Standard Hours")
+    StockOptionLevel = st.number_input("Stock Option Level")
+    TotalWorkingYears = st.number_input("Total Working Years")
+    TrainingTimesLastYear = st.number_input("Training Times Last Year")
+    WorkLifeBalance = st.number_input("Work Life Balance")
+    YearsAtCompany = st.number_input("Years At Company")
+    YearsInCurrentRole = st.number_input("Years In Current Role")
+    YearsSinceLastPromotion = st.number_input("Years Since Last Promotion")
+    YearsWithCurrManager = st.number_input("Years With Current Manager")
+    
+ 
+    # Create a feature list from the user inputs
+    features = [[Age,BusinessTravel,DailyRate,Department,DistanceFromHome,Education,EducationField,EmployeeCount,EmployeeNumber,EnvironmentSatisfaction,Gender,HourlyRate,JobInvolvement,JobLevel,JobRole,JobSatisfaction,MaritalStatus,MonthlyIncome,MonthlyRate,NumCompaniesWorked,Over18,OverTime,PercentSalaryHike,PerformanceRating,RelationshipSatisfaction,StandardHours,StockOptionLevel,TotalWorkingYears,TrainingTimesLastYear,WorkLifeBalance,YearsAtCompany,YearsInCurrentRole,YearsSinceLastPromotion,YearsWithCurrManager]]
+    
+    # Load the model
+    model = load_model()
+    
+    # Make a prediction when the user clicks the "Predict" button
+    if st.button('Predict Attrition'):
+        predicted_value = model_prediction(model, features)
+        if predicted_value == 0:
+            st.success(f"The Employee will leave the company")
+        else:
+            st.success(f"The Employee will not leave the company")        
+
+
+def about_hidevs():
+
+        components.html("""
+        <div>
+        <h4>🚀 Unlock Your Dream Job with HiDevs Community!</h4>
+        <p class="subtitle">🔍 Seeking the perfect job? HiDevs Community is your gateway to career success in the tech industry. Explore free expert courses, job-seeking support, and career transformation tips.</p>
+        <p class="subtitle">💼 We offer an upskill program in <b>Gen AI, Data Science, Machine Learning</b>, and assist startups in adopting <b>Gen AI</b> at minimal development costs.</p>
+        <p class="subtitle">🆓 Best of all, everything we offer is <b>completely free</b>! We are dedicated to helping society.</p>
+        <p class="subtitle">Book free of cost 1:1 mentorship on any topic of your choice — <a class="link" href="https://topmate.io/deepakchawla1307">topmate</a></p>
+        <p class="subtitle">✨ We dedicate over 30 minutes to each applicant’s resume, LinkedIn profile, mock interview, and upskill program. If you’d like our guidance, check out our services <a class="link" href="https://hidevscommunity.wixsite.com/hidevs">here</a></p>
+        <p class="subtitle">💡 Join us now, and turbocharge your career!</p>
+        <p class="subtitle"><a class="link" href="https://hidevscommunity.wixsite.com/hidevs" target="__blank">Website</a>
+        <a class="link" href="https://www.youtube.com/@HidevsCommunity1307/" target="__blank">YouTube</a>
+        <a class="link" href="https://www.instagram.com/hidevs_community/" target="__blank">Instagram</a>
+        <a class="link" href="https://medium.com/@hidevscommunity" target="__blank">Medium</a>
+        <a class="link" href="https://www.linkedin.com/company/hidevs-community/" target="__blank">LinkedIn</a>
+        <a class="link" href="https://github.com/hidevscommunity" target="__blank">GitHub</a></p>
+      </div>
+        """,
+                    height=600)
+
+def main():
+
+        # Set the app title and add your website name and logo
+        st.set_page_config(
+        page_title="Employee Attrition Prediction",
+        page_icon=":chart_with_upwards_trend:",
+        )
+    
+        st.title("Welcome to our Employee Attrition Prediction App!")
+    
+        app_design()    
+        st.header("About HiDevs Community")
+        about_hidevs()
+
+if __name__ == '__main__':
+        main()