Use gdown python API to automatically download test data during launch

.gitignore

@@ -1,5 +1,4 @@
-secrets.toml
 venv/
 data/
 .DS_Store
-config.json
+config.json

README.md

@@ -20,14 +20,20 @@ pip install -r requirements.txt
 export OPENAI_API_KEY=<insert key here>
 ```
 
-3. Download test data:
+3. Create the `config.json` file and fill in the relevant info:
 ```
-gdown https://drive.google.com/drive/folders/1uDSAWtLvp1YPzfXUsK_v6DeWta16pq6y -O ./data/ --folder
+{
+    "CHATGPT_MODEL":"<insert model name>",
+    "OPENAI_API_BASE":"https://<insert-openai-service-name>.openai.azure.com",
+    "OPENAI_API_VERSION":"<insert version>",
+    "ENGINE": "<insert deployment model name>",
+    "ENGINE_EMBEDDING": "<insert deployment embedding name>"
+}
 ```
 
-3. Launch the app:
+4. Launch the app:
 ```
 streamlit run app.py
 ```
 
-You can then access the app in your browser at `http://localhost:8501`
+A Streamlit browser window should automatically open. If not, the app can be accessed at `http://localhost:8501`

app.py

@@ -2,6 +2,7 @@ import json
 import os
 
 import streamlit as st
+from gdown import download_folder
 from llama_index import ServiceContext
 from llama_index import SimpleDirectoryReader
 from llama_index import VectorStoreIndex
@@ -9,6 +10,7 @@ from llama_index import set_global_service_context
 from llama_index.embeddings import OpenAIEmbedding
 from llama_index.llms import AzureOpenAI
 
+
 # Initialize message history
 st.header("Chat with André's research 💬 📚")
 
@@ -20,6 +22,12 @@ with open(r"config.json") as config_file:
     config_details = json.load(config_file)
 
 
+def download_test_data():
+    url = "https://drive.google.com/drive/folders/1uDSAWtLvp1YPzfXUsK_v6DeWta16pq6y"
+    with st.spinner(text="Downloading test data. Might take a few seconds."):
+        download_folder(url, quiet=True, use_cookies=False, output="./data/")
+
+
 @st.cache_resource(show_spinner=False)
 def load_data():
     with st.spinner(text="Loading and indexing the provided dataset – hang tight! This may take a few seconds."):
@@ -53,6 +61,7 @@ def load_data():
 
 
 def main():
+    download_test_data()
     index = load_data()
     chat_engine = index.as_chat_engine(chat_mode="condense_question", verbose=True)