adding app

.gitignore

@@ -0,0 +1,2 @@
+venv/
+*.pyc

README.md

@@ -5,6 +5,7 @@ colorFrom: yellow
 colorTo: purple
 sdk: streamlit
 sdk_version: 1.17.0
+python_version: 3.10.6
 app_file: app.py
 pinned: false
 ---

app.py

@@ -0,0 +1,43 @@
+import pandas as pd
+from pathlib import Path
+import requests
+import streamlit as st
+from langchain.agents import create_pandas_dataframe_agent
+from langchain.llms import OpenAI
+from contextlib import contextmanager, redirect_stdout
+from io import StringIO
+from time import sleep
+
+data_path = Path('umpire-full-text.csv')
+if not data_path.exists():
+    r = requests.get('https://upenn.box.com/shared/static/dyxc1heqrfrgp22ntwpet3f57sir34c3.csv')
+    data_path.write_bytes(r.content)
+
+@contextmanager
+def st_capture(output_func):
+    with StringIO() as stdout, redirect_stdout(stdout):
+        old_write = stdout.write
+
+        def new_write(string):
+            ret = old_write(string)
+            output_func(stdout.getvalue())
+            return ret
+        
+        stdout.write = new_write
+        yield
+
+
+df = pd.read_csv('umpire-full-text.csv')
+agent = create_pandas_dataframe_agent(OpenAI(temperature=0), df, verbose=True)
+
+st.dataframe(df)
+
+query = st.text_input('Enter query here:', '')
+answer = st.empty()
+
+if query:
+    output = st.empty()
+    with st_capture(output.info):
+        response = agent.run(query)
+        answer = st.write(response)
+        

requirements.txt

@@ -0,0 +1,66 @@
+aiohttp==3.8.4
+aiosignal==1.3.1
+altair==4.2.2
+async-timeout==4.0.2
+attrs==22.2.0
+blinker==1.5
+cachetools==5.3.0
+certifi==2022.12.7
+charset-normalizer==3.1.0
+click==8.1.3
+dataclasses-json==0.5.7
+decorator==5.1.1
+entrypoints==0.4
+frozenlist==1.3.3
+gitdb==4.0.10
+GitPython==3.1.31
+greenlet==2.0.2
+idna==3.4
+importlib-metadata==6.1.0
+Jinja2==3.1.2
+jsonschema==4.17.3
+langchain==0.0.116
+markdown-it-py==2.2.0
+MarkupSafe==2.1.2
+marshmallow==3.19.0
+marshmallow-enum==1.5.1
+mdurl==0.1.2
+multidict==6.0.4
+mypy-extensions==1.0.0
+numpy==1.24.2
+openai==0.27.2
+packaging==23.0
+pandas==1.5.3
+Pillow==9.4.0
+protobuf==3.20.3
+pyarrow==11.0.0
+pydantic==1.10.6
+pydeck==0.8.0
+Pygments==2.14.0
+Pympler==1.0.1
+pyrsistent==0.19.3
+python-dateutil==2.8.2
+pytz==2022.7.1
+pytz-deprecation-shim==0.1.0.post0
+PyYAML==6.0
+requests==2.28.2
+rich==13.3.2
+semver==2.13.0
+six==1.16.0
+smmap==5.0.0
+SQLAlchemy==1.4.47
+streamlit==1.20.0
+tenacity==8.2.2
+toml==0.10.2
+toolz==0.12.0
+tornado==6.2
+tqdm==4.65.0
+typing-inspect==0.8.0
+typing_extensions==4.5.0
+tzdata==2022.7
+tzlocal==4.3
+urllib3==1.26.15
+validators==0.20.0
+watchdog==3.0.0
+yarl==1.8.2
+zipp==3.15.0