Update app.py

app.py

@@ -12,17 +12,11 @@ model = st.sidebar.selectbox("Choose a model", models)
 
 uploaded_file = st.file_uploader("Choose a .txt file", type="txt")
 
+# Add slider to the sidebar for the scale value
+scale_percentage = st.sidebar.slider('Scale %', min_value=1, max_value=100, value=50)
+
 if uploaded_file is not None:
     user_input = uploaded_file.read().decode('utf-8')
-    total_length = len(user_input.split())
-
-    # Add slider to the sidebar for the scale value
-    scale_percentage = st.sidebar.slider('Scale %', min_value=1, max_value=100, value=50)
-    min_length_percentage = max(scale_percentage - 10, 1)  # Ensure min_length_percentage is not less than 1
-    max_length_percentage = min(scale_percentage + 10, 100)  # Ensure max_length_percentage is not more than 100
-
-    st.sidebar.text(f'Minimum Length %: {min_length_percentage}')
-    st.sidebar.text(f'Maximum Length %: {max_length_percentage}')
 
     if st.button('Summarize'):
         summarizer = pipeline('summarization', model=model)
@@ -33,8 +27,11 @@ if uploaded_file is not None:
         
         # Summarize each chunk
         for chunk in chunks:
-            min_length = max(int(total_length * min_length_percentage / 100), 1)  # Calculate min_length based on the percentage of the total length
-            max_length = int(total_length * max_length_percentage / 100)  # Calculate max_length based on the percentage of the total length
+            chunk_length = len(chunk.split())
+            min_length_percentage = max(scale_percentage - 10, 1)  # Ensure min_length_percentage is not less than 1
+            max_length_percentage = min(scale_percentage + 10, 100)  # Ensure max_length_percentage is not more than 100
+            min_length = max(int(chunk_length * min_length_percentage / 100), 1)  # Calculate min_length based on the percentage of the chunk length
+            max_length = int(chunk_length * max_length_percentage / 100)  # Calculate max_length based on the percentage of the chunk length
             summarized = summarizer(chunk, max_length=max_length, min_length=min_length, do_sample=False)
             summarized_text += summarized[0]['summary_text'] + " "