Update app.py

app.py

@@ -91,34 +91,16 @@ def download_gitlab_repo():
         print("Upload complete")
 
 def get_all_files_in_folder(temp_dir, partial_paths):
+   
+    print("inner method of get all files in folder")
+    target_dir = os.path.join(temp_dir, partial_path)
+    print(target_dir)
 
-    # List to store paths of all subordinate directories
-    subordinate_dirs = []
-
-    # Walk through the directory structure
-    for root, dirs, files in os.walk(temp_dir, topdown=True):
-        for dir_name in dirs:
-            dir_path = os.path.join(root, dir_name)
-            subordinate_dirs.append(dir_path)
-
-    # Print all subordinate directory paths
-    for path in subordinate_dirs:
-        print(path)
-
-    
-    print("inside the get all files in folder method")
-    all_files = []
-
-    for partial_path in partial_paths:
-        print("inner method of get all files in folder")
-        target_dir = os.path.join(temp_dir, partial_path)
-        print(target_dir)
-
-        for root, dirs, files in os.walk(target_dir):
-            print(f"Files in current directory ({root}): {files}")
-            for file in files:
-                print(f"Processing file: {file}")
-                all_files.append(os.path.join(root, file))
+    for root, dirs, files in os.walk(target_dir):
+        print(f"Files in current directory ({root}): {files}")
+        for file in files:
+            print(f"Processing file: {file}")
+            all_files.append(os.path.join(root, file))
 
     return all_files
 
@@ -159,9 +141,17 @@ def process_directory(directory, partial_paths=None, file_paths=None):
             print(f"Extracted {zip_file_path} to {tmpdirname}")
             
             files = []
+            
+            unzipped_root = os.listdir(tmpdirname)
+            if len(unzipped_root) == 1 and os.path.isdir(os.path.join(tmpdirname, unzipped_root[0])):
+                tmpsubdirpath= os.path.join(tmpdirname, unzipped_root[0])
+            else:
+                tmpsubdirpath = tmpdirname
+                    
             if partial_paths:
                 print("Go in partial_paths")
-                files = get_all_files_in_folder(tmpdirname, partial_paths)
+                for path in partial_paths:
+                files += get_all_files_in_folder(tmpsubdirpath, partial_paths)
             else:
                 print("wtf")
                 for root, _, files_list in os.walk(tmpdirname):