Update app.py

app.py

@@ -16,8 +16,7 @@ This tool allows you to analyze various file types:
 - View information about .dll files
 """)
 
-# Maximum file size (10MB)
-MAX_FILE_SIZE = 10 * 1024 * 1024
+# No size limit
 
 def unpack_exe(file_path, output_dir):
     """Extract information from an EXE file using pefile"""
@@ -166,61 +165,58 @@ def process_zip_file(file_path, temp_dir):
 uploaded_file = st.file_uploader("Upload a file (.zip, .exe, or .dll)", type=["zip", "exe", "dll"])
 
 if uploaded_file is not None:
-    # Check file size
-    if uploaded_file.size > MAX_FILE_SIZE:
-        st.error(f"File too large. Maximum size is {MAX_FILE_SIZE/1024/1024}MB.")
-    else:
-        with tempfile.TemporaryDirectory() as temp_dir:
-            # Save the uploaded file to the temporary directory
-            file_path = os.path.join(temp_dir, uploaded_file.name)
-            with open(file_path, "wb") as f:
-                f.write(uploaded_file.getbuffer())
+    # Process file without size check
+    with tempfile.TemporaryDirectory() as temp_dir:
+        # Save the uploaded file to the temporary directory
+        file_path = os.path.join(temp_dir, uploaded_file.name)
+        with open(file_path, "wb") as f:
+            f.write(uploaded_file.getbuffer())
+        
+        st.success(f"File uploaded: {uploaded_file.name}")
+        
+        # Process based on file type
+        if uploaded_file.name.lower().endswith('.zip'):
+            st.subheader("ZIP File Contents")
+            output_dir = os.path.join(temp_dir, "extracted")
+            os.makedirs(output_dir, exist_ok=True)
             
-            st.success(f"File uploaded: {uploaded_file.name}")
+            result = process_zip_file(file_path, output_dir)
             
-            # Process based on file type
-            if uploaded_file.name.lower().endswith('.zip'):
-                st.subheader("ZIP File Contents")
-                output_dir = os.path.join(temp_dir, "extracted")
-                os.makedirs(output_dir, exist_ok=True)
+            if 'error' in result:
+                st.error(f"Error processing ZIP file: {result['error']}")
+            else:
+                with st.expander("ZIP Contents", expanded=True):
+                    st.write(f"Total files: {len(result['file_list'])}")
+                    st.json(result['file_list'])
                 
-                result = process_zip_file(file_path, output_dir)
-                
-                if 'error' in result:
-                    st.error(f"Error processing ZIP file: {result['error']}")
-                else:
-                    with st.expander("ZIP Contents", expanded=True):
-                        st.write(f"Total files: {len(result['file_list'])}")
-                        st.json(result['file_list'])
-                    
-                    if result['nested_files']:
-                        st.subheader("Detected Executable Files")
-                        for file_path, file_info in result['nested_files'].items():
-                            with st.expander(f"{file_path} ({file_info['type'].upper()})"):
-                                st.json(file_info['info'])
+                if result['nested_files']:
+                    st.subheader("Detected Executable Files")
+                    for file_path, file_info in result['nested_files'].items():
+                        with st.expander(f"{file_path} ({file_info['type'].upper()})"):
+                            st.json(file_info['info'])
+        
+        elif uploaded_file.name.lower().endswith('.exe'):
+            st.subheader("EXE File Analysis")
+            output_dir = os.path.join(temp_dir, "exe_unpacked")
+            os.makedirs(output_dir, exist_ok=True)
             
-            elif uploaded_file.name.lower().endswith('.exe'):
-                st.subheader("EXE File Analysis")
-                output_dir = os.path.join(temp_dir, "exe_unpacked")
-                os.makedirs(output_dir, exist_ok=True)
+            try:
+                exe_info = unpack_exe(file_path, output_dir)
+                st.json(exe_info)
                 
-                try:
-                    exe_info = unpack_exe(file_path, output_dir)
-                    st.json(exe_info)
-                    
-                    # Check if resources were extracted
-                    resource_dir = os.path.join(output_dir, "resources")
-                    if os.path.exists(resource_dir) and os.listdir(resource_dir):
-                        st.subheader("Extracted Resources")
-                        for resource_file in os.listdir(resource_dir):
-                            st.text(f"Resource: {resource_file}")
-                except Exception as e:
-                    st.error(f"Error analyzing EXE file: {str(e)}")
-            
-            elif uploaded_file.name.lower().endswith('.dll'):
-                st.subheader("DLL File Information")
-                try:
-                    dll_info = analyze_dll(file_path)
-                    st.json(dll_info)
-                except Exception as e:
-                    st.error(f"Error analyzing DLL file: {str(e)}")
+                # Check if resources were extracted
+                resource_dir = os.path.join(output_dir, "resources")
+                if os.path.exists(resource_dir) and os.listdir(resource_dir):
+                    st.subheader("Extracted Resources")
+                    for resource_file in os.listdir(resource_dir):
+                        st.text(f"Resource: {resource_file}")
+            except Exception as e:
+                st.error(f"Error analyzing EXE file: {str(e)}")
+        
+        elif uploaded_file.name.lower().endswith('.dll'):
+            st.subheader("DLL File Information")
+            try:
+                dll_info = analyze_dll(file_path)
+                st.json(dll_info)
+            except Exception as e:
+                st.error(f"Error analyzing DLL file: {str(e)}")