add buttons with suggested prompts

app.py

@@ -18,6 +18,7 @@ from dotenv import load_dotenv
 
 load_dotenv()
 ss = st.session_state
+ss.prompt = None
 
 icon = Image.open('assets/logo0.png')
 st.set_page_config(
@@ -59,6 +60,29 @@ def on_api_key_change():
     if not oai_key_isvalid(api_key):
         st.write("Please input a valid OpenAI API key.")
 
+pre_prompts = [
+    'What is the molecular weight of sugar',
+    'Can I safely mix caffeine and sodium hydroxide?',
+    'How is safinamide synthesized?',
+    'How similar is morphine to heroin?'
+]
+
+
+def run_prompt(prompt):
+
+    st.chat_message("user").write(prompt)
+    with st.chat_message("assistant"):
+        st_callback = StreamlitCallbackHandlerChem(
+            st.container(),
+            max_thought_containers = 4,
+            collapse_completed_thoughts = False,
+            output_placeholder=ss
+        )
+        with wandb_tracing_enabled():
+            response = agent.run(prompt, callbacks=[st_callback])
+            st.write(response)
+
+
 # sidebar
 with st.sidebar:
     chemcrow_logo = Image.open('assets/chemcrow-logo-bold-new.png')
@@ -74,28 +98,42 @@ with st.sidebar:
         label_visibility="collapsed"
     )
 
+    # Display prompt examples
+    st.markdown('# What can I ask?')
+    cols = st.columns(2)
+    with cols[0]:
+        st.button(
+            pre_prompts[0],
+            on_click=lambda: run_prompt(pre_prompts[0]),
+        )
+        st.button(
+            pre_prompts[1],
+            on_click=lambda: run_prompt(pre_prompts[1]),
+        )
+    with cols[1]:
+        st.button(
+            pre_prompts[2],
+            on_click=lambda: run_prompt(pre_prompts[2]),
+        )
+        st.button(
+            pre_prompts[3],
+            on_click=lambda: run_prompt(pre_prompts[3]),
+        )
+
+
     # Display available tools
-    st.markdown(f"# Available tools: {len(tool_list)}")
+    st.markdown(f"# {len(tool_list)} available tools")
     st.dataframe(
         tool_list,
         use_container_width=True,
         hide_index=True,
-        height=300
+        height=200
     )
 
 
-# Agent execution
-if prompt := st.chat_input():
-    st.chat_message("user").write(prompt)
-    with st.chat_message("assistant"):
-        st_callback = StreamlitCallbackHandlerChem(
-            st.container(),
-            max_thought_containers = 4,
-            collapse_completed_thoughts = False,
-            output_placeholder=st.session_state
-        )
-        with wandb_tracing_enabled():
-            response = agent.run(prompt, callbacks=[st_callback])
-            st.write(response)
 
+prompt = None
 
+# Determine the prompt to use
+if user_input := st.chat_input():
+    run_prompt(user_input)

requirements.txt

@@ -8,3 +8,5 @@ wandb
 chemcrow==0.3.11
 pydantic==1.10.7
 langchain==0.0.275
+typing-inspect==0.8.0
+typing_extensions==4.5.0