fix: update data loader

app.py

@@ -1,39 +1,35 @@
 import streamlit as st
 import requests
-import json
+from datasets import load_dataset
+NAME='bigcode/bigcodebench'
 
-FILE = "wild-code-bench-v0.1.0.jsonl"
-# Function to load data from a JSONL
+# Function to load data
 @st.cache_data
-def load_data_from_jsonl(file):
-    with open(file, 'r') as f:
-        jsonl_data = f.readlines()
-    return [json.loads(line) for line in jsonl_data]
+def load_data(file):
+    return load_dataset('bigcode/bigcodebench', split='train')
 
 # Function to filter data based on search keyword
 def filter_data(data, keyword):
     if not keyword:
         return data
-    filtered_data = [item for item in data if keyword.lower() in item['prompt'].lower() or keyword.lower() in item['instruction'].lower()]
+    filtered_data = [item for item in data if keyword.lower() in item['complete_prompt'].lower()]
     return filtered_data
 
 # Streamlit UI Setup
 st.title('BigCodeBench Viewer')
-data = load_data_from_jsonl(FILE)
+data = load_data(NAME)
 
 search_keyword = st.sidebar.text_input('Search by keyword', '')
 filtered_data = filter_data(data, search_keyword)
-
-index = st.sidebar.number_input('Select Index', min_value=0, max_value=len(filtered_data)-1, value=0, step=1)
-show_solution = st.sidebar.checkbox("Show Solution")
-
 if filtered_data:
-    snippet1 = filtered_data[index]['prompt']
-    snippet2 = filtered_data[index]['instruction']
-    st.subheader("Code to Code")
+    index = st.sidebar.number_input('Select Index', min_value=0, max_value=len(filtered_data)-1, value=0, step=1)
+    show_solution = st.sidebar.checkbox('Show Solution')
+    snippet1 = filtered_data[index]['complete_prompt']
+    snippet2 = filtered_data[index]['instruct_prompt']
+    st.subheader("Code Completion")
     st.code(snippet1, language='python')
     
-    st.subheader("Natural Language to Code")
+    st.subheader("Natural Language Instruction to Code")
     st.code(snippet2, language='python')
     
     if show_solution: