Update app.py

app.py

@@ -11,7 +11,7 @@ from keyphrase_extraction import extract_sentences_with_obligations
 dataset = load_dataset("EE21/ToS-Summaries")
 
 # Extract titles or identifiers for the ToS
-tos_titles = [f"Document {i}" for i in range(len(dataset['train']))]
+tos_titles = ["None"] + [f"Document {i}" for i in range(len(dataset['train']))]
 
 
 # Set page to wide mode
@@ -52,10 +52,10 @@ def main():
         uploaded_file = st.file_uploader("Upload a PDF", type="pdf")
         
         # Dropdown for selecting the document
-        tos_selection_index = st.selectbox("Select 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:
+            if uploaded_file and user_input and tos_selection_index:
                 st.warning("Please provide either text input or a PDF file, not both.")
                 return
             elif uploaded_file:
@@ -64,7 +64,8 @@ def main():
                 st.write("PDF uploaded successfully.")
             elif user_input:
                 file_content = user_input
-            elif tos_selection_index is not None:  # Assuming tos_selection_index won't be None
+            elif tos_selection != "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:
                 st.warning("Please upload a PDF, enter some text, or select a document to summarize.")