Update app.py

app.py

@@ -13,6 +13,7 @@ import time
 import random
 from google.genai.types import GenerateContentConfig
 import json
+from PIL import ImageEnhance, ImageFilter
 
 
 def extract_zip_and_collect_files(zip_file_path):
@@ -47,6 +48,19 @@ def process_pdfs(pdf_files, dpi):
 
     return all_images
 
+def preprocess_image(img):
+    # Enhance contrast and sharpness
+    img = ImageEnhance.Contrast(img).enhance(1.5)
+    img = ImageEnhance.Sharpness(img).enhance(2.0)
+    
+    # Convert to grayscale for better OCR performance
+    img = img.convert('L')
+    
+    # Resize for optimal resolution (maintain aspect ratio)
+    max_size = (1600, 1600)
+    img.thumbnail(max_size, Image.LANCZOS)
+    
+    return img
 
 # Function to analyze the extracted image using Google GenAI
 def gemini_analysis(images, tanggal_berangkat, tanggal_pulang, tanggal_biometrik):
@@ -56,27 +70,26 @@ def gemini_analysis(images, tanggal_berangkat, tanggal_pulang, tanggal_biometrik
     Pastikan data seperti nama, tanggal, tujuan dan lainnya konsisten antar dokumen.
     ---
     DAFTAR SYARAT DOKUMEN YANG WAJIB DIPERIKSA:
-1. **Paspor**: Aktif min. 6 bulan setelah tanggal pulang, ada tanda tangan, lampirkan paspor lama jika ada, lampirkan fotocopy / scan paspor sponsor jika disponsori.
-2. **Paspor Lama**: Opsional. Halaman depan & tanda tangan, semua visa sebelumnya (terutama 5 tahun terakhir).
-3. **Fotokopi Paspor**: Wajib. Halaman depan & tanda tangan, semua visa sebelumnya (terutama 5 tahun terakhir).
-4. **Pas Foto**: Background putih, wajah terlihat 80%, alis terlihat, tidak berbayang.
-5. **Kartu Keluarga**: Wajib. Versi 2019 atau ada barcode, sesuai status, translate jika untuk VFS Germany.
-6. **Bukti Nikah/Cerai**: Wajib jika sudah menikah/bercerai, sertakan halaman biodata suami/istri & surat izin suami jika istri ikut.
+1. **Paspor**. Wajib. Aktif min. 6 bulan setelah tanggal pulang, ada tanda tangan, lampirkan paspor lama jika ada, lampirkan fotocopy / scan paspor sponsor jika disponsori.
+2. **Paspor Lama**. Opsional. Halaman depan & tanda tangan, semua visa sebelumnya (terutama 5 tahun terakhir).
+3. **Fotokopi Paspor**. Wajib. Halaman depan & tanda tangan, semua visa sebelumnya (terutama 5 tahun terakhir).
+4. **Pas Foto**. Background putih, wajah terlihat 80%, alis terlihat, tidak berbayang.
+5. **Kartu Keluarga**. Wajib. Versi 2019 atau ada barcode, sesuai status, translate jika untuk VFS Germany.
+6. **Bukti Nikah/Cerai**. Wajib jika sudah menikah/bercerai, sertakan halaman biodata suami/istri & surat izin suami jika istri ikut.
 7. **KTP**: Wajib. Nama harus sama dengan paspor & dokumen lain. Jika beda nama, wajib ada surat keterangan beda nama.
 8. **KTP Sponsor**: Wajib jika disponsori.
-9. **Bukti Kelahiran/Ijazah**: Optional. Wajib jika anak-anak atau disponsori orang lain.
+9. **Bukti Kelahiran/Ijazah**: Optional. Wajib untuk anak-anak atau disponsori orang lain.
 10. **Surat Sponsor**: Berbahasa Inggris, berisi tujuan, tanggal trip, penanggung biaya, jaminan kembali ke Indonesia.
-11. **Status Pekerjaan (pilih salah satu)**:
+11. **Status Pekerjaan (hanya salah satu)**:
      - Pegawai: Surat kerja + slip gaji 3 bulan
      - Pemilik usaha: NIB/SIUP + surat jaminan staf
      - Pelajar: Surat sekolah/universitas + kartu pelajar
      - Freelancer/Online shop: Kontrak + 5 bukti transaksi
      - Pensiun: Surat pensiun + guarantee keluarga
-12. **Rekening Koran Peserta**: Atas nama pribadi, tertera cap & logo bank, nama & nomor rekening, saldo minimal Rp 40 juta/orang, wajib terupdate hingga. 7 hari sebelum tanggal biometrik. Laporkan hingga hasil analisis.
-13. **Rekening Koran Sponsor**: Wajib ada jika disponsori. tertera cap & logo bank, nama & nomor rekening, saldo minimal Rp 40 juta/orang, wajib terupdate hingga. 7 hari sebelum tanggal biometrik. Laporkan hingga hasil analisis.
-14. **Slip Gaji**: Wajib jika pekerja. 3 bulan terakhir. Tidak perlu jika istri disponsori suami.
-15. **Surat Referensi Bank**: Wajib ada. Surat resmi dari bank yang menyatakan peserta adalah nasabah. Berbeda dengan rekening koran dan tidak bisa digantikan dengan rekening koran.
-16. **Surat Referensi Bank Sponsor**: Wajib jika disponsori. Harus berupa surat resmi dari bank, bukan rekening koran saja. Tidak bisa digantikan dengan rekening koran.
+12. **Rekening Koran Peserta**. Wajib. Atas nama pribadi, tertera cap & logo bank, nama & nomor rekening, saldo minimal Rp 40 juta/orang, wajib terupdate hingga. 7 hari sebelum tanggal biometrik. Laporkan hingga hasil analisis.
+13. **Rekening Koran Sponsor**. Optional. Wajib ada jika disponsori. tertera cap & logo bank, nama & nomor rekening, saldo minimal Rp 40 juta/orang, wajib terupdate hingga. 7 hari sebelum tanggal biometrik. Laporkan hingga hasil analisis.
+14. **Surat Referensi Bank**: Wajib ada. Surat resmi dari bank yang menyatakan peserta adalah nasabah. Berbeda dengan rekening koran dan tidak bisa digantikan dengan rekening koran.
+15. **Surat Referensi Bank Sponsor**: Wajib jika disponsori. Harus berupa surat resmi dari bank, bukan rekening koran saja. Tidak bisa digantikan dengan rekening koran.
     ---
     TEMPLATE OUTPUT JAWABAN DALAM FORMAT JSON:
     {
@@ -125,13 +138,13 @@ def gemini_analysis(images, tanggal_berangkat, tanggal_pulang, tanggal_biometrik
     ---
     TEMPLATE PESAN PEMBERITAHUAN (notice_msg):
     Berikut kami informasikan kekurangan dokumen yang *WAJIB* dibawa saat biometric visa schengen nanti ya :
-    1. Pas Foto (Wajib berukuran 3,5 x 4,5 cm dan diserahkan di lokasi biometrik) # ini wajib dicantumkan dan dibawa terlepas valid atau tidak
+    1. Pas Foto (Wajib berukuran 3,5 x 4,5 cm dan diserahkan di lokasi biometrik)    ## ini wajib dicantumkan dan dibawa
     2. ...
     3. ...
     4. ...
     5. ...
     ---
-    # cantumkan dokumen lain yang masih Invalid dan perlu perbaikan.
+    # cantumkan dokumen lain yang masih Invalid beserta alasan invalidnya.
     ---
     TEMPLATE SUMMARY:
     List Dokumen yang sudah valid: .... \n
@@ -266,7 +279,8 @@ def main_process(files, tanggal_berangkat, tanggal_pulang, tanggal_biometrik, dp
             raise ValueError(f"File {file_path} is not a valid image, PDF, or ZIP.")
 
     # Generate summary from images
-    raw_output, analysis_str, summary, invalid_list_str, notice_msg, form_filling_str, token_usage = gemini_analysis(all_images, tanggal_berangkat, tanggal_pulang, tanggal_biometrik)
+    preprocessed_images = [preprocess_image(img) for img in all_images]
+    raw_output, analysis_str, summary, invalid_list_str, notice_msg, form_filling_str, token_usage = gemini_analysis(preprocessed_images, tanggal_berangkat, tanggal_pulang, tanggal_biometrik)
     rdf = random.randint(5, 10)
     time.sleep(rdf)