Update app.py

app.py

@@ -24,7 +24,7 @@ def extract_and_combine_zip(zip_file):
                     data = json.loads(decoded_content)
                     combined_data = flatten_json(data, combined_data)
                 except (UnicodeDecodeError, json.JSONDecodeError) as e:
-                    st.warning(f"Warning: Could not decode {json_file}. Error: {str(e)}")
+                    st.warning(f"Warning: Could not decode {json_file} in {zip_file.name}. Error: {str(e)}")
     return combined_data
 
 # Improved function to flatten and merge JSON data
@@ -57,26 +57,36 @@ def to_excel(df):
     return processed_data
 
 # Streamlit app setup
-st.title("ZIP JSON Extractor & Flattener")
+st.title("Multi-ZIP JSON Extractor & Flattener")
 
-# File uploader widget
-uploaded_zip = st.file_uploader("Upload ZIP file containing JSON files:", type="zip")
+# File uploader widget for multiple files
+uploaded_zips = st.file_uploader("Upload ZIP files containing JSON files:", type="zip", accept_multiple_files=True)
 
-if uploaded_zip:
-    # Combine and flatten JSON data
-    flattened_json = extract_and_combine_zip(uploaded_zip)
+if uploaded_zips:
+    # Process all uploaded ZIP files
+    all_flattened_json = {}
+    total_files_processed = 0
+    
+    for zip_file in uploaded_zips:
+        flattened_json = extract_and_combine_zip(zip_file)
+        all_flattened_json.update(flattened_json)
+        total_files_processed += 1
 
     # Create a DataFrame from the flattened JSON data
-    df = pd.DataFrame([flattened_json])
+    df = pd.DataFrame([all_flattened_json])
 
     # Convert all object columns to string to avoid Arrow conversion issues
     for col in df.select_dtypes(include=['object']).columns:
         df[col] = df[col].astype(str)
 
     # Create a downloadable JSON
-    flattened_json_str = json.dumps(flattened_json, indent=4)
+    flattened_json_str = json.dumps(all_flattened_json, indent=4)
     json_bytes = flattened_json_str.encode()
 
+    # Display processing summary
+    st.write(f"Processed {total_files_processed} ZIP file(s)")
+    st.write(f"Total JSON objects flattened: {len(all_flattened_json)}")
+
     # Create columns for download buttons
     col1, col2 = st.columns(2)
 
@@ -100,4 +110,4 @@ if uploaded_zip:
         )
 
     # Add a success message
-    st.success("JSON data has been successfully processed and flattened into a single object. You can now download the flattened JSON file or the Excel file.")
+    st.success(f"All ZIP files have been successfully processed. JSON data has been flattened into a single object. You can now download the flattened JSON file or the Excel file.")