spider dataset work update 1

app.py

@@ -1,4 +1,5 @@
 import os
+import random
 import gradio as gr
 from huggingface_hub import InferenceClient
 from datasets import load_dataset
@@ -6,7 +7,11 @@ from datasets import load_dataset
 # import torch
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
-spider_data = load_dataset("spider")
+
+# ------------------------
+# Load Spider Dataset (Hugging Face Datasets)
+# ------------------------
+spider_dataset = load_dataset("spider", split="train")
 
 # ------------------------
 # API Clients
@@ -86,27 +91,30 @@ def run_model_with_history(model_name, user_input, chat_history):
 
     return chat_transcript, chat_history
 
+# ------------------------
+# Get Random Spider Question
+# ------------------------
+def get_random_spider_question():
+    sample = random.choice(spider_dataset)
+    return sample["question"]
+
 # ------------------------
 # Gradio UI
 # ------------------------
 with gr.Blocks() as demo:
     gr.Markdown("## 🧠 Generative AI Model Evaluation with Context")
 
-    model_choice = gr.Dropdown(
-        choices=["LLaMA 4", "Qwen3 14B"],  # "Mixtral 8x7B" removed
-        label="Select Model",
-        value="LLaMA 4"
-    )
+    with gr.Row():
+        model_choice = gr.Dropdown(
+            choices=["LLaMA 4", "Qwen3 14B"],
+            label="Select Model",
+            value="LLaMA 4"
+        )
+        load_spider_btn = gr.Button("🔀 Load Random Spider Prompt")
 
     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
@@ -117,6 +125,12 @@ with gr.Blocks() as demo:
         inputs=[model_choice, prompt_input, chat_memory],
         outputs=[chat_display, chat_memory]
     )
+    
+    load_spider_btn.click(
+        fn=get_random_spider_question,
+        inputs=[],
+        outputs=prompt_input
+    )
 
 # Launch app
 demo.launch()