Update app.py

app.py

@@ -20,6 +20,9 @@ try:
 except Exception as e:
     print(f"🛑 Terjadi error saat inisialisasi: {e}")
 
+# --- KONSTANTA BATAS TOKEN OUTPUT ---
+MAX_OUTPUT_TOKENS = 1024
+
 # --- FUNGSI-FUNGSI UTAMA ---
 
 def ekstrak_teks_dari_pdf(path_file_pdf):
@@ -44,6 +47,21 @@ def generate_search_links(keywords):
     }
     return links
 
+def log_token_usage(usage_metadata):
+    """Log penggunaan token dari usage_metadata ke console."""
+    if usage_metadata is None:
+        print("⚠️  Token usage: data tidak tersedia.")
+        return
+    prompt_tokens    = getattr(usage_metadata, 'prompt_token_count',     'N/A')
+    candidate_tokens = getattr(usage_metadata, 'candidates_token_count', 'N/A')
+    total_tokens     = getattr(usage_metadata, 'total_token_count',      'N/A')
+    print("=" * 40)
+    print("📊 TOKEN USAGE")
+    print(f"   🔼 Input  (prompt)  : {prompt_tokens}")
+    print(f"   🔽 Output (response): {candidate_tokens}  [limit: {MAX_OUTPUT_TOKENS}]")
+    print(f"   ➕ Total            : {total_tokens}")
+    print("=" * 40)
+
 def analyze_career_path(cv_file):
     """Fungsi utama pipeline: Analisis CV -> Buat Laporan JSON -> Buat Link -> Gabungkan."""
     if not API_CONFIGURED:
@@ -73,8 +91,15 @@ def analyze_career_path(cv_file):
         Pastikan output hanya berupa JSON saja.
         """
         
-        generation_config = genai.types.GenerationConfig(response_mime_type="application/json")
+        # ✅ Tambahan: max_output_tokens untuk membatasi token output
+        generation_config = genai.types.GenerationConfig(
+            response_mime_type="application/json",
+            max_output_tokens=MAX_OUTPUT_TOKENS,
+        )
         response = model.generate_content(prompt_analisis_karir, generation_config=generation_config)
+
+        # ✅ Tambahan: log penggunaan token ke console
+        log_token_usage(getattr(response, 'usage_metadata', None))
         
         response_json = json.loads(response.text)
         print("✅ Laporan karir komprehensif berhasil diterima.")
@@ -83,14 +108,10 @@ def analyze_career_path(cv_file):
         keywords_from_analysis = response_json.get("jabatan_ideal", "")
         search_links = generate_search_links(keywords_from_analysis)
         
-        # ==================================================================
-        # PERUBAHAN PENTING: Menambahkan link ke dalam JSON
-        # ==================================================================
         response_json["tautan_pencarian"] = search_links
         print("✅ Tautan pencarian ditambahkan ke JSON.")
 
         print("--- Proses Selesai ---")
-        # Mengembalikan dictionary/JSON mentah
         return response_json
 
     except Exception as e:
@@ -108,7 +129,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             analyze_button = gr.Button("🔍 Analisis Karir Saya", variant="primary")
         
         with gr.Column(scale=2):
-            # Menggunakan gr.JSON untuk menampilkan dan menghasilkan data JSON sebagai output API
             output_analysis = gr.JSON(label="Output JSON dari API")
             
     analyze_button.click(