Update app.py

app.py

@@ -17,15 +17,23 @@ def extract_text_from_pdf(pdf_file):
 
 # Function to generate discussion points
 def generate_discussion_points(text):
-    summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
-    summary = summarizer(text, max_length=130, min_length=30, do_sample=False)
-    return summary[0]['summary_text']
+    try:
+        summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+        summary = summarizer(text, max_length=130, min_length=30, do_sample=False)
+        return summary[0]['summary_text']
+    except Exception as e:
+        st.error(f"Error during summarization: {e}")
+        return None
 
 # Function to convert text to speech
 def text_to_speech(text, output_file="output.mp3"):
-    tts = gTTS(text)
-    tts.save(output_file)
-    return output_file
+    try:
+        tts = gTTS(text)
+        tts.save(output_file)
+        return output_file
+    except Exception as e:
+        st.error(f"Error during text-to-speech conversion: {e}")
+        return None
 
 # Streamlit app starts here
 st.title("📄 PDF Discussion Points Generator")
@@ -45,37 +53,34 @@ if uploaded_file:
 
         # Generate discussion points
         with st.spinner("Generating discussion points..."):
-            try:
-                discussion_points = generate_discussion_points(pdf_text)
+            discussion_points = generate_discussion_points(pdf_text)
+
+            if discussion_points:
                 st.write("### Discussion Points")
                 st.text_area("Discussion Points", discussion_points, height=150)
-            except Exception as e:
-                st.error(f"Error during summarization: {e}")
 
-            # Convert discussion points to audio
-            with st.spinner("Converting discussion points to audio..."):
-                try:
+                # Convert discussion points to audio
+                with st.spinner("Converting discussion points to audio..."):
                     audio_file = text_to_speech(discussion_points)
-                    st.success("Audio file generated successfully!")
-                except Exception as e:
-                    st.error(f"Error during text-to-speech conversion: {e}")
-                    audio_file = None
 
-            if audio_file:
-                # Audio playback
-                st.write("### Listen to the Discussion Points")
-                audio_bytes = open(audio_file, "rb").read()
-                st.audio(audio_bytes, format="audio/mp3")
+                    if audio_file:
+                        st.success("Audio file generated successfully!")
+
+                        # Audio playback
+                        st.write("### Listen to the Discussion Points")
+                        audio_bytes = open(audio_file, "rb").read()
+                        st.audio(audio_bytes, format="audio/mp3")
 
-                # Option to download the audio file
-                st.download_button(
-                    label="Download Audio",
-                    data=audio_bytes,
-                    file_name="discussion_points.mp3",
-                    mime="audio/mp3"
-                )
+                        # Option to download the audio file
+                        st.download_button(
+                            label="Download Audio",
+                            data=audio_bytes,
+                            file_name="discussion_points.mp3",
+                            mime="audio/mp3"
+                        )
+            else:
+                st.error("Failed to generate discussion points. Please try again.")
     else:
         st.error("No text was extracted from the uploaded PDF. Please try with another file.")
 else:
     st.info("Upload a PDF file to start.")
-