Update app.py

app.py

@@ -9,26 +9,34 @@ from io import BytesIO
 import base64
 from imutils.video import VideoStream
 import cv2
+from pydub import AudioSegment
+from pydub.playback import play
+import pypdf
+import numpy as np
 
-# 🌟🔥 Initialize session state like a galactic DJ spinning tracks!
+# 🌟🔥 Initialize session state
 if 'file_history' not in st.session_state:
     st.session_state['file_history'] = []
 if 'auto_capture_running' not in st.session_state:
     st.session_state['auto_capture_running'] = False
 
-# 📜💾 Save to history like a time-traveling scribe!
-def save_to_history(file_type, file_path, img_data):
+# 📜💾 Save to history
+def save_to_history(file_type, file_path, data=None):
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    # Convert RGB to BGR for OpenCV saving
-    img_bgr = cv2.cvtColor(img_data, cv2.COLOR_RGB2BGR)
-    cv2.imwrite(file_path, img_bgr)
+    if file_type == "🖼️ Image" and data is not None:
+        cv2.imwrite(file_path, cv2.cvtColor(data, cv2.COLOR_RGB2BGR))
+    elif file_type == "🎵 Audio" and data is not None:
+        data.export(file_path, format="wav")
+    elif file_type == "📜 PDF" and data is not None:
+        with open(file_path, "wb") as f:
+            f.write(data)
     st.session_state['file_history'].append({
         "Timestamp": timestamp,
         "Type": file_type,
         "Path": file_path
     })
 
-# 📸⏰ Auto-capture every 10 secs like a sneaky shutterbug!
+# 📸⏰ Auto-capture every 10 secs
 def auto_capture(streams):
     while st.session_state['auto_capture_running']:
         for i, stream in enumerate(streams):
@@ -38,7 +46,7 @@ def auto_capture(streams):
                 save_to_history("🖼️ Image", filename, frame)
         time.sleep(10)
 
-# 🎛️ Sidebar config like a spaceship control panel!
+# 🎛️ Sidebar config
 with st.sidebar:
     st.header("🎚️📸 Snap Shack")
     streams = [
@@ -51,21 +59,36 @@ with st.sidebar:
     if st.button("⏹️ Stop Auto-Snap"):
         st.session_state['auto_capture_running'] = False
 
-    # 📂 Sidebar file outline with emoji flair!
+    # Audio recording control
+    if st.button("🎙️ Record Audio (5s)"):
+        audio = AudioSegment.silent(duration=5000)  # Placeholder; real mic capture needs pyaudio
+        filename = f"audio_{datetime.now().strftime('%Y%m%d_%H%M%S')}.wav"
+        save_to_history("🎵 Audio", filename, audio)
+
     st.subheader("📜 Snap Stash")
     if st.session_state['file_history']:
         images = [f for f in st.session_state['file_history'] if f['Type'] == "🖼️ Image"]
+        audios = [f for f in st.session_state['file_history'] if f['Type'] == "🎵 Audio"]
+        pdfs = [f for f in st.session_state['file_history'] if f['Type'] == "📜 PDF"]
         if images:
             st.write("🖼️ Images")
             for img in images:
                 st.write(f"- {img['Path']} @ {img['Timestamp']}")
+        if audios:
+            st.write("🎵 Audios")
+            for aud in audios:
+                st.write(f"- {aud['Path']} @ {aud['Timestamp']}")
+        if pdfs:
+            st.write("📜 PDFs")
+            for pdf in pdfs:
+                st.write(f"- {pdf['Path']} @ {pdf['Timestamp']}")
     else:
         st.write("🕳️ Empty Stash!")
 
-# 🌍🎨 Main UI kicks off like a cosmic art show!
-st.title("📸 Imutils Snap Craze")
+# 🌍🎨 Main UI
+st.title("📸 Multi-Lib Snap Craze")
 
-# 📸📷 Camera snap zone!
+# 📸📷 Camera snap zone
 st.header("📸🎥 Snap Zone")
 cols = st.columns(2)
 placeholders = [cols[0].empty(), cols[1].empty()]
@@ -74,31 +97,37 @@ for i, (placeholder, stream) in enumerate(zip(placeholders, streams)):
     if frame is not None:
         placeholder.image(frame, caption=f"Live Cam {i}", use_container_width=True)
     else:
-        placeholder.error(f"🚨 No feed from Cam {i}. Check connection or index.")
+        placeholder.error(f"🚨 No feed from Cam {i}")
     
     if st.button(f"📸 Snap Cam {i}", key=f"snap{i}"):
         frame = stream.read()
         if frame is not None:
             filename = f"cam{i}_snap_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
             save_to_history("🖼️ Image", filename, frame)
-        else:
-            st.error(f"🚨 Failed to snap Cam {i}")
 
-# 📂 Upload zone like a media drop party!
+# 📂 Upload zone (PDFs and Images)
 st.header("📥🎉 Drop Zone")
-uploaded_files = st.file_uploader("📸 Toss Pics", accept_multiple_files=True, type=['jpg', 'png'])
+uploaded_files = st.file_uploader("📜📸 Upload PDFs or Images", accept_multiple_files=True, type=['pdf', 'jpg', 'png'])
 if uploaded_files:
     for uploaded_file in uploaded_files:
-        file_path = f"uploaded_{uploaded_file.name}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
-        img = Image.open(uploaded_file)
-        img_array = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)  # Convert for consistency
-        save_to_history("🖼️ Image", file_path, img_array)
-        st.image(img, caption=uploaded_file.name, use_container_width=True)
+        file_type = "📜 PDF" if uploaded_file.type == "application/pdf" else "🖼️ Image"
+        file_path = f"uploaded_{uploaded_file.name}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.{uploaded_file.name.split('.')[-1]}"
+        if file_type == "🖼️ Image":
+            img = Image.open(uploaded_file)
+            img_array = cv2.cvtColor(np.array(img), cv2.COLOR_RGB2BGR)
+            save_to_history(file_type, file_path, img_array)
+            st.image(img, caption=uploaded_file.name, use_container_width=True)
+        else:
+            save_to_history(file_type, file_path, uploaded_file.getvalue())
+            with pypdf.PdfReader(uploaded_file) as reader:
+                st.write(f"PDF Pages: {len(reader.pages)}")
 
-# 🖼️ Gallery like a media circus!
+# 🖼️ Gallery
 st.header("🎪 Snap Show")
 if st.session_state['file_history']:
     images = [f for f in st.session_state['file_history'] if f['Type'] == "🖼️ Image"]
+    audios = [f for f in st.session_state['file_history'] if f['Type'] == "🎵 Audio"]
+    pdfs = [f for f in st.session_state['file_history'] if f['Type'] == "📜 PDF"]
     if images:
         st.subheader("🖼️ Pic Parade")
         cols = st.columns(3)
@@ -109,12 +138,25 @@ if st.session_state['file_history']:
                     with open(img['Path'], "rb") as f:
                         img_data = f.read()
                     st.markdown(f'<a href="data:image/jpeg;base64,{base64.b64encode(img_data).decode()}" download="{os.path.basename(img["Path"])}">📥 Snag It!</a>', unsafe_allow_html=True)
-                else:
-                    st.warning(f"🚨 Missing file: {img['Path']}")
+    if audios:
+        st.subheader("🎵 Audio Alley")
+        for aud in audios:
+            if os.path.exists(aud['Path']):
+                st.audio(aud['Path'])
+                with open(aud['Path'], "rb") as f:
+                    aud_data = f.read()
+                st.markdown(f'<a href="data:audio/wav;base64,{base64.b64encode(aud_data).decode()}" download="{os.path.basename(aud["Path"])}">📥 Snag It!</a>', unsafe_allow_html=True)
+    if pdfs:
+        st.subheader("📜 PDF Plaza")
+        for pdf in pdfs:
+            if os.path.exists(pdf['Path']):
+                with open(pdf['Path'], "rb") as f:
+                    pdf_data = f.read()
+                st.markdown(f'<a href="data:application/pdf;base64,{base64.b64encode(pdf_data).decode()}" download="{os.path.basename(pdf["Path"])}">📥 Download {os.path.basename(pdf["Path"])}</a>', unsafe_allow_html=True)
 else:
     st.write("🚫 No snaps yet!")
 
-# 📜 History log like a time machine!
+# 📜 History log
 st.header("⏳ Snap Saga")
 if st.session_state['file_history']:
     df = pd.DataFrame(st.session_state['file_history'])
@@ -122,7 +164,7 @@ if st.session_state['file_history']:
 else:
     st.write("🕳️ Nothing snapped yet!")
 
-# Cleanup on app exit
+# Cleanup
 def cleanup():
     for stream in streams:
         stream.stop()