Update ui.py

ui.py

@@ -62,56 +62,81 @@ def APP():
             st.markdown('<div class="blinking-text"> CAUTION ! </div>', unsafe_allow_html=True)
             
             with tab1:
-                with st.form("bio",border=False):
 
-                    uid=None
-                    project_name=None
+
+                option_map = {
+                    0: ":material/pill:",
+                    1: ":material/vaccines:",
+    
+                    }
+
+                selection = st.pills(
+                    "BIOLOGICS",
+                    options=option_map.keys(),
+                    format_func=lambda option: option_map[option],
+                    selection_mode="single",
+                    )
+                match selection:
+                    case 0:
+                        st.markdown("<p style='color:white;background-color:orange'> Nanobody [CANCER targeted]</p>",unsafe_allow_html=True)
+                        with st.form("bio",border=False):
+
+                            uid=None
+                            project_name=None
                     
-                    with st.expander("setup",icon=":material/settings:"):
-                        uid=st.text_input("enter username")
-                        project_name=st.text_input("enter project name ")
+                            with st.expander("setup",icon=":material/settings:"):
+                                uid=st.text_input("enter username")
+                                project_name=st.text_input("enter project name ")
 
 
-                    if not uid or not project_name:
-                        st.markdown(":orange-badge[⚠️ Set Username and Projectname]")
+                            if not uid or not project_name:
+                                st.markdown(":orange-badge[⚠️ Set Username and Projectname]")
 
                         
-                    #else:
-                    bio_input = st.text_area(
-                            "Protein Engineering Query",
-                            placeholder="Type your query here."
-                        )
-                    execute_button=st.form_submit_button("execute")
                     
-                    if execute_button:
-                            if uid and project_name:  # Only process if fields are filled
-                                st.session_state.projectname = project_name
-                                st.session_state.username = uid 
+                            bio_input = st.text_area(
+                                "Protein Engineering Query",
+                                placeholder="Type your query here."
+                            )
+                            execute_button=st.form_submit_button("execute")
+                    
+                            if execute_button:
+                                if uid and project_name:  # Only process if fields are filled
+                                    st.session_state.projectname = project_name
+                                    st.session_state.username = uid 
                             
-                                payload={
-                                    "uid":uid,
-                                    "pid":project_name,
-                                    "high_level_bio_query":bio_input
-                                }
+                                    payload={
+                                        "uid":uid,
+                                        "pid":project_name,
+                                        "high_level_bio_query":bio_input
+                                    }
 
                             
-                                response=requests.post("https://thexforce-combat-backend.hf.space/bio_context_language_plan",json=payload,headers={
+                                    response=requests.post("https://thexforce-combat-backend.hf.space/bio_context_language_plan",json=payload,headers={
                     
-                                    "Content-Type":"application/json",
-                                    "Authorization":f"Bearer {tok}"
-                                    })
+                                        "Content-Type":"application/json",
+                                        "Authorization":f"Bearer {tok}"
+                                        })
                             
-                                plan_response=response.json()
-                                if plan_response.get("status")=="active":
+                                    plan_response=response.json()
+                                    if plan_response.get("status")=="active":
                                 
                                 
-                                    st.code(f"> operation execution successfull", language="rust")
-                                    st.success("Task completed!")
-                                else:
-                                    st.warning(">>>Error")
+                                        st.code(f"> operation execution successfull", language="rust")
+                                        st.success("Task completed!")
+                                    else:
+                                        st.warning(">>>Error")
 
-                            else:
+                                else:
                                     st.error("Please fill in both username and project name before submitting")
+
+
+                    case 1:
+                        st.markdown("<p style='color:white;background-color:orange'>Vaccine [Supported] </p>",unsafe_allow_html=True)
+                        st.code("coming soon")
+                        
+
+            
             with tab2:
                 st.markdown("### newMATTER Bio Lab Operations")
                 response=requests.get(f"https://thexforce-combat-backend.hf.space/user/operations/{uid}",headers={