Update app.py

app.py

@@ -7,13 +7,13 @@ from keyword_extraction import extract_keywords
 from keyphrase_extraction import extract_sentences_with_obligations
 #from blanc import BlancHelp
 
+
 # Load in ToS
 dataset = load_dataset("EE21/ToS-Summaries")
 
 # Extract titles or identifiers for the ToS
 tos_titles = ["None"] + [f"Document {i}" for i in range(len(dataset['train']))]
 
-
 # Set page to wide mode
 st.set_page_config(layout="wide")
 
@@ -52,7 +52,7 @@ def main():
         uploaded_file = st.file_uploader("Upload a PDF", type="pdf")
         
         # Dropdown for selecting the document
-        tos_selection_index = st.selectbox("Select a Terms of Service Document", range(len(tos_titles)), format_func=lambda x: tos_titles[x])
+        tos_selection_index  = st.selectbox("Select a Terms of Service Document", range(len(tos_titles)), format_func=lambda x: tos_titles[x])
         
         if st.button("Summarize"):
             if uploaded_file and user_input and tos_selection_index:
@@ -64,7 +64,7 @@ def main():
                 st.write("PDF uploaded successfully.")
             elif user_input:
                 file_content = user_input
-            elif tos_selection != "None":
+            elif tos_selection_index != "None":
                 tos_selection_index = tos_titles.index(tos_selection) - 1  # Adjust for the added 'None' option
                 file_content = dataset['train'][tos_selection_index]['plain_text']
             else: