Create app.py

app.py

@@ -0,0 +1,67 @@
+import gradio as gr
+import os
+import zipfile
+
+# Define paths
+ZIP_FILE = "nes.zip"
+EXTRACTED_FOLDER = "nes_extracted"
+
+# Ensure extraction directory exists
+os.makedirs(EXTRACTED_FOLDER, exist_ok=True)
+
+# Function to extract ROMs and nested zips
+def extract_roms():
+    if os.path.exists(ZIP_FILE):
+        # Extract the main nes.zip
+        with zipfile.ZipFile(ZIP_FILE, 'r') as zip_ref:
+            for file in zip_ref.namelist():
+                if not file.startswith("__MACOSX/"):  # Skip MacOS metadata
+                    # Extract the files from nes.zip
+                    zip_ref.extract(file, EXTRACTED_FOLDER)
+                    # If the file is a zip (potential ROM zip), extract it too
+                    if file.endswith(".zip"):
+                        nested_zip_path = os.path.join(EXTRACTED_FOLDER, file)
+                        extract_nested_zip(nested_zip_path)
+    else:
+        print("Error: `nes.zip` is missing. Upload it to the root directory.")
+
+# Function to extract nested zip files (ROM zips)
+def extract_nested_zip(nested_zip_path):
+    with zipfile.ZipFile(nested_zip_path, 'r') as nested_zip:
+        nested_folder = os.path.splitext(nested_zip_path)[0]  # Get folder name from zip filename
+        os.makedirs(nested_folder, exist_ok=True)
+        nested_zip.extractall(nested_folder)
+        os.remove(nested_zip_path)  # Optional: remove the nested zip after extracting
+
+# Ensure ROMs are extracted on startup
+extract_roms()
+
+# Function to list available ROMs
+def list_roms():
+    if not os.path.exists(EXTRACTED_FOLDER):
+        return []
+    # Flatten all files from subdirectories
+    files = []
+    for root, dirs, filenames in os.walk(EXTRACTED_FOLDER):
+        for filename in filenames:
+            if not filename.startswith("__MACOSX"):  # Skip MacOS metadata files
+                files.append(os.path.join(root, filename))
+    return files if files else ["No ROMs available"]
+
+# Function to return file path for download
+def get_rom(file_name):
+    return file_name if os.path.exists(file_name) else "File not found"
+
+# Gradio App
+with gr.Blocks() as app:
+    gr.Markdown("# 🎮 Retro ROM File Server")
+
+    # Dropdown to select ROM
+    rom_dropdown = gr.Dropdown(choices=list_roms(), label="Select a ROM to Download", interactive=True)
+    download_button = gr.Button("Download Selected ROM")
+    file_output = gr.File(label="Download Here")
+
+    # Button Click Event
+    download_button.click(get_rom, inputs=rom_dropdown, outputs=file_output)
+
+app.launch()