Update ui.py

ui.py

@@ -1,4 +1,25 @@
 import streamlit as st 
+import requests
+import os 
+
+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=""
+
+
+InitSession()
+
+
+
 
 def APP():
 
@@ -43,15 +64,18 @@ def APP():
             with tab1:
                 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 ")
 
 
-                    if uid and 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:
-                        user_input = st.text_area(
+                        bio_input = st.text_area(
                             "Protein Engineering Query",
                             placeholder="Type your query here."
                         )
@@ -59,22 +83,46 @@ def APP():
                     
                         if execute_button:
 
-                            st.session_state.projectname = project_name
-                            output=None
-                            if output:
-                                with console_container:
-                                    st.code(f"> {output}", language="rust")
+                            st.session_state.projectname =project_name
+                            st.session_state.username=uid 
+                            
+                            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={
+                    
+                                "Content-Type":"application/json",
+                                "Authorization":f"Bearer {tok}"
+                                })
+                            
+                            plan_response=response.json()
+                            if plan_response.get("status")=="active":
+                                
+                                
+                                    st.code(f"> operation execution successfull", language="rust")
                                     st.success("Task completed!")
                             else:
-                                with console_container:
                                     st.warning(">>>Error")
 
 
             with tab2:
-                st.markdown("### Operations")
-    
+                st.markdown("### newMATTER Bio Lab Operations")
+                response=requests.get(f"https://thexforce-combat-backend.hf.space/user/operations/{user_id}",headers={
+                    
+                    "Content-Type":"application/json",
+                    "Authorization":f"Bearer {tok}"
+                })
+                useroperations_json=response.json()
+                with st.expander("operations"):
+                    st.json(useroperations_json)
+                
+                
             with tab3:
-                st.markdown("### Output")
+                st.markdown("### newMATTER Bio Lab Outputs")
 
             
         SHOWTABS()