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

# 🌟🔥 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:
    st.session_state['auto_capture_running'] = False

# 📜💾 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({
        "Timestamp": timestamp,
        "Type": file_type,
        "Path": 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!")
        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)
    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")
    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!")

# 🌍🎨 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)

# 📂 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'])
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:
            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)

# 🖼️🎵🎥 Gallery like a media circus!
st.header("🎪 Media Mania")
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']}")
else:
    st.write("🚫 No loot yet!")

# 📜 History log like a time machine!
st.header("⏳ Snap Saga")
if st.session_state['file_history']:
    df = pd.DataFrame(st.session_state['file_history'])
    st.dataframe(df)
else:
    st.write("🕳️ Nothing snapped yet!")