added default index

app.py

@@ -106,38 +106,40 @@ with st.expander("Upload csv file"):
 if uploaded_file is not None:
     df = load_data(uploaded_file)
 
-    # Run inference on first example
-    user_idx = st.text_input('Enter index')
-    first_example = df['review'][int(user_idx)]
-    question = "How was the recipe improved?"
-
-    if postprocessing == "no postprocessing":
-        resp = question_answer(question=question,
-                             context=first_example,
-                             top_k=5)
-    elif postprocessing == "remove substrings":
-        resp = question_answer(question=question,
-                             context=first_example,
-                             top_k=5)
-        resp = remove_substring(resp)
-    elif postprocessing == "iteration":
-        resp = remove_substring_iter(question_answer,
-                                     question,
-                                     first_example)
-
-    # Present results
-    st.markdown(f"""
-    # Results
-
-    The review provided was:
-
-    {first_example}
-
-    The question asked was:
-
-    {question}
-
-    The answers were:
-
-    {resp}
-    """)
+    # Ask user for index to try out
+    user_idx = st.text_input(f'Enter index (max {df.shape[0] - 1}', "0")
+
+    # Only run rest of script if user provides index
+    if user_idx is not None:
+        first_example = df['review'][int(user_idx)]
+        question = "How was the recipe improved?"
+        if postprocessing == "no postprocessing":
+            resp = question_answer(question=question,
+                                context=first_example,
+                                top_k=5)
+        elif postprocessing == "remove substrings":
+            resp = question_answer(question=question,
+                                context=first_example,
+                                top_k=5)
+            resp = remove_substring(resp)
+        elif postprocessing == "iteration":
+            resp = remove_substring_iter(question_answer,
+                                        question,
+                                        first_example)
+
+        # Present results
+        st.markdown(f"""
+        # Results
+
+        The review provided was:
+
+        {first_example}
+
+        The question asked was:
+
+        {question}
+
+        The answers were:
+
+        {resp}
+        """)