Update app.py

app.py

@@ -1,28 +1,24 @@
 import streamlit as st
 import pandas as pd
 from datetime import datetime
-import cv2
-from pydub import AudioSegment
-import imageio
-import av
-import moviepy as mp
 import os
-import numpy as np
-from io import BytesIO
 import threading
 import time
+from PIL import Image
+from io import BytesIO
+import aiofiles
+import asyncio
+import base64
 
 # 🌟🔥 Initialize session state like a galactic DJ spinning tracks!
 if 'file_history' not in st.session_state:
     st.session_state['file_history'] = []
-if 'ping_code' not in st.session_state:
-    st.session_state['ping_code'] = ""
-if 'uploaded_files' not in st.session_state:
-    st.session_state['uploaded_files'] = []
-if 'auto_capture_running' not in st.session_state:
+if 'auto_capture_running' not_in st.session_state:
     st.session_state['auto_capture_running'] = False
+if 'cam_file' not_in st.session_state:
+    st.session_state['cam_file'] = None
 
-# 📜💾 Save to history like a time-traveling scribe! | 📅✨ save_to_history("Image", "pic.jpg") - Stamps a pic in the history books like a boss!
+# 📜💾 Save to history like a time-traveling scribe! | 📅✨ save_to_history("🖼️ Image", "pic.jpg") - Stamps a pic in the history books like a boss!
 def save_to_history(file_type, file_path):
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     st.session_state['file_history'].append({
@@ -33,464 +29,75 @@ def save_to_history(file_type, file_path):
 
 # 📸⏰ Auto-capture every 10 secs like a sneaky shutterbug!
 def auto_capture():
-    if st.session_state['auto_capture_running']:
-        cap = None
-        for i in range(10):  # Try indices 0-9
-            cap = cv2.VideoCapture(i)
-            if cap.isOpened():
-                break
-            cap.release()
-        if cap and cap.isOpened():
-            ret, frame = cap.read()
-            if ret:
-                file_path = f"auto_snap_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
-                cv2.imwrite(file_path, frame)
-                save_to_history("🖼️ Image", file_path)
-            cap.release()
-        else:
-            st.warning("🚨 No camera found!")
+    if st.session_state['auto_capture_running'] and 'cam_img' in st.session_state:
+        cam_img = st.session_state['cam_img']
+        if cam_img:
+            filename = f"auto_snap_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
+            with open(filename, "wb") as f:
+                f.write(cam_img.getvalue())
+            save_to_history("🖼️ Image", filename)
         threading.Timer(10, auto_capture).start()
 
 # 🎛️ Sidebar config like a spaceship control panel!
 with st.sidebar:
-    st.header("🎚️📸 Tune-Up Zone")
-    library_choice = st.selectbox("📚 Pick a Tool", ["OpenCV", "PyDub", "ImageIO", "PyAV", "MoviePy", "JS Audio"])
-    resolution = st.select_slider("📏 Snap Size", options=["320x240", "640x480", "1280x720"], value="640x480")
-    fps = st.slider("⏱️ Speed Snap", 1, 60, 30)
+    st.header("🎚️📸 Snap Shack")
     if st.button("⏰ Start Auto-Snap"):
         st.session_state['auto_capture_running'] = True
         auto_capture()
     if st.button("⏹️ Stop Auto-Snap"):
         st.session_state['auto_capture_running'] = False
 
-    # 🔢 DTMF ping code like a retro phone hacker!
-    st.subheader("🔐 Ping-a-Tron")
-    col1, col2, col3, col4 = st.columns(4)
-    with col1:
-        digit1 = st.selectbox("1️⃣", [str(i) for i in range(10)], key="d1")
-    with col2:
-        digit2 = st.selectbox("2️⃣", [str(i) for i in range(10)], key="d2")
-    with col3:
-        digit3 = st.selectbox("3️⃣", [str(i) for i in range(10)], key="d3")
-    with col4:
-        digit4 = st.selectbox("4️⃣", [str(i) for i in range(10)], key="d4")
-    ping_code = digit1 + digit2 + digit3 + digit4
-    st.session_state['ping_code'] = ping_code
-    st.write(f"🔑 Code: {ping_code}")
-
     # 📂 Sidebar file outline with emoji flair!
-    st.subheader("📜 File Vault")
+    st.subheader("📜 Snap Stash")
     if st.session_state['file_history']:
         images = [f for f in st.session_state['file_history'] if f['Type'] == "🖼️ Image"]
-        videos = [f for f in st.session_state['file_history'] if f['Type'] in ["🎥 Video", "🎞️ GIF"]]
-        audios = [f for f in st.session_state['file_history'] if f['Type'] == "🎵 Audio"]
         if images:
             st.write("🖼️ Images")
             for img in images:
                 st.write(f"- {img['Path']} @ {img['Timestamp']}")
-        if videos:
-            st.write("🎥 Videos/GIFs")
-            for vid in videos:
-                st.write(f"- {vid['Path']} @ {vid['Timestamp']}")
-        if audios:
-            st.write("🎵 Audios")
-            for aud in audios:
-                st.write(f"- {aud['Path']} @ {aud['Timestamp']}")
     else:
-        st.write("🕳️ Empty Vault!")
+        st.write("🕳️ Empty Stash!")
 
 # 🌍🎨 Main UI kicks off like a cosmic art show!
-st.title("📸🎙️ Capture Craze")
-
-# 📸📚 Library showcase like a tech talent show!
-st.header("📸🎙️ Tool Titans")
-
-# 1. OpenCV - 📷 Pixel party time!
-with st.expander("1️⃣ 📷 OpenCV Fiesta"):
-    st.write("🎥 Snap Star: Real-time pixel magic!")
-    st.subheader("🔥 Top Tricks")
-    
-    st.write("📸🎥 Video Snap")
-    if st.button("🎬 Go Live", key="opencv_1"):
-        # 📷🎥 Snags webcam like a paparazzi pro! | 📸 cap = cv2.VideoCapture(0) - Grabs live feed faster than gossip spreads!
-        cap = None
-        for i in range(10):
-            cap = cv2.VideoCapture(i)
-            if cap.isOpened():
-                break
-            cap.release()
-        if cap and cap.isOpened():
-            frame_placeholder = st.empty()
-            for _ in range(50):
-                ret, frame = cap.read()
-                if ret:
-                    frame_placeholder.image(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
-            cap.release()
-        else:
-            st.error("🚨 No camera found!")
-
-    st.write("🖌️✨ Gray Snap")
-    if st.button("📷 Save Gray", key="opencv_2"):
-        # 🖌️✨ Saves grayscale like an artsy ghost! | 📝 cv2.imwrite("gray.jpg", cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)) - Ditches color like a moody poet!
-        cap = None
-        for i in range(10):
-            cap = cv2.VideoCapture(i)
-            if cap.isOpened():
-                break
-            cap.release()
-        if cap and cap.isOpened():
-            ret, frame = cap.read()
-            if ret:
-                gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
-                file_path = f"opencv_gray_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
-                cv2.imwrite(file_path, gray)
-                save_to_history("🖼️ Image", file_path)
-                st.image(file_path, caption="🖤 Gray Vibes")
-            cap.release()
-        else:
-            st.error("🚨 No camera found!")
-
-    st.write("🕵️‍♂️🔍 Edge Snap")
-    if st.button("🔪 Edge It", key="opencv_3"):
-        # 🕵️‍♂️🔍 Finds edges sharper than a detective’s wit! | 🖼️ edges = cv2.Canny(frame, 100, 200) - Outlines like a crime scene sketch!
-        cap = None
-        for i in range(10):
-            cap = cv2.VideoCapture(i)
-            if cap.isOpened():
-                break
-            cap.release()
-        if cap and cap.isOpened():
-            ret, frame = cap.read()
-            if ret:
-                edges = cv2.Canny(frame, 100, 200)
-                file_path = f"opencv_edges_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
-                cv2.imwrite(file_path, edges)
-                save_to_history("🖼️ Image", file_path)
-                st.image(file_path, caption="🔍 Edge Lord")
-            cap.release()
-        else:
-            st.error("🚨 No camera found!")
-
-# 2. PyDub - 🎙️ Audio mixing madness!
-with st.expander("2️⃣ 🎙️ PyDub Party"):
-    st.write("🔊 Audio Ace: Mixmaster vibes!")
-    st.subheader("🔥 Top Tricks")
-
-    st.write("🎤🌩️ Load Jam")
-    if st.button("🎵 Load Sound", key="pydub_1"):
-        # 🎤🌩️ Grabs audio like a sonic thief! | 🎙️ AudioSegment.from_file("sound.wav") - Nabs tracks like a sound bandit!
-        uploaded_audio = st.file_uploader("🎙️ Drop a WAV", type=['wav'], key="pydub_load")
-        if uploaded_audio:
-            sound = AudioSegment.from_file(uploaded_audio)
-            file_path = f"pydub_load_{datetime.now().strftime('%Y%m%d_%H%M%S')}.wav"
-            sound.export(file_path, format="wav")
-            save_to_history("🎵 Audio", file_path)
-            st.audio(file_path)
-
-    st.write("🔊📡 Export Snap")
-    if st.button("🎶 MP3 It", key="pydub_2"):
-        # 🔊📡 Spits out tracks like a beat factory! | 🎵 sound.export("out.mp3") - Pumps tunes like a hit machine!
-        uploaded_audio = st.file_uploader("🎙️ Drop a WAV", type=['wav'], key="pydub_export")
-        if uploaded_audio:
-            sound = AudioSegment.from_file(uploaded_audio)
-            file_path = f"pydub_mp3_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp3"
-            sound.export(file_path, format="mp3")
-            save_to_history("🎵 Audio", file_path)
-            st.audio(file_path)
-
-    st.write("🎶💾 Reverse Blast")
-    if st.button("🔄 Flip It", key="pydub_3"):
-        # 🎶💾 Flips sound like a time-travel DJ! | 🎧 sound.reverse() - Spins audio back like a retro remix!
-        uploaded_audio = st.file_uploader("🎙️ Drop a WAV", type=['wav'], key="pydub_reverse")
-        if uploaded_audio:
-            sound = AudioSegment.from_file(uploaded_audio)
-            reversed_sound = sound.reverse()
-            file_path = f"pydub_rev_{datetime.now().strftime('%Y%m%d_%H%M%S')}.wav"
-            reversed_sound.export(file_path, format="wav")
-            save_to_history("🎵 Audio", file_path)
-            st.audio(file_path)
-
-# 3. ImageIO - 🖼️ Pixel playtime!
-with st.expander("3️⃣ 📹 ImageIO Bash"):
-    st.write("🎥 Easy Snap: Pixel lightweight champ!")
-    st.subheader("🔥 Top Tricks")
-
-    st.write("📹👀 Frame Peek")
-    if st.button("📸 Snap It", key="imageio_1"):
-        # 📹👀 Grabs frames like a shy stalker! | 📷 reader = imageio.get_reader('<video0>') - Sneaks a peek at your cam!
-        try:
-            reader = imageio.get_reader('<video0>')
-            frame = reader.get_next_data()
-            file_path = f"imageio_frame_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
-            imageio.imwrite(file_path, frame)
-            save_to_history("🖼️ Image", file_path)
-            st.image(file_path)
-        except Exception as e:
-            st.error(f"🚨 Camera snag: {str(e)}")
-
-    st.write("🖨️😂 Crunch Snap")
-    if st.button("📏 Slim Pic", key="imageio_2"):
-        # 🖨️😂 Compresses like a diet guru! | 🖼️ imageio.imwrite("pic.jpg", frame, quality=85) - Shrinks pics like a tight squeeze!
-        try:
-            reader = imageio.get_reader('<video0>')
-            frame = reader.get_next_data()
-            file_path = f"imageio_comp_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
-            imageio.imwrite(file_path, frame, quality=85)
-            save_to_history("🖼️ Image", file_path)
-            st.image(file_path, caption="😂 Slim Fit")
-        except Exception as e:
-            st.error(f"🚨 Camera snag: {str(e)}")
-
-    st.write("🎞️🤡 GIF Blast")
-    if st.button("🎉 GIF It", key="imageio_3"):
-        # 🎞️🤡 Makes GIFs like a circus juggler! | 🎥 imageio.mimwrite("gif.gif", [f1, f2], fps=5) - Flips frames into fun!
-        try:
-            reader = imageio.get_reader('<video0>')
-            frames = [reader.get_next_data() for _ in range(10)]
-            file_path = f"imageio_gif_{datetime.now().strftime('%Y%m%d_%H%M%S')}.gif"
-            imageio.mimwrite(file_path, frames, fps=5)
-            save_to_history("🎞️ GIF", file_path)
-            st.image(file_path, caption="🤡 GIF Party")
-        except Exception as e:
-            st.error(f"🚨 Camera snag: {str(e)}")
-
-# 4. PyAV - 🎬 AV rock fest!
-with st.expander("4️⃣ 🎬 PyAV Rave"):
-    st.write("🎥 AV King: FFmpeg-powered chaos!")
-    st.subheader("🔥 Top Tricks")
-
-    st.write("🎥🔥 Video Jam")
-    if st.button("🎬 Roll It", key="pyav_1"):
-        # 🎥🔥 Captures like a rockstar shredding! | 📹 container = av.open('/dev/video0') - Rocks the cam like a live gig!
-        try:
-            container = av.open('/dev/video0')
-            stream = container.streams.video[0]
-            file_path = f"pyav_vid_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp4"
-            output = av.open(file_path, 'w')
-            out_stream = output.add_stream('h264', rate=30)
-            for i, frame in enumerate(container.decode(stream)):
-                if i > 30: break
-                out_frame = frame.reformat(out_stream.width, out_stream.height)
-                output.mux(out_stream.encode(out_frame))
-            output.close()
-            container.close()
-            save_to_history("🎥 Video", file_path)
-            st.video(file_path)
-        except Exception as e:
-            st.error(f"🚨 Camera snag: {str(e)}")
-
-    st.write("🎬🍿 Audio Rip")
-    if st.button("🎵 Snag Sound", key="pyav_2"):
-        # 🎬🍿 Pulls audio like a popcorn thief! | 🎙️ frames = [f for f in av.open('vid.mp4').decode()] - Steals sound like a ninja!
-        try:
-            container = av.open('/dev/video0', 'r', format='v4l2')
-            file_path = f"pyav_audio_{datetime.now().strftime('%Y%m%d_%H%M%S')}.wav"
-            output = av.open(file_path, 'w')
-            out_stream = output.add_stream('pcm_s16le', rate=44100)
-            for packet in container.demux():
-                for frame in packet.decode():
-                    if frame.is_corrupt: continue
-                    if hasattr(frame, 'to_ndarray'):
-                        output.mux(out_stream.encode(frame))
-                if os.path.getsize(file_path) > 100000: break
-            output.close()
-            container.close()
-            save_to_history("🎵 Audio", file_path)
-            st.audio(file_path)
-        except Exception as e:
-            st.error(f"🚨 Camera snag: {str(e)}")
-
-    st.write("🖤✨ Flip Snap")
-    if st.button("🔄 Twist It", key="pyav_3"):
-        # 🖤✨ Flips colors like a rebel teen! | 🎨 graph = av.filter.Graph().add("negate").pull() - Inverts like a goth makeover!
-        try:
-            container = av.open('/dev/video0')
-            stream = container.streams.video[0]
-            file_path = f"pyav_filter_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp4"
-            output = av.open(file_path, 'w')
-            out_stream = output.add_stream('h264', rate=30)
-            graph = av.filter.Graph()
-            filt = graph.add("negate")
-            for i, frame in enumerate(container.decode(stream)):
-                if i > 30: break
-                filt.push(frame)
-                out_frame = filt.pull()
-                output.mux(out_stream.encode(out_frame.reformat(out_stream.width, out_stream.height)))
-            output.close()
-            container.close()
-            save_to_history("🎥 Video", file_path)
-            st.video(file_path, caption="🖤 Flip Vibes")
-        except Exception as e:
-            st.error(f"🚨 Camera snag: {str(e)}")
-
-# 5. MoviePy - 🎥 Editing extravaganza!
-with st.expander("5️⃣ 📼 MoviePy Gala"):
-    st.write("🎥 Edit Queen: Video diva vibes!")
-    st.subheader("🔥 Top Tricks")
-
-    st.write("🎞️💃 Frame Dance")
-    if st.button("🎬 Spin It", key="moviepy_1"):
-        # 🎞️💃 Twirls frames like a dance diva! | 🎥 clip = mp.ImageSequenceClip([f1, f2], fps=15) - Makes vids like a pro choreographer!
-        cap = None
-        for i in range(10):
-            cap = cv2.VideoCapture(i)
-            if cap.isOpened():
-                break
-            cap.release()
-        if cap and cap.isOpened():
-            frames = [cv2.cvtColor(cap.read()[1], cv2.COLOR_BGR2RGB) for _ in range(30)]
-            cap.release()
-            file_path = f"moviepy_seq_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp4"
-            clip = mp.ImageSequenceClip(frames, fps=15)
-            clip.write_videofile(file_path)
-            save_to_history("🎥 Video", file_path)
-            st.video(file_path)
-        else:
-            st.error("🚨 No camera found!")
-
-    st.write("📏👑 Size Snap")
-    if st.button("✂️ Trim It", key="moviepy_2"):
-        # 📏👑 Resizes like a royal tailor! | ✂️ clip = mp.VideoFileClip("vid.mp4").resize((320, 240)) - Fits vids like a bespoke suit!
-        cap = None
-        for i in range(10):
-            cap = cv2.VideoCapture(i)
-            if cap.isOpened():
-                break
-            cap.release()
-        if cap and cap.isOpened():
-            frames = [cv2.cvtColor(cap.read()[1], cv2.COLOR_BGR2RGB) for _ in range(30)]
-            cap.release()
-            temp_path = "temp.mp4"
-            mp.ImageSequenceClip(frames, fps=15).write_videofile(temp_path)
-            clip = mp.VideoFileClip(temp_path).resize((320, 240))
-            file_path = f"moviepy_resized_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp4"
-            clip.write_videofile(file_path)
-            save_to_history("🎥 Video", file_path)
-            st.video(file_path, caption="👑 Tiny King")
-        else:
-            st.error("🚨 No camera found!")
-
-    st.write("🎬🤝 Join Jam")
-    if st.button("🔗 Link It", key="moviepy_3"):
-        # 🎬🤝 Stitches clips like a love guru! | 🎥 final = mp.concatenate_videoclips([clip1, clip2]) - Hooks up vids like a matchmaker!
-        cap = None
-        for i in range(10):
-            cap = cv2.VideoCapture(i)
-            if cap.isOpened():
-                break
-            cap.release()
-        if cap and cap.isOpened():
-            frames1 = [cv2.cvtColor(cap.read()[1], cv2.COLOR_BGR2RGB) for _ in range(15)]
-            frames2 = [cv2.cvtColor(cap.read()[1], cv2.COLOR_BGR2RGB) for _ in range(15)]
-            cap.release()
-            clip1 = mp.ImageSequenceClip(frames1, fps=15)
-            clip2 = mp.ImageSequenceClip(frames2, fps=15)
-            final_clip = mp.concatenate_videoclips([clip1, clip2])
-            file_path = f"moviepy_concat_{datetime.now().strftime('%Y%m%d_%H%M%S')}.mp4"
-            final_clip.write_videofile(file_path)
-            save_to_history("🎥 Video", file_path)
-            st.video(file_path, caption="🤝 Duo Dance")
-        else:
-            st.error("🚨 No camera found!")
-
-# 6. JS Audio - 🎵 Browser beats!
-with st.expander("6️⃣ 🎵 JS Audio Jam"):
-    st.write("🔊 Web Wizard: Browser-based sound sorcery!")
-    st.subheader("🔥 Top Tricks")
-
-    # 🎤🌩️ Record audio with MediaRecorder
-    record_js = """
-    <div>
-        <button id="recordBtn" onclick="startRecording()">🎙️ Record</button>
-        <button id="stopBtn" onclick="stopRecording()" disabled>⏹️ Stop</button>
-        <audio id="audioPlayback" controls></audio>
-    </div>
-    <script>
-        let mediaRecorder;
-        let audioChunks = [];
-        navigator.mediaDevices.getUserMedia({ audio: true })
-            .then(stream => {
-                mediaRecorder = new MediaRecorder(stream);
-                mediaRecorder.ondataavailable = e => audioChunks.push(e.data);
-                mediaRecorder.onstop = () => {
-                    const audioBlob = new Blob(audioChunks, { type: 'audio/wav' });
-                    const audioUrl = URL.createObjectURL(audioBlob);
-                    document.getElementById('audioPlayback').src = audioUrl;
-                    audioChunks = [];
-                };
-            });
-        function startRecording() {
-            mediaRecorder.start();
-            document.getElementById('recordBtn').disabled = true;
-            document.getElementById('stopBtn').disabled = false;
-        }
-        function stopRecording() {
-            mediaRecorder.stop();
-            document.getElementById('recordBtn').disabled = false;
-            document.getElementById('stopBtn').disabled = true;
-        }
-    </script>
-    """
-    st.write("🎤🌩️ Mic Drop")
-    st.markdown(record_js, unsafe_allow_html=True)
-
-    # 🔊📡 Play tone with Web Audio API
-    tone_js = """
-    <button onclick="playTone()">🎶 Beep!</button>
-    <script>
-        function playTone() {
-            const audioCtx = new (window.AudioContext || window.webkitAudioContext)();
-            const oscillator = audioCtx.createOscillator();
-            oscillator.type = 'sine';
-            oscillator.frequency.setValueAtTime(440, audioCtx.currentTime);
-            oscillator.connect(audioCtx.destination);
-            oscillator.start();
-            setTimeout(() => oscillator.stop(), 500);
-        }
-    </script>
-    """
-    st.write("🔊📡 Tone Snap")
-    st.markdown(tone_js, unsafe_allow_html=True)
+st.title("📸 Snap Craze")
+
+# 📸📷 Camera snap zone!
+st.header("📸🎥 Snap Zone")
+cam_img = st.camera_input("📷 Snap It!", key="cam")
+if cam_img:
+    filename = f"cam_snap_{datetime.now().strftime('%Y%m%d_%H%M%S')}.jpg"
+    if st.session_state['cam_file'] and os.path.exists(st.session_state['cam_file']):
+        os.remove(st.session_state['cam_file'])
+    with open(filename, "wb") as f:
+        f.write(cam_img.getvalue())
+    st.session_state['cam_file'] = filename
+    st.session_state['cam_img'] = cam_img  # Store for auto-capture
+    save_to_history("🖼️ Image", filename)
+    st.image(Image.open(filename), caption="Latest Snap", use_column_width=True)
 
 # 📂 Upload zone like a media drop party!
 st.header("📥🎉 Drop Zone")
-uploaded_files = st.file_uploader("📸🎵🎥 Toss Media", accept_multiple_files=True, type=['jpg', 'png', 'mp4', 'wav', 'mp3'])
+uploaded_files = st.file_uploader("📸 Toss Pics", accept_multiple_files=True, type=['jpg', 'png'])
 if uploaded_files:
     for uploaded_file in uploaded_files:
-        file_type = uploaded_file.type.split('/')[0]
         file_path = f"uploaded_{uploaded_file.name}"
-        with open(file_path, 'wb') as f:
+        with open(file_path, "wb") as f:
             f.write(uploaded_file.read())
-        emoji_type = "🖼️ Image" if file_type == 'image' else "🎥 Video" if file_type == 'video' else "🎵 Audio"
-        save_to_history(emoji_type, file_path)
+        save_to_history("🖼️ Image", file_path)
 
-# 🖼️🎵🎥 Gallery like a media circus!
-st.header("🎪 Media Mania")
+# 🖼️ Gallery like a media circus!
+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"]
-    videos = [f for f in st.session_state['file_history'] if f['Type'] in ["🎥 Video", "🎞️ GIF"]]
-
     if images:
         st.subheader("🖼️ Pic Parade")
         cols = st.columns(3)
         for i, img in enumerate(images):
             with cols[i % 3]:
                 st.image(img['Path'], caption=img['Path'], use_column_width=True)
-
-    if audios:
-        st.subheader("🎵 Sound Splash")
-        for audio in audios:
-            st.audio(audio['Path'], format=f"audio/{audio['Path'].split('.')[-1]}")
-            st.write(f"🎤 {audio['Path']}")
-
-    if videos:
-        st.subheader("🎥 Vid Vortex")
-        for video in videos:
-            st.video(video['Path'])
-            st.write(f"🎬 {video['Path']}")
+                st.markdown(f'<a href="data:image/jpeg;base64,{base64.b64encode(open(img["Path"], "rb").read()).decode()}" download="{os.path.basename(img["Path"])}">📥 Snag It!</a>', unsafe_allow_html=True)
 else:
-    st.write("🚫 No loot yet!")
+    st.write("🚫 No snaps yet!")
 
 # 📜 History log like a time machine!
 st.header("⏳ Snap Saga")