Add wormneurodata tab

app.py

@@ -1,6 +1,8 @@
 import streamlit
 import streamlit as st
 
+#st.set_page_config(layout="wide")
+
 from ring import LLM_GPT35
 from ring import LLM_GPT4
 from ring import LLM_LLAMA2
@@ -23,7 +25,12 @@ with col1:
 with col2:
     st.image("images/OpenWormLogo.png")
 
-tab_free, tab_panel, tab_pubs = st.tabs(["Individual LLMs", "Panel discussion", "Publications"])
+
+tab_free, tab_panel, tab_pubs, tab_data, tab_model = st.tabs(["Individual LLMs", 
+                                         "Panel discussion", 
+                                         "Publications",
+                                         "Structured data",
+                                         "Run model"])
 
 
 with tab_free:
@@ -87,4 +94,45 @@ with tab_pubs:
             response = find_basis_paper(query=text, 
                                         result_limit=num)
 
+            st.info(response)
+
+
+with tab_data:
+
+    st.markdown("**Query structured datasets**")
+
+    with st.form("form_data"):
+
+        from datasources import DS_WORMNEUROATLAS
+        from datasources import FORMATS
+        from datasources import query_data_source
+
+        
+        text = st.text_area("Which neuron would you like to know about:", "AVBL")
+
+        source = st.selectbox('Which data source to use?', (DS_WORMNEUROATLAS,))
+        format = st.selectbox('Return format', FORMATS)
+
+        submitted = st.form_submit_button("Submit")
+
+        if submitted:
+            response = query_data_source(text, source, format)
+
+            st.info(response)
+
+with tab_model:
+
+    st.markdown("**Run a _C. elegans_ cell model**")
+
+    with st.form("form_model"):
+
+        from model import run_model
+
+        text = st.text_area("Current injection level:", "100pA")
+
+        submitted = st.form_submit_button("Submit")
+
+        if submitted:
+            response = run_model(text)
+
             st.info(response)

datasources.py

@@ -0,0 +1,44 @@
+# Based on: https://github.com/allenai/s2-folks/tree/main/examples/python
+# 
+# 
+
+import os
+import re
+
+FORMAT_TEXT = 'Text'
+FORMAT_DICT = 'Dict'
+FORMAT_JSON = 'JSON'
+
+FORMATS = (FORMAT_TEXT, FORMAT_DICT)
+
+DS_WORMNEUROATLAS = 'WormNeuroAtlas'
+
+import wormneuroatlas as wa
+import pprint
+
+pp = pprint.PrettyPrinter(depth=6)
+
+
+def query_data_source(text, source, format):
+
+    if source == DS_WORMNEUROATLAS:
+
+        atlas = wa.NeuroAtlas()
+        ds, ss = atlas.everything_about(text, return_values=True, return_text=True, print_text=False)
+
+        if format==FORMAT_TEXT:
+            info = '''Extracting info on neuron %s from %s...
+'''%(text, source)
+            info += """```%s```"""%ss
+        elif format==FORMAT_DICT:
+            info = """```
+%s```"""%pp.pformat(ds)
+        elif format==FORMAT_JSON:
+            import json
+            jj =  json.dumps(ds)
+            info = """```
+%s```"""%pp.pformat(jj)
+
+    return info
+
+

model.py

@@ -0,0 +1,17 @@
+# Based on: https://github.com/allenai/s2-folks/tree/main/examples/python
+# 
+# 
+
+import os
+import re
+
+
+def run_model(text):
+
+    info = 'Running models with: %s'%text
+
+    
+
+    return info
+
+

requirements.txt

@@ -3,4 +3,6 @@ openai
 langchain_openai
 llamaapi
 asyncio
-langchain_experimental
+langchain_experimental
+wormneuroatlas
+pprint