Update app.py

app.py

@@ -2,8 +2,16 @@ import gradio as gr
 import pandas as pd
 from utils.dataset_loader import load_dataset
 
-# Load dataset
-df = load_dataset()
+# Function to safely load dataset
+def load_and_validate_dataset():
+    df = load_dataset()
+    if df is not None and "prompt_text" in df.columns:
+        return df
+    else:
+        return None
+
+# Load dataset with validation
+df = load_and_validate_dataset()
 
 # Function to simulate conversation with model selection
 def chat_interface(user_input, selected_model, prompt_id=None):
@@ -34,15 +42,18 @@ with gr.Blocks(css="./static/styles.css") as demo:
             value=models[0],
             elem_classes="retro-terminal"
         )
-        if prompt_ids:
+        
+        # Conditionally display the prompt selector only if dataset is available
+        if df is not None:
             prompt_selector = gr.Dropdown(
                 choices=prompt_ids,
                 label="Select Debugging Prompt ID",
-                value=prompt_ids[0],
+                value=prompt_ids[0] if prompt_ids else None,
                 elem_classes="retro-terminal"
             )
         else:
             prompt_selector = None
+        
     with gr.Row():
         response_box = gr.Textbox(
             label="Model Response:",