added spider dataset

app.py

@@ -1,10 +1,12 @@
 import os
 import gradio as gr
 from huggingface_hub import InferenceClient
+from datasets import load_dataset
 # from transformers import AutoTokenizer, AutoModelForCausalLM
 # import torch
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
+spider_data = load_dataset("spider")
 
 # ------------------------
 # API Clients
@@ -99,6 +101,12 @@ with gr.Blocks() as demo:
     chat_display = gr.Textbox(label="Chat History", lines=20, interactive=False)
     prompt_input = gr.Textbox(label="Your Prompt", lines=3, placeholder="Ask your BI question...")
 
+    random_index = 0
+    if spider_data:
+        spider_prompt = spider_data[random_index]["question"]
+        chat_memory = []
+        chat_display.update(f"👤 User: {spider_prompt}")
+
     run_button = gr.Button("Send")
 
     # Hidden chat history state

requirements.txt

@@ -3,4 +3,6 @@ torch>=2.2.0
 gradio>=4.24.0
 huggingface_hub>=0.22.2
 sqlalchemy>=2.0.41
-pymssql>=2.3.4
+pymssql>=2.3.4
+datasets>=4.0.0
+jsonlines>=4.0.0