import streamlit as st 
import requests
import os 
#from auth import authenticator

tok=os.getenv("TOK")



def InitSession():
    
    if "username" not  in st.session_state:
        st.session_state.username=""

    if "projectname" not in st.session_state:
        st.session_state.projectname=""

    #if "loggedin" not in st.session_state:
     #   st.session_state.loggedin=False
        

#InitSession()




               

def APP():

        st.markdown("""
            <h2>new <i style='color:red'>MATTER</i> </h2>
            """,unsafe_allow_html=True)
            
        tab_map = {
                    
                    0: "BIO ENGINEERING LAB @newMATTER",
                    1: "EXECUTED OPERATIONS @newMATTER",
                    2: "LAB OUTPUT @newMATTER"
    
                    }

        tab_selection=st.pills(
            "TABS",
            options=tab_map.keys(),
            format_func=lambda option:tab_map[option],
            selection_mode="single"
            
        )
        
        def SHOWTABS():
            
            
            if tab_selection == 0 :


                option_map = {
                    
                    0: " @newMATTER Nanobody Engineering:",
                    1: "@newMATTER Vaccine Engineering:",
                    2: "Research",
                    3: "Information"
    
                    }

                selection = st.pills(
                    "BIOLOGICS",
                    options=option_map.keys(),
                    format_func=lambda option: option_map[option],
                    selection_mode="single",
                    )
                #match selection:
                if selection == 0:
                        st.markdown("<p style='color:white;background-color:orange;font-weight:bold'> Nanobody [CANCER targeted]</p>",unsafe_allow_html=True)
                        #with st.expander("info"):
                            #st.info("This Interface lets u specify a high level biological query (Protein Engineering Query) and execute the pipeline for the end product i.e Engineered Nanobody",icon=":material/info:")
                        
                        with st.form("bio",border=False):

                           
                            
                            project_name=None
                            target=None
                            
                            
                            with st.expander("project settings",icon=":material/settings:"):
                                project_name=st.text_input("enter project name ")
                                target=st.text_input("Set up the target antigen ")
                            
                           
                                                        
                    
                            bio_input = st.text_area(
                                "Protein Engineering Query",
                                placeholder="Type your query here."
                            )
                            execute_button=st.form_submit_button("execute")
                    
                            if execute_button:

                                
                                    ##st.session_state.projectname = project_name
                                    #st.session_state.username = uid 
                                    if not project_name or not target:
                                        st.markdown(":orange-badge[⚠️ Set Projectname and Target Antigen before proceeding]")
                                    else:
                                        
                                        payload={
                                            "uid":st.user.email ,
                                            "pid":project_name,
                                            "target":target,
                                            "high_level_bio_query":bio_input
                                        }

                            
                                        response=requests.post("https://thexforce-combat-backend.hf.space/bio_context_language_plan",json=payload,headers={
                    
                                            "Content-Type":"application/json",
                                            "Authorization":f"Bearer {tok}"
                                            })
                            
                                        plan_response=response.json()
                                        if plan_response.get("status")=="active":
                                
                                
                                            st.markdown(f"## {plan_response.get('plan')}")
                                        else:
                                            st.warning(">>>Error")

                                    #else:
                                       # st.error("Please fill in both username and project name before submitting")


                if selection == 1:
                        st.markdown("<p style='color:white;background-color:orange;font-weight:bold'>Vaccine [Supported] </p>",unsafe_allow_html=True)
                        with st.expander("info"):
                            st.info("This Interface lets u specify a high level biological query  and execute the pipeline for the end product i.e Vaccine",icon=":material/info:")
                        st.code("coming soon")

                if selection ==2:
                    st.markdown("<p style='color:white;background-color:orange;font-weight:bold'>Operation Details </p>",unsafe_allow_html=True)

                if selection==3:
                    st.markdown("<p style='color:white;background-color:orange;font-weight:bold'> This system is running in trial phase </p>",unsafe_allow_html=True)
                    
                        

            
            if tab_selection == 1:
                st.markdown("### newMATTER Bio Lab Operations")

                @st.cache_data(ttl=10)
                def fetch_ops():
                    response=requests.get(f"https://thexforce-combat-backend.hf.space/user/operations/{st.user.email}",headers={
                    
                        "Content-Type":"application/json",
                        "Authorization":f"Bearer {tok}"
                    })
                    useroperations_json=response.json()
                    return useroperations_json

                    

                userops=fetch_ops()
                    
                with st.expander("operations"):
                    st.json(userops)
                
                
            if tab_selection == 2:
                st.markdown("### newMATTER Bio Lab Outputs")
                projectname=st.text_input("projectname to look the results for ")
                if st.button("lookup"):
                    
                    response=requests.get(
                        
                        f"https://thexforce-combat-backend.hf.space/{st.session_state.get('username')}/{projectname}/individual/experiment",
                        headers={
                    
                        "Content-Type":"application/json",
                        "Authorization":f"Bearer {tok}"
                    })
                    
                    ie=response.json()
                    st.json(ie)




    
    
        if st.user.is_logged_in:
            
            if st.button("🚪 Logout"):
                st.logout()
                st.rerun()
    
            #Then show the main UI
            SHOWTABS()
        else:
            #Show login when NOT logged in
            st.info("Please log in to access the Bio Lab")
            if st.button("Log in"):
                st.login("auth0")
                st.stop()