Update app.py

app.py

@@ -5,22 +5,22 @@ import os
 def unzip_file(file):
     extract_path = "extracted_files"  # Define a path to extract files
     os.makedirs(extract_path, exist_ok=True)  # Create the directory if it doesn't exist
-    file_names = []
+    jpg_files = []  # List to store paths of JPG files
     with zipfile.ZipFile(file, "r") as zip_ref:
         zip_ref.extractall(extract_path)  # Extract files into the specified directory
-        # Walk through the directory structure and add files to the list
+        # Walk through the directory structure and look for JPG files
         for root, dirs, files in os.walk(extract_path):
             for file in files:
-                if not file.startswith('.'):
-                    # Construct the file's full path
+                if file.lower().endswith('.jpg'):  # Check if the file is a JPG
                     full_path = os.path.join(root, file)
-                    # Subtract the base extraction path to show a relative path
-                    relative_path = os.path.relpath(full_path, extract_path)
-                    file_names.append(relative_path)
+                    jpg_files.append(full_path)
 
-    return '\n'.join(file_names)  # Join the list into a single string separated by newlines
+    if not jpg_files:
+        return ["No JPG files found in the zip."]  # Return a message if no JPGs are found
+    return jpg_files  # Return the list of JPG file paths
 
-interface = gr.Interface(fn=unzip_file, inputs="file", outputs="text")
+# Define the Gradio interface, specifying image display for multiple images
+interface = gr.Interface(fn=unzip_file, inputs="file", outputs=gr.Gallery())
 interface.launch()