import streamlit as st
import os
import glob
import zipfile

def extract_zip(zip_path, extract_to="extracted_images"):
    with zipfile.ZipFile(zip_path, 'r') as zip_ref:
        zip_ref.extractall(extract_to)
    return extract_to

search_urls = {
    "📖": lambda k: f"https://en.wikipedia.org/wiki/{quote(k)}",
    "🔍": lambda k: f"https://www.google.com/search?q={quote(k)}",
    "▶️": lambda k: f"https://www.youtube.com/results?search_query={quote(k)}",
    "🔎": lambda k: f"https://www.bing.com/search?q={quote(k)}",
    "🎲": lambda k: f"https://huggingface.co/spaces/awacke1/AI-ChatGPT-CPT-Body-Map-Cost?q={quote(k)}",
}

def generate_search_links(filename):
    # Remove the file extension and replace underscores with spaces for search terms
    search_term = filename.rsplit('.', 1)[0].replace("_", " ")
    # Generate markdown links for each search URL
    links_md = ' '.join([f"[{emoji}]({url(search_term)})" for emoji, url in search_urls.items()])
    # Display the markdown with Streamlit
    st.markdown(f"Explore **{search_term}** in: {links_md}", unsafe_allow_html=True)


def main():
    st.title('Zip File Image Gallery')

    # Check for zip files in the current directory
    zip_files = glob.glob('*.zip')
    if zip_files:
        zip_file = zip_files[0]
        st.write(f"Found zip file: {zip_file}")
        file_size = os.path.getsize(zip_file)
        st.write(f"Size: {file_size / (1024 * 1024):.2f} MB")

        # Extract the zip file
        if st.button('Extract Zip File'):
            extract_path = extract_zip(zip_file)
            st.success(f'Extracted to {extract_path}')

            # Display video gallery
            #videos = glob.glob(f'{extract_path}/*.mp4')
            #for video in videos:
            #    st.video(video, format="video/mp4", start_time=0)

            # Display image gallery
            images = glob.glob(f'{extract_path}/*.png')  # Adjusted to include all image file types
            for image in images:
                st.image(image, caption=os.path.basename(image), width=300)
                generate_search_links(os.path.basename(image))
                
    else:
        st.write("No zip files found in the directory.")

if __name__ == "__main__":
    main()