show solution

app.py

@@ -25,6 +25,7 @@ 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']
@@ -34,5 +35,10 @@ if filtered_data:
     
     st.subheader("Natural Language to Code")
     st.code(snippet2, language='python')
+    
+    if show_solution:
+        snippet3 = filtered_data[index]['canonical_solution']
+        st.subheader("Solution")
+        st.code(snippet3, language='python')
 else:
     st.error("No data available. Check the URL or search criteria.")