Update app.py

app.py

@@ -3,13 +3,30 @@ import os
 import glob
 import zipfile
 
-def extract_zip(zip_path, extract_to="extracted_videos"):
+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 Video Gallery')
+    st.title('Zip File Image Gallery')
 
     # Check for zip files in the current directory
     zip_files = glob.glob('*.zip')
@@ -25,10 +42,16 @@ def main():
             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)
+            #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.")