Update app.py

app.py

@@ -140,7 +140,7 @@ Berikut kami informasikan kekurangan dokumen yang *WAJIB* dibawa saat biometric
       )
     analysis = response_1.text
     # Perform valid list generation
-    prompt_2 = "Dari analisa diatas, Rangkum menjadi list/daftar dokumen yang valid saja. Misal: 'paspor', 'pasfoto', 'kartukeluarga', 'buktinikah', 'KTP', 'suratkelahiran', 'suratsponsor', 'suratkerja', 'NIB', 'SIUP', 'suratjaminanstaff', 'suratsekolah', 'kontrakkerja', 'suratpensiun', 'rekeningkoran', 'slipgaji'. Jangan tambahkan kata-kata atau kalimat lain. Tidak boleh ada file di luar contoh."
+    prompt_2 = "Dari analisa diatas, Rangkum menjadi list/daftar dokumen YANG TIDAK VALID SAJA. Misal: 'paspor', 'pasfoto', 'kartukeluarga', 'buktinikah', 'KTP', 'suratkelahiran', 'suratsponsor', 'suratkerja', 'NIB', 'SIUP', 'suratjaminanstaff', 'suratsekolah', 'kontrakkerja', 'suratpensiun', 'rekeningkoran', 'slipgaji'. Jangan tambahkan kata-kata atau kalimat lain. Tidak boleh ada file di luar contoh."
     response_2 = client.models.generate_content(
         model="gemini-2.0-flash-lite",
         contents=[analysis, prompt_2],  # Pass prompt and image files
@@ -163,8 +163,7 @@ Berikut kami informasikan kekurangan dokumen yang *WAJIB* dibawa saat biometric
     return analysis, docs_list
 
 
-def process_and_zip_valid_images(images, docs_list, api_key, zip_name="Valid_Docs.zip"):
-
+def process_and_zip_all_images(images, api_key, zip_name="Valid_Docs.zip"):
     # Inisialisasi Gemini client
     client = genai.Client(api_key=api_key)
 
@@ -178,7 +177,6 @@ Jangan tambahkan penjelasan, tanda kutip, titik, atau teks tambahan lainnya.
 Contoh:
 Gambar : [tampak gambar KTP]
 Output: KTP
-
 Gambar: [gambar sebenarnya]
 Output:
 '''
@@ -197,19 +195,13 @@ Output:
         filename = response.text.strip().lower()
         renamed_images.append({"image": image, "filename": filename})
 
-    # Step 2: Filter sesuai valid_list
-    filtered_images = [
-        item for item in renamed_images
-        if os.path.splitext(item["filename"])[0] in [v.lower() for v in docs_list]
-    ]
-
-    # Step 3: Kelompokkan berdasarkan nama file (tanpa .jpg)
+    # Step 2: Kelompokkan berdasarkan nama file (tanpa ekstensi)
     grouped = {}
-    for item in filtered_images:
+    for item in renamed_images:
         name = os.path.splitext(item["filename"])[0]
         grouped.setdefault(name, []).append(item["image"])
 
-    # Step 4: Simpan ke PDF dan masukkan ke ZIP
+    # Step 3: Simpan ke PDF dan masukkan ke ZIP
     temp_dir = tempfile.mkdtemp()
     zip_path = os.path.join(tempfile.gettempdir(), zip_name)
 
@@ -271,7 +263,7 @@ def main_process(files, tanggal_berangkat, tanggal_pulang, api_key):
     doc.save(temp_docx_path)
 
     # Filtering the file
-    zip_file_path = process_and_zip_valid_images(all_images, valid_list, api_key, zip_name=f'Valid_ZIP_{base_name}.zip')
+    zip_file_path = process_and_zip_valid_images(all_images, api_key, zip_name=f'Valid_ZIP_{base_name}.zip')
     return temp_docx_path, valid_list, zip_file_path