Update app.py

app.py

@@ -11,19 +11,53 @@ EXTRACTED_FOLDER = "nes"  # Directory to serve extracted ROMs
 # 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):
-        with zipfile.ZipFile(ZIP_FILE, 'r') as zip_ref:
-            for file in zip_ref.namelist():
-                if not file.startswith("__MACOSX/"):
-                    zip_ref.extract(file, EXTRACTED_FOLDER)
-                    extracted_path = os.path.join(EXTRACTED_FOLDER, file)
-                    print(f"Extracted: {extracted_path}")  # Debug output
-                    if file.endswith(".zip"):
-                        extract_nested_zip(extracted_path)
-    else:
-        print("Error: `nes.zip` is missing. Upload it to the root directory.")
+#def extract_roms():
+    """
+    Extract ROM files from a zip archive into an extracted folder.
+    
+    Raises:
+        FileNotFoundError: If the zip file is missing or not found in the current directory.
+        PermissionError: When attempting to write data somewhere without sufficient permissions.
+        
+    Returns:
+        list: A list of paths for extracted ROMs.
+    """
+    try:
+        if os.path.exists(ZIP_FILE):
+            with zipfile.ZipFile(ZIP_FILE, 'r') as zip_ref:
+                for file in zip_ref.namelist():
+                    # Ignore special files that do not contain actual data
+                    if not file.startswith("__MACOSX/"):
+                        zip_ref.extract(file, EXTRACTED_FOLDER)
+                        extracted_path = os.path.join(EXTRACTED_FOLDER, file)
+                        print(f"Extracted: {extracted_path}")  
+        else:
+            raise FileNotFoundError("Error: `nes.zip` is missing. Please upload it to the root directory.")
+    
+    except PermissionError as e:
+        # Handle permission errors gracefully
+        print(f"Permission error encountered while extracting ROMs - {e}")
+        
+    return [str(extracted_path) for extracted_path in os.listdir(EXTRACTED_FOLDER)]
+
+def list_roms():
+    """
+    List all available ROM files and construct a dropdown menu with the file paths.
+    
+    Returns:
+        list: A list of paths to extracted ROMs, which are used as options for Gradio's dropdown input.
+        
+    Raises:
+        FileNotFoundError: If the directory containing extracted files does not exist or if `nes.zip` is missing.
+    """
+    try:
+        # Check that a directory exists where our data will be stored and download from 'nes' zip archive
+        rom_files = extract_roms()
+    
+    except FileNotFoundError as e:
+        print(f"Error: ROM folder has been removed. The program should not continue with this action - {e}")
+        
+    return [str(rom) for rom in rom_files]
 
 # Function to extract nested zip files (ROM zips)
 def extract_nested_zip(nested_zip_path):