Create app.py

app.py

@@ -0,0 +1,116 @@
+import gradio as gr
+import zipfile
+import os
+import requests
+from pathlib import Path
+
+# Configuration
+# SECURE: Get the link from the environment variable "link"
+ZIP_URL = os.environ.get("link")
+ZIP_FILE = "consolidated_epstein_media.zip"
+EXTRACT_PATH = "extracted_media"
+
+def get_images():
+    """Scans the extraction folder for images."""
+    image_extensions = {".jpg", ".jpeg", ".png", ".bmp", ".gif", ".webp"}
+    images = []
+    if os.path.exists(EXTRACT_PATH):
+        for root, dirs, files in os.walk(EXTRACT_PATH):
+            for file in files:
+                if Path(file).suffix.lower() in image_extensions:
+                    images.append(os.path.join(root, file))
+    return sorted(images)
+
+def process_files(progress=gr.Progress()):
+    """Downloads and unzips the file with progress bars."""
+    
+    # Security Check
+    if not ZIP_URL:
+        return [("Error: Secret 'link' not found. Please add it in Settings > Repository Secrets.", "Error")]
+
+    # 1. Download Phase
+    if not os.path.exists(ZIP_FILE):
+        print("Starting download...")
+        try:
+            response = requests.get(ZIP_URL, stream=True)
+            if response.status_code != 200:
+                return [f"Error downloading: HTTP {response.status_code}"]
+                
+            total_size = int(response.headers.get('content-length', 0))
+            block_size = 1024 * 1024  # 1MB chunks
+            
+            with open(ZIP_FILE, 'wb') as file:
+                downloaded = 0
+                for data in response.iter_content(block_size):
+                    file.write(data)
+                    downloaded += len(data)
+                    if total_size > 0:
+                        # Report download progress (0.0 to 0.5 of total bar)
+                        percentage = (downloaded / total_size) * 0.5
+                        progress(percentage, desc=f"Downloading zip... {int((downloaded/total_size)*100)}%")
+        except Exception as e:
+            return [f"Error downloading file: {str(e)}"]
+
+    # 2. Unzip Phase
+    if not os.path.exists(EXTRACT_PATH):
+        print("Starting unzip...")
+        try:
+            with zipfile.ZipFile(ZIP_FILE, 'r') as zip_ref:
+                file_list = zip_ref.namelist()
+                total_files = len(file_list)
+                
+                os.makedirs(EXTRACT_PATH, exist_ok=True)
+                
+                for index, file in enumerate(file_list):
+                    if index % 10 == 0:
+                        # Report unzip progress (0.5 to 1.0 of total bar)
+                        percentage = 0.5 + ((index / total_files) * 0.5)
+                        progress(percentage, desc=f"Unzipping files... {int((index/total_files)*100)}%")
+                    
+                    zip_ref.extract(file, EXTRACT_PATH)
+        except zipfile.BadZipFile:
+            return ["Error: The downloaded file is not a valid zip archive."]
+            
+    progress(1.0, desc="Done!")
+    return get_images()
+
+# CSS for clean, sharp UI
+custom_css = """
+.gradio-container {
+    background-color: white !important;
+}
+/* Force sharp corners and black borders on images */
+button.gallery-item {
+    border-radius: 0px !important;
+    border: 2px solid black !important;
+    overflow: hidden;
+}
+img {
+    border-radius: 0px !important;
+}
+/* Hide default rounded containers */
+.block {
+    border: none !important;
+    background: transparent !important;
+}
+"""
+
+with gr.Blocks(css=custom_css, title="Media Viewer") as demo:
+    gr.Markdown("### Consolidated Media Viewer")
+    
+    # Gallery with lazy loading
+    gallery = gr.Gallery(
+        label="Media Gallery", 
+        show_label=False, 
+        columns=[4], 
+        rows=[2], 
+        object_fit="contain", 
+        height="80vh",
+        interactive=False
+    )
+
+    # Trigger process on load
+    demo.load(process_files, inputs=None, outputs=gallery)
+
+if __name__ == "__main__":
+    demo.queue().launch(server_name="0.0.0.0", server_port=7860)