Upload ScienceWorld demo

README.md

@@ -1,10 +1,10 @@
 ---
 title: ScienceWorld
-emoji: 💩
+emoji: 👨‍🔬🔬
 colorFrom: pink
 colorTo: green
 sdk: streamlit
-sdk_version: 1.9.0
+sdk_version: 1.10.0
 app_file: app.py
 pinned: false
 license: apache-2.0

app.py

@@ -0,0 +1,89 @@
+import streamlit as st
+import streamlit.components.v1 as components
+
+from scienceworld import ScienceWorldEnv
+
+description = """
+[Project Page](https://sciworld.apps.allenai.org) | [ArXiv Paper](https://arxiv.org/abs/2203.07540) | [Github Repo](https://github.com/allenai/ScienceWorld)
+"""
+st.title("ScienceWorld Demo")
+st.markdown(description)
+
+env = st.session_state.get("env")
+if env is None:
+    env = ScienceWorldEnv("")
+    st.session_state["env"] = env
+
+seed = st.session_state.get("seed")
+obs = st.session_state.get("obs")
+infos = st.session_state.get("infos")
+history = st.session_state.get("history")
+if history is None:
+    history = []
+    st.session_state["history"] = history
+
+def clear_history():
+    history.clear()
+
+
+with st.sidebar:
+    st.title("ScienceWorld Demo")
+    st.markdown(description)
+    task = st.selectbox("Task:", env.getTaskNames(), on_change=clear_history)
+
+if len(history) == 0:
+    env.load(task, 0, "")
+    obs, infos = env.reset()
+    st.session_state["obs"] = obs
+    st.session_state["infos"] = infos
+    history.append(("", env.getTaskDescription()))
+    history.append(("look around", obs))
+
+def step():
+    act = st.session_state.action
+    if act:
+        obs, reward, done, infos = env.step(act)
+        history.append((act, obs))
+        st.session_state["obs"] = obs
+        st.session_state["infos"] = infos
+
+        if act == "reset":
+            clear_history()
+
+
+with st.sidebar:
+    st.warning(env.getTaskDescription())
+    st.success(f"Score:  {infos['score']}")
+
+    valid_actions = [""] + sorted(infos["valid"])
+    if infos['score'] == 100:
+        valid_actions = ["", "reset"]
+
+    act = st.selectbox('Action:', options=valid_actions, index=0, on_change=step, key="action")
+
+for act, obs in history:
+    if act:
+        st.write("> " + act)
+
+    if obs:
+        st.info(obs.replace('\n\t', '\n- '))
+
+
+if infos['score'] == 100:
+    with st.sidebar:
+        st.balloons()
+
+    st.success("Congratulations! You have completed the task.")
+
+
+
+# Auto scroll at the bottom of the page.
+components.html(
+f"""
+    <p>{st.session_state.obs}</p>
+    <script>
+        window.parent.document.querySelector('section.main').scrollTo(0, window.parent.document.querySelector('section.main').scrollHeight);
+    </script>
+""",
+height=0
+)

packages.txt

@@ -0,0 +1,2 @@
+default-jre
+default-jdk

requirements.txt

@@ -1,3 +1,2 @@
-textworld
+py4j
 scienceworld
-jericho