Update app.py

app.py

@@ -29,7 +29,7 @@ def delete_asset(path):
         st.error(f"Error deleting file: {e}")
     st.rerun()
 
-# --- New Function to Generate Combined PDF ---
+# --- Function to Generate Combined PDF ---
 def generate_combined_pdf(selected_asset_paths):
     """Generates a single PDF from selected markdown and image file paths."""
     buf = io.BytesIO()
@@ -76,7 +76,7 @@ def generate_combined_pdf(selected_asset_paths):
         c.setFont(font_family, font_size)
 
         # Estimate line height and character width for text wrapping
-        # ReportLab units are points. Approximating char width for wrapping.
+        # ReportLab measures in points. Approximating char width for wrapping.
         # A common approximation for average character width is font_size * 0.6
         avg_char_width_points = font_size * 0.6
         # wrap_width is the number of characters that fit in one line of a column
@@ -176,6 +176,7 @@ def generate_combined_pdf(selected_asset_paths):
                     c.showPage() # Start a new page for this image
 
                 # Draw the image onto the current page
+                # Use the path directly with c.drawImage for files on disk
                 c.drawImage(path, pos_x, pos_y, width=draw_w, height=draw_h, preserveAspectRatio=True)
 
             except Exception as e:
@@ -190,7 +191,7 @@ def generate_combined_pdf(selected_asset_paths):
     c.save() # Finalize the PDF
     buf.seek(0) # Rewind the buffer to the beginning
     return buf.getvalue() # Return the PDF bytes
-# --- End of New Function ---
+# --- End of Combined PDF Function ---
 
 
 # Tabs setup
@@ -371,15 +372,17 @@ with tab1:
 
     st.markdown("---")
     st.subheader("📂 Available Assets")
-    st.markdown("Select assets below and click 'Generate Combined PDF'.")
+    st.markdown("Select assets below to include in a combined PDF.")
 
     # Get all files and filter out unwanted ones
     all_assets = glob.glob("*.*")
+    # Removed '.txt' from excluded extensions
     excluded_extensions = ['.py', '.ttf']
-    excluded_files = ['README.md', 'index.html'] # Added index.html here
+    excluded_files = ['README.md', 'index.html']
 
     assets = sorted([
         a for a in all_assets
+        # Check if extension is in excluded list OR if the full name is in excluded files
         if not (a.lower().endswith(tuple(excluded_extensions)) or os.path.basename(a) in excluded_files)
     ])
 
@@ -430,24 +433,29 @@ with tab1:
                     # Audio player takes up too much space here, just offer download
                     with open(a, 'rb') as mp3:
                          cols[2].download_button("📥", data=mp3, file_name=a, mime="audio/mpeg", key=f"download_{a}")
-                # Add more file types here if needed (e.g., images)
-                elif ext in ['png', 'jpg', 'jpeg', 'gif']:
-                     # Can't preview image directly in this column, offer download
+                # Offer download for common text files like txt, csv etc.
+                elif ext in ['md', 'txt', 'csv', 'json', 'xml', 'log']:
+                     with open(a, 'r', encoding='utf-8') as text_file:
+                         cols[2].download_button("⬇️", data=text_file.read(), file_name=a, mime="text/plain", key=f"download_{a}")
+                # Offer download for common image files
+                elif ext in ['png', 'jpg', 'jpeg', 'gif', 'bmp', 'tiff']:
                      with open(a, 'rb') as img_file:
                          cols[2].download_button("⬇️", data=img_file.read(), file_name=a, mime=f"image/{ext}", key=f"download_{a}")
-                elif ext == 'md':
-                    # Offer download for markdown files
-                    with open(a, 'r', encoding='utf-8') as md_file:
-                        cols[2].download_button("⬇️", data=md_file.read(), file_name=a, mime="text/markdown", key=f"download_{a}")
-                # Handle other file types - maybe just offer download
+                # Offer download for common video files (Streamlit doesn't have easy preview here)
+                elif ext in ['mp4', 'webm', 'ogg', 'avi', 'mov']:
+                     with open(a, 'rb') as video_file:
+                         cols[2].download_button("⬇️", data=video_file.read(), file_name=a, mime=f"video/{ext}", key=f"download_{a}")
+                # Handle other file types - maybe just offer download with guessed mime
                 else:
                      with open(a, 'rb') as other_file:
                          cols[2].download_button("⬇️", data=other_file.read(), file_name=a, key=f"download_{a}") # Mime type is guessed by streamlit
 
+
                 # Delete button in the fourth column
                 cols[3].button("🗑️", key=f"del_{a}", on_click=delete_asset, args=(a,))
             except Exception as e:
-                cols[3].error(f"Error handling file {a}: {e}") # Place error in the delete column or add a separate status
+                # Display error next to the file if handling fails
+                cols[3].error(f"Error: {e}")
 
 
     # --- Combined PDF Generation Button ---
@@ -477,11 +485,62 @@ with tab1:
                             )
                             st.success("Combined PDF generated!")
                         else:
-                            st.warning("Generated PDF is empty. No valid markdown or image files were selected.")
+                            # This case might happen if selected files couldn't be read/processed
+                            st.warning("Generated PDF is empty. Check selected files or console for errors.")
 
                     except Exception as e:
                         st.error(f"An unexpected error occurred during PDF generation: {e}")
 
+    # --- Image Gallery ---
+    st.markdown("---")
+    st.subheader("🖼️ Image Gallery")
+    # Find common image file types
+    image_files = sorted(glob.glob("*.png") + glob.glob("*.jpg") + glob.glob("*.jpeg") + glob.glob("*.gif") + glob.glob("*.bmp") + glob.glob("*.tiff"))
+
+    if not image_files:
+        st.info("No image files found in the directory.")
+    else:
+        # Slider to control the number of columns in the gallery
+        image_cols = st.slider("Image Gallery Columns", min_value=1, max_value=10, value=5)
+        # Ensure image_cols is at least 1
+        image_cols = max(1, image_cols)
+
+        # Display images in columns
+        cols = st.columns(image_cols)
+        for idx, image_file in enumerate(image_files):
+            with cols[idx % image_cols]: # Cycle through columns
+                try:
+                    img = Image.open(image_file)
+                    st.image(img, caption=os.path.basename(image_file), use_container_width=True)
+                except Exception as e:
+                    st.warning(f"Could not display image {image_file}: {e}")
+
+
+    # --- Video Gallery ---
+    st.markdown("---")
+    st.subheader("🎥 Video Gallery")
+    # Find common video file types
+    video_files = sorted(glob.glob("*.mp4") + glob.glob("*.webm") + glob.glob("*.ogg") + glob.glob("*.avi") + glob.glob("*.mov"))
+
+    if not video_files:
+        st.info("No video files found in the directory.")
+    else:
+        # Slider to control the number of columns in the gallery
+        video_cols = st.slider("Video Gallery Columns", min_value=1, max_value=5, value=3)
+        # Ensure video_cols is at least 1
+        video_cols = max(1, video_cols)
+
+
+        # Display videos in columns
+        cols = st.columns(video_cols)
+        for idx, video_file in enumerate(video_files):
+            with cols[idx % video_cols]: # Cycle through columns
+                 try:
+                      # Streamlit's built-in video player is simpler than custom HTML
+                      st.video(video_file, caption=os.path.basename(video_file))
+                 except Exception as e:
+                      st.warning(f"Could not display video {video_file}: {e}")
+
 
 with tab2:
     st.header("🧪 Python Code Executor & Demo")