init

app.py

@@ -0,0 +1,18 @@
+import os
+from dataset_loader import load_dataset
+
+if not os.path.isdir('agent_function'):
+    load_dataset()
+
+import gradio as gr
+from agent_function.agent import Agent
+
+iface = gr.Interface(
+    fn=Agent().ask,
+    inputs=gr.components.Textbox(lines=7, label="Enter your text"),
+    outputs="text",
+    title="Savoir Group E-commerce Stock Inquiry Demo",
+    description="""This is a demo bot for Savoir Group."""
+    )
+
+iface.launch(share=False)

dataset_loader.py

@@ -0,0 +1,16 @@
+from huggingface_hub import Repository
+from huggingface_hub import login
+import os
+
+def load_dataset():
+    login(token = os.environ['HUB_TOKEN'])
+
+    repo = Repository(
+        local_dir="agent_function",
+        repo_type="dataset",
+        clone_from="gheng/savoir_group",
+        token=True
+    )
+    repo.git_pull()
+
+load_dataset()