Update app.py

app.py

@@ -55,16 +55,18 @@ Response:
 st.title(":blue[Venus] Annotation System 🪐")
 
 # Step 1: Load the problem set
+language = ste.selectbox("Select a problem here", ['python3', 'cpp', 'rust', 'javascript', 'golang', 'java'])
+
 my_bar = st.progress(0, text="Loading the problem set...")
 
-my_bar.progress(10, text="Loading [Elfsong/Venus] datasets...")
+my_bar.progress(10, text="Loading [Elfsong/Venus]-[{language}] datasets...")
 if "raw_ds" not in st.session_state.keys():
-    st.session_state["raw_ds"] = load_dataset("Elfsong/Venus", "python3")
+    st.session_state["raw_ds"] = load_dataset("Elfsong/Venus", language)
 raw_ds = st.session_state["raw_ds"]
 
-my_bar.progress(55, text="Loading [Elfsong/venus_case] datasets...")
+my_bar.progress(55, text=f"Loading [Elfsong/venus_case]-[{language}] datasets...")
 if "case_ds" not in st.session_state.keys():
-    st.session_state["case_ds"] = load_dataset("Elfsong/venus_case", "python3")
+    st.session_state["case_ds"] = load_dataset("Elfsong/venus_case", language)
 case_ds = st.session_state["case_ds"]
 
 my_bar.progress(90, text="Filtering out the cases that already exist...")