Update app.py

app.py

@@ -19,6 +19,9 @@ keywords = st.text_input("Enter keywords (comma-separated)")
 # Add slider to the sidebar for the scale value
 scale_percentage = st.sidebar.slider('Scale %', min_value=1, max_value=100, value=50)
 
+# Add slider for the chunk size
+chunk_size = st.sidebar.slider('Chunk size', min_value=100, max_value=1000, value=500)
+
 if uploaded_file is not None and keywords:
     user_input = uploaded_file.read().decode('utf-8')
     keywords = [keyword.strip() for keyword in keywords.split(",")]
@@ -32,8 +35,8 @@ if uploaded_file is not None and keywords:
         summarizer = pipeline('summarization', model=model)
         summarized_text = ""
 
-        # Split the filtered text into chunks of approximately 500 words each
-        chunks = textwrap.wrap(filtered_text, 500)
+        # Split the filtered text into chunks of approximately the selected chunk size each
+        chunks = textwrap.wrap(filtered_text, chunk_size)
         
         # Summarize each chunk
         for chunk in chunks: