Update app.py

app.py

@@ -1,7 +1,4 @@
 import streamlit as st
-import pandas as pd
-from datetime import datetime
-import os
 import cv2
 from PIL import Image
 import numpy as np
@@ -16,7 +13,7 @@ import fitz  # pymupdf
 import borb.pdf
 import pdf2image
 import vidgear.gears
-import moviepy.editor
+import moviepy.editor as mp  # Corrected import
 import av
 import pydub
 import sounddevice as sd
@@ -28,8 +25,8 @@ st.write("Watch these libraries flex their muscles with live demos producing rea
 
 # 📜 Demo Functions
 def demo_pypdf():
-    with open("demo.pdf", "w") as f:
-        f.write("Hello from pypdf!")
+    with open("demo.pdf", "wb") as f:  # Binary mode for PDF
+        f.write(b"Hello from pypdf!")  # Binary string
     reader = pypdf.PdfReader("demo.pdf")
     return reader.pages[0].extract_text() if reader.pages else "No text found!"
 
@@ -40,8 +37,8 @@ def demo_pikepdf():
     return "demo_pikepdf.pdf"
 
 def demo_pdfminer():
-    with open("demo.pdf", "w") as f:
-        f.write("PDFminer magic!")
+    with open("demo.pdf", "wb") as f:
+        f.write(b"PDFminer magic!")
     text = pdfminer.high_level.extract_text("demo.pdf")
     return text
 
@@ -58,8 +55,8 @@ def demo_pdfkit():
     return "demo_pdfkit.pdf"
 
 def demo_pdfplumber():
-    with open("demo.pdf", "w") as f:
-        f.write("Plumb this text!")
+    with open("demo.pdf", "wb") as f:
+        f.write(b"Plumb this text!")
     with pdfplumber.open("demo.pdf") as pdf:
         return pdf.pages[0].extract_text() if pdf.pages else "No text!"
 
@@ -78,8 +75,8 @@ def demo_borb():
     return "demo_borb.pdf"
 
 def demo_pdf2image():
-    with open("demo.pdf", "w") as f:
-        f.write("Image this!")
+    with open("demo.pdf", "wb") as f:
+        f.write(b"Image this!")
     images = pdf2image.convert_from_path("demo.pdf")
     images[0].save("demo_pdf2image.jpg")
     return "demo_pdf2image.jpg"
@@ -131,33 +128,36 @@ if demo_type == "PDF":
     ])
     if st.button("Run PDF Demo"):
         with st.spinner("Running demo..."):
-            if demo == "pypdf":
-                result = demo_pypdf()
-                st.write("Extracted Text:", result)
-            elif demo == "pikepdf":
-                result = demo_pikepdf()
-                st.download_button("Download PDF", open(result, "rb").read(), file_name=result)
-            elif demo == "pdfminer.six":
-                result = demo_pdfminer()
-                st.write("Extracted Text:", result)
-            elif demo == "fpdf2":
-                result = demo_fpdf()
-                st.download_button("Download PDF", open(result, "rb").read(), file_name=result)
-            elif demo == "pdfkit":
-                result = demo_pdfkit()
-                st.download_button("Download PDF", open(result, "rb").read(), file_name=result)
-            elif demo == "pdfplumber":
-                result = demo_pdfplumber()
-                st.write("Extracted Text:", result)
-            elif demo == "pymupdf":
-                result = demo_pymupdf()
-                st.download_button("Download PDF", open(result, "rb").read(), file_name=result)
-            elif demo == "borb":
-                result = demo_borb()
-                st.download_button("Download PDF", open(result, "rb").read(), file_name=result)
-            elif demo == "pdf2image":
-                result = demo_pdf2image()
-                st.image(result, caption="Converted PDF to Image")
+            try:
+                if demo == "pypdf":
+                    result = demo_pypdf()
+                    st.write("Extracted Text:", result)
+                elif demo == "pikepdf":
+                    result = demo_pikepdf()
+                    st.download_button("Download PDF", open(result, "rb").read(), file_name=result)
+                elif demo == "pdfminer.six":
+                    result = demo_pdfminer()
+                    st.write("Extracted Text:", result)
+                elif demo == "fpdf2":
+                    result = demo_fpdf()
+                    st.download_button("Download PDF", open(result, "rb").read(), file_name=result)
+                elif demo == "pdfkit":
+                    result = demo_pdfkit()
+                    st.download_button("Download PDF", open(result, "rb").read(), file_name=result)
+                elif demo == "pdfplumber":
+                    result = demo_pdfplumber()
+                    st.write("Extracted Text:", result)
+                elif demo == "pymupdf":
+                    result = demo_pymupdf()
+                    st.download_button("Download PDF", open(result, "rb").read(), file_name=result)
+                elif demo == "borb":
+                    result = demo_borb()
+                    st.download_button("Download PDF", open(result, "rb").read(), file_name=result)
+                elif demo == "pdf2image":
+                    result = demo_pdf2image()
+                    st.image(result, caption="Converted PDF to Image")
+            except Exception as e:
+                st.error(f"Demo failed: {str(e)}")
 
 # 📸 Image Capture Demos
 elif demo_type == "Image Capture":
@@ -165,18 +165,21 @@ elif demo_type == "Image Capture":
     demo = st.selectbox("Pick an Image Demo", ["opencv-python", "imutils"])
     if st.button("Run Image Demo"):
         with st.spinner("Capturing image..."):
-            if demo == "opencv-python":
-                result = demo_opencv_image()
-                if result:
-                    st.image(result, caption="Snapped with OpenCV!")
-                else:
-                    st.error("Camera not detected!")
-            elif demo == "imutils":
-                result = demo_imutils()
-                if result:
-                    st.image(result, caption="Snapped with Imutils!")
-                else:
-                    st.error("Camera not detected!")
+            try:
+                if demo == "opencv-python":
+                    result = demo_opencv_image()
+                    if result:
+                        st.image(result, caption="Snapped with OpenCV!")
+                    else:
+                        st.error("Camera not detected!")
+                elif demo == "imutils":
+                    result = demo_imutils()
+                    if result:
+                        st.image(result, caption="Snapped with Imutils!")
+                    else:
+                        st.error("Camera not detected!")
+            except Exception as e:
+                st.error(f"Demo failed: {str(e)}")
 
 # 🎥 Video Capture Demos
 elif demo_type == "Video Capture":
@@ -184,18 +187,21 @@ elif demo_type == "Video Capture":
     demo = st.selectbox("Pick a Video Demo", ["opencv-python", "vidgear"])
     if st.button("Run Video Demo"):
         with st.spinner("Capturing video frame..."):
-            if demo == "opencv-python":
-                result = demo_opencv_image()  # Using image capture as a proxy
-                if result:
-                    st.image(result, caption="Frame from OpenCV Video!")
-                else:
-                    st.error("Camera not detected!")
-            elif demo == "vidgear":
-                result = demo_vidgear()
-                if result:
-                    st.image(result, caption="Frame from VidGear Video!")
-                else:
-                    st.error("Camera not detected!")
+            try:
+                if demo == "opencv-python":
+                    result = demo_opencv_image()  # Using image capture as a proxy
+                    if result:
+                        st.image(result, caption="Frame from OpenCV Video!")
+                    else:
+                        st.error("Camera not detected!")
+                elif demo == "vidgear":
+                    result = demo_vidgear()
+                    if result:
+                        st.image(result, caption="Frame from VidGear Video!")
+                    else:
+                        st.error("Camera not detected!")
+            except Exception as e:
+                st.error(f"Demo failed: {str(e)}")
 
 # 🎵 Audio Capture Demos
 elif demo_type == "Audio Capture":
@@ -203,10 +209,13 @@ elif demo_type == "Audio Capture":
     demo = st.selectbox("Pick an Audio Demo", ["sounddevice"])
     if st.button("Run Audio Demo"):
         with st.spinner("Recording audio (2 seconds)..."):
-            if demo == "sounddevice":
-                result = demo_sounddevice()
-                st.audio(result, format="audio/wav")
-                st.download_button("Download Audio", open(result, "rb").read(), file_name=result)
+            try:
+                if demo == "sounddevice":
+                    result = demo_sounddevice()
+                    st.audio(result, format="audio/wav")
+                    st.download_button("Download Audio", open(result, "rb").read(), file_name=result)
+            except Exception as e:
+                st.error(f"Demo failed: {str(e)}")
 
 # 📊 Results Display
 st.write("### Results")