Update app.py

app.py

@@ -291,15 +291,21 @@ def process_zip(zip_file, threshold):
         with zipfile.ZipFile(zip_file.name, 'r') as zip_ref:
             zip_ref.extractall(temp_dir)
         
-        all_files = [os.path.join(temp_dir, f) for f in os.listdir(temp_dir)]
+        all_files = []
+        for root, _, files in os.walk(temp_dir):
+            for file in files:
+                all_files.append(os.path.join(root, file))
+        
         image_files = [f for f in all_files if is_valid_image(f)]
         results = process_images(image_files, threshold)
     
         temp_file = NamedTemporaryFile(delete=False, suffix=".zip")
         with zipfile.ZipFile(temp_file, "w") as zip_ref:
             for image_name, text_no_impl, _ in results:
-                with zip_ref.open(''.join(image_name.split('.')[:-1]) + ".txt", 'w') as file:
+                txt_filename = ''.join(image_name.split('.')[:-1]) + ".txt"
+                with zip_ref.open(txt_filename, 'w') as file:
                     file.write(text_no_impl.encode())
+        
         temp_file.seek(0)
         df = pd.DataFrame([(os.path.basename(f), t) for f, t, _ in results], columns=['Image', 'Tags'])