update readme + cc ver

README.md

@@ -1,8 +1,8 @@
 ---
 title: ChemCrow
-emoji: 🏆
-colorFrom: pink
-colorTo: purple
+emoji: 🐦‍⬛
+colorFrom: green
+colorTo: black
 sdk: streamlit
 sdk_version: 1.25.0
 app_file: app.py

app.py

@@ -13,7 +13,7 @@ from langchain.chat_models import ChatOpenAI
 from PIL import Image
 from rmrkl import ChatZeroShotAgent, RetryAgentExecutor
 
-from chemcrow.agents import ChemCrow
+from chemcrow.agents import ChemCrow, make_tools
 from chemcrow.agents.prompts import (FORMAT_INSTRUCTIONS, QUESTION_PROMPT,
                                      SUFFIX)
 from chemcrow.frontend.streamlit_callback_handler import \
@@ -48,6 +48,18 @@ st.markdown(
 # Session state
 # st.session_state['molecule'] = "CCO"
 
+agent = ChemCrow(
+    model='gpt-4',
+    temp=0.1,
+    api_key=ss.get('api_key')
+).agent_executor
+
+tools = agent.tools
+
+tool_list = pd.Series(
+    {f"✅ {t.name}":t.description for t in tools}
+).reset_index()
+tool_list.columns = ['Tool', 'Description']
 
 # sidebar
 with st.sidebar:
@@ -59,33 +71,15 @@ with st.sidebar:
     st.text_input('OpenAI API key', type='password', key='api_key', label_visibility="collapsed")
 
     # Display available tools
-    #st.markdown(f"# Available tools: {len(tools)}")
-    #st.dataframe(
-    #    tool_list,
-    #    use_container_width=True,
-    #    hide_index=True,
-    #    height=300
-
-    #)
-
-
-agent = ChemCrow(
-    #tools,
-    model='gpt-4',
-    temp=0.1,
-    api_key=ss.get('api_key')
-).agent_executor
-
-
-#tool_list = pd.Series(
-#    {f"✅ {t.name}":t.description for t in tools}
-#).reset_index()
-#tool_list.columns = ['Tool', 'Description']
-
-
+    st.markdown(f"# Available tools: {len(tool_list)}")
+    st.dataframe(
+        tool_list,
+        use_container_width=True,
+        hide_index=True,
+        height=300
+    )
 
 
-print(st.session_state)
 # Agent execution
 if prompt := st.chat_input():
     st.chat_message("user").write(prompt)
@@ -96,11 +90,10 @@ if prompt := st.chat_input():
             collapse_completed_thoughts = False,
             output_placeholder=st.session_state
         )
-        #try:
-        # TODO Modify this, not taking callbacks 
-        response = agent.run(prompt, callbacks=[st_callback])
-        st.write(response)
-        #except:
-        #    st.write("Please input a valid OpenAI API key.")
+        try:
+            response = agent.run(prompt, callbacks=[st_callback])
+            st.write(response)
+        except:
+            st.write("Please input a valid OpenAI API key.")
 
 

requirements.txt

@@ -2,5 +2,5 @@ ipython
 paper-scraper @ git+https://github.com/blackadad/paper-scraper.git
 streamlit
 python-dotenv
-chemcrow==0.3.2
+chemcrow