from utlis.helper import *
import sqlite3
import hashlib
import io



initialize_session_state()

with st.sidebar:
    st.image("logo.png", width=170)
    st.title("Config Settings")
    # Get List of models
   
    st.session_state.genre = st.radio(
    "Choose option",
    [ "Select Service", "Add service"])#,"Delete service"])
    
    if st.session_state.genre=="Add service":
        st.title('Add service')
        # Check service status
        # Get all available services
        add_new_service = st.checkbox("Add new service")
        if add_new_service:
            new_service = st.text_input("Enter service name")
            # Get list of Embedding models
            
            if  new_service and st.button('Add'):
                add_service(st.session_state.token,new_service)
        data = {"token": st.session_state.token}
        json_data = json.dumps(data)
        headers = {'Content-Type': 'application/json'}
        services  = requests.get(SERVICES_API,data=json_data, headers=headers)
        services =json.loads(services.text)
        if len(services)>0:
            st.session_state.service = st.selectbox("Choose Service",services)

            st.session_state.uploaded_files = st.file_uploader("Upload CVs",  type=["pdf","doc"], accept_multiple_files=True)
            if st.session_state.uploaded_files:
                st.session_state.process = st.button('Process')
                if st.session_state.process:
                    with st.spinner("Processing..."):
                        add_document(st.session_state.token,st.session_state.service)

    elif st.session_state.genre=="Select Service":
        st.title('Choose Service')
        data = {"token": st.session_state.token}
        json_data = json.dumps(data)
        headers = {'Content-Type': 'application/json'}
        services  = requests.get(SERVICES_API,data=json_data, headers=headers)
        services =json.loads(services.text)

        if len(services)>0:
            st.session_state.service_slected_to_chat = st.selectbox(" ",services)
            st.session_state.top_k=st.slider("Number of Candidates", min_value=1, max_value=10, value=3, step=1)
            # data = {"token": st.session_state.token, "servicename": st.session_state.service_slected_to_chat}
            # json_data = json.dumps(data)
            # headers = {'Content-Type': 'application/json'}
            # history_document  = requests.get(DOCUMENT_API,data=json_data, headers=headers)
            # history_document =json.loads(history_document.text).get("documents",[])
            # history_document = [doc["documentname"] for doc in history_document]
            

            
    # elif st.session_state.genre == "Delete service":
    #     st.title('Delete Service')
    #     data = {"token": st.session_state.token}
    #     json_data = json.dumps(data)
    #     headers = {'Content-Type': 'application/json'}
    #     services  = requests.get(SERVICES_API,data=json_data, headers=headers)
    #     services =json.loads(services.text)
    #     if len(services)>=2:
    #         services.append("ALL")
    #         # Get list of documents from histrory
    #     if "ALL" in services:
    #         service_slected = st.multiselect(
    #                 "",services ,default="ALL"
    #                 )
    #     elif len(services)==1:
    #         service_slected = st.multiselect(
    #                 "",services,default=services[0]
    #                 )
    #     else:
    #         service_slected = st.multiselect(
    #                 "",services
    #                 )
    #     if "ALL" in service_slected:
    #         service_slected = services
    #         service_slected.remove("ALL")
    #     st.write("You selected:", service_slected)

    #     if len(service_slected) > 0:
    #         st.session_state.delete = st.button('Delete')
    #         if st.session_state.delete:
    #             delete_service(st.session_state.token ,service_slected)
        
    # elif st.session_state.genre == "Delete CV(s)":
    #     st.title('Delete CV(s)')
    #     data = {"token": st.session_state.token}
    #     json_data = json.dumps(data)
    #     headers = {'Content-Type': 'application/json'}
    #     services  = requests.get(SERVICES_API,data=json_data, headers=headers)
    #     services =json.loads(services.text)
    #     if len(services)>0:
    #         service = st.selectbox("Choose Service",services)
    #         data = {"token": st.session_state.token, "servicename": service}
    #         json_data = json.dumps(data)
    #         headers = {'Content-Type': 'application/json'}
        

    #         st.write("You selected:", document_slected_to_delete)
    #         if len(document_slected_to_delete) > 0:
    #             st.session_state.delete = st.button('Delete')
    #             if st.session_state.delete:
    #                 delete_document(st.session_state.token,st.session_state.service ,document_slected_to_delete)

# css_style = """
# <style>
# .title {
#     white-space: nowrap;
# }
# </style>
# """

# st.markdown(css_style, unsafe_allow_html=True)
st.markdown("""
    <style>
    .st-bm {
        color: #1E90FF; /* DodgerBlue color */
    }
            .card {
    border-radius: 10px;
    box-shadow: 0 2px 4px rgba(0,0,0,0.1);
    padding: 15px;
    margin: 10px 0;
    transition: box-shadow 0.3s ease-in-out;
}
.card:hover {
    box-shadow: 0 4px 8px rgba(0,0,0,0.2);
}
.highlight {
    color:  #1E90FF; /* Custom color */
}
    </style>
    """, unsafe_allow_html=True)

with st.container():
    st.markdown('<h1 class="title">SmartHire Matcher</h1>', unsafe_allow_html=True)
    col1, col2 = st.columns([3, 1])
    if st.session_state.genre=="Select Service" and st.session_state.service_slected_to_chat:
        query = st.text_area("Add description of your offer:", height=300)
        
        if query and st.button('Process') :
            with st.spinner("Finding Matching CVs..."):
                results = search_document( index_name= "cvindex",token= "abcd",service_name= st.session_state.service_slected_to_chat,query= query,top_k= st.session_state.top_k)
                st.session_state.results_str = results.decode('utf-8')

            # Displaying results
        try:
            if st.session_state.results_str:  # check if there are results
                results = json.loads(st.session_state.results_str)

                for index, item in enumerate(results):
                    with st.container():
                        col1, col2 = st.columns([3, 1])
                        with col1:
                            st.markdown(f"<div class='card'><h3><span class='highlight'>Top:</span> {index+1}</h3><p><span class='highlight'>Score:</span> {round(item['score']*100, 3)}%</p><p><span class='highlight'>Document Name:</span> {item['documentname'].replace('_pdf', ' ')}</p></div>", unsafe_allow_html=True)

                            #st.markdown(f"**Tag:** {item['tag']}")
                            #st.markdown(f"**Score:** {round(item['score'], 3)}")
                            #st.markdown(f"**Document Name:** {item['documentname'].replace('_pdf', ' ')}")
                        with col2:
                            b64_pdf = item['encoded_cv']
                            pdf = base64.b64decode(b64_pdf)
                            pdf_file = io.BytesIO(pdf)
                            st.download_button("Download CV", data=pdf_file, file_name=item['documentname'].replace('_', '.'), mime='application/pdf', key=f"download_{index}_{item['documentname']}")
                if not results:
                    st.error("No results found.")
        except Exception as e:
            st.error("Failed to load results. Please try again later."+ str(e))