Update ui.py

ui.py

@@ -119,13 +119,14 @@ def APP():
                                 projectname=p_name
                                 
                     
-                            bio_input = st.text_area(
-                                "Protein Engineering Query",
-                                placeholder="Type your query here."
-                            )
-                            execute_button=st.form_submit_button("execute")
-
-
+                            #bio_input = st.text_area(
+                             #   "Protein Engineering Query",
+                              #  placeholder="Type your query here."
+                            #)
+                            #execute_button=st.form_submit_button("execute")
+                            bio_input = st.chat_input(" Ready for Action ! ")
+                            
+                            
 
 
                             """----------------experimental--------------------"""
@@ -140,14 +141,23 @@ def APP():
                                 useroperations_json=response.json()
                                 return useroperations_json
 
-                            fetch_ops()
-                            """---------------experimental---------------------"""
                             
+                            """---------------experimental-----------------------"""
+
+                            with st.container(border=True):
+                                if len(st.session_state.messages) > 0:
+                                    for msg in st.session_state.messages:
+                                        with st.chat_message(msg["role"]):
+                                            st.markdown(msg["content"])
                     
-                            if execute_button:
+                            if bio_input:
+
+                                    st.session_state.messages.append({"role":"user","content":bio_input})
+                                    with st.chat_message("user"):
+                                        st.markdown(user_input)
 
                                     if projectname == None:
-                                        st.markdown(":orange-badge[⚠️ Set Projectname]")
+                                        st.markdown(":orange-badge[⚠️Set Projectname]")
                                     else:
                                         
                                         payload={
@@ -167,8 +177,13 @@ def APP():
                                         plan_response=response.json()
                                         
                                 
-                                
-                                        st.markdown(f"## {plan_response}")
+                                        #st.markdown(f"## {plan_response}")
+                                        with st.chat_message("assistant"):
+                                            st.markdown(plan_response)
+
+                                        
+                                            
+                                        st.session_state.messages.append({"role":"assistant","content":plan_response})
 
 
                 if selection == 1: