mkaramb's picture
Update app.py
c8197d8 verified
raw
history blame
1.24 kB
import zipfile
import gradio as gr
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
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 look for JPG files
for root, dirs, files in os.walk(extract_path):
for file in files:
if file.lower().endswith('.jpg'): # Check if the file is a JPG
full_path = os.path.join(root, file)
jpg_files.append(full_path)
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
# Define the Gradio interface, specifying image display for multiple images
interface = gr.Interface(fn=unzip_file, inputs="file", outputs=gr.Gallery())
interface.launch()
# def greet(name):
# return "Hello " + name + "!!"
#iface = gr.Interface(fn=greet, inputs="text", outputs="text")
#iface.launch()