Update app.py

app.py

@@ -91,30 +91,32 @@ def download_gitlab_repo():
         print("Upload complete")
 
 
+def extract_single_zip_file(directory):
+    zip_file = [file for file in os.listdir(directory) if file.endswith('.zip')]
+
+    if not zip_file:
+        print("No zip file found in the directory.")
+    else:
+        zip_filename = zip_file  # Get the only zip file
+        zip_file_path = os.path.join(directory, zip_file)
+        
+        # Create a temporary directory for the zip file
+        with tempfile.TemporaryDirectory() as tmpdirname:
+            # Unzip the file into the temporary directory
+            with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
+                zip_ref.extractall(tmpdirname)
+            print(f"Extracted {zip_filename} to {tmpdirname}")
+            return tmpdirname
+
+    
+
 def process_directory(directory):
     all_texts = []
     file_references = []
 
-    zip_files = [file for file in os.listdir(directory) if file.endswith('.zip')]
-    
-    if not zip_files:
-        print("No zip files found in the directory.")
-    else:
-        for zip_filename in zip_files:
-            zip_file_path = os.path.join(directory, zip_filename)
-            
-            # Create a temporary directory for each zip file
-            with tempfile.TemporaryDirectory() as tmpdirname:
-                # Unzip the file into the temporary directory
-                with zipfile.ZipFile(zip_file_path, 'r') as zip_ref:
-                    zip_ref.extractall(tmpdirname)
-                
-                # Process the temporary directory
-                temp_texts, temp_references = process_directory(tmpdirname)
-                all_texts.extend(temp_texts)
-                file_references.extend(temp_references)
+    directory_path = extract_single_zip_file(directory)
     
-    for root, _, files in os.walk(directory):
+    for root, _, files in os.walk(directory_path):
         for file in files:
             file_path = os.path.join(root, file)
             file_ext = os.path.splitext(file_path)[1]