Update app.py

app.py

@@ -15,12 +15,16 @@ def extract_text_from_pdf(pdf_file):
         st.error(f"Error reading the PDF: {e}")
     return text
 
-# Function to generate discussion points
+# Function to generate discussion points with adjustable summary length
 def generate_discussion_points(text, user_prompt=None):
     summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+    
+    # Optionally include the user's prompt to guide the summary
     if user_prompt:
-        text = user_prompt + " " + text  # Include the user's prompt in the summary input
-    summary = summarizer(text, max_length=130, min_length=30, do_sample=False)
+        text = user_prompt + " " + text
+
+    # Adjust the max_length and min_length for a longer summary
+    summary = summarizer(text, max_length=300, min_length=100, do_sample=False)
     return summary[0]['summary_text']
 
 # Function to convert text to speech
@@ -31,13 +35,13 @@ def text_to_speech(text, output_file="output.mp3"):
 
 # Streamlit app starts here
 st.title("📄 PDF Discussion Points Generator with User Prompts")
-st.write("Upload a small PDF file to generate discussion points and listen to them.")
+st.write("Upload a PDF file to generate discussion points and listen to them. Optionally, provide a prompt to guide the summary.")
 
 # File uploader
 uploaded_file = st.file_uploader("Upload a PDF", type=["pdf"])
 
-# User prompt input
-user_prompt = st.text_input("Enter a prompt for the summary (optional)")
+# Input for user-defined summary prompt
+user_prompt = st.text_area("Enter a prompt to guide the summary (optional)", height=100)
 
 if uploaded_file:
     # Extract text from uploaded PDF