update app

app.py

@@ -8,6 +8,8 @@ import logging
 import json
 import gradio as gr
 import pandas as pd
+from datasets import load_dataset
+import random
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -18,6 +20,9 @@ co = cohere.Client(api_key=os.environ.get("COHERE_API_KEY"))
 sentence_model = SentenceTransformer('all-MiniLM-L6-v2')
 qa_pipeline = pipeline("question-answering", model="distilbert-base-cased-distilled-squad")
 
+# Load the dataset
+dataset = load_dataset("serhany/scaling-qa")
+
 # Define sample inputs
 samples = [
     {
@@ -166,10 +171,16 @@ def gradio_interface(context: str, answer: str) -> Tuple[pd.DataFrame, pd.DataFr
 def use_sample(sample_index: int) -> Tuple[str, str]:
     return samples[sample_index]["context"], samples[sample_index]["answer"]
 
+def get_random_entry():
+    # Get a random entry from the dataset
+    random_index = random.randint(0, len(dataset['train']) - 1)
+    entry = dataset['train'][random_index]
+    return entry['context'], entry['answer']
+
 # Create Gradio interface with improved layout and sample buttons
 with gr.Blocks(theme=gr.themes.Default()) as iface:
     gr.Markdown("# Question Generator and Ranker")
-    gr.Markdown("Enter a context and an answer to generate and rank questions, or use one of the sample inputs.")
+    gr.Markdown("Enter a context and an answer to generate and rank questions, use one of the sample inputs, or get a random entry from the dataset.")
     
     with gr.Row():
         with gr.Column(scale=1):
@@ -179,6 +190,7 @@ with gr.Blocks(theme=gr.themes.Default()) as iface:
             
             with gr.Row():
                 sample_buttons = [gr.Button(f"Sample {i+1}") for i in range(3)]
+                random_button = gr.Button("Random Dataset Entry")
         
         with gr.Column(scale=2):
             best_question_output = gr.Textbox(label="Best Question")
@@ -216,5 +228,11 @@ with gr.Blocks(theme=gr.themes.Default()) as iface:
             outputs=[context_input, answer_input]
         )
 
+    # Set up random button functionality
+    random_button.click(
+        fn=get_random_entry,
+        outputs=[context_input, answer_input]
+    )
+
 # Launch the app
 iface.launch()