Update app.py

app.py

@@ -144,10 +144,7 @@ def gemini_analysis(images, tanggal_berangkat, tanggal_pulang, api_key):
               response_mime_type="application/json"
             )
       )
-    token_counter = client.models.count_tokens(
-    model="gemini-2.0-flash-lite", contents=[prompt_with_date] + images)
-    total_tokens = token_counter.total_tokens
-    total_tokens_in_idr = total_tokens*5/1000
+    token_usage = response.usage_metadata
     raw_output = response.text
     # ✅ Inisialisasi variabel default
     analysis = {}
@@ -172,7 +169,7 @@ Halo Kak selamat malam 😊
         form_filling_str = json.dumps(form_filling, indent=2, ensure_ascii=False)
     except Exception as e:
         print(f"Error parsing JSON: {e}")
-    return raw_output, analysis_str, summary, invalid_list_str, notice_msg, form_filling_str, total_tokens_in_idr
+    return raw_output, analysis_str, summary, invalid_list_str, notice_msg, form_filling_str, token_usage
 
 def process_and_zip_all_images(images, api_key, zip_name="All_PDF_Docs.zip"):
     # Inisialisasi Gemini client
@@ -253,7 +250,7 @@ def main_process(files, tanggal_berangkat, tanggal_pulang, api_key, dpi):
             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, total_tokens_in_idr = gemini_analysis(all_images, tanggal_berangkat, tanggal_pulang, api_key)
+    raw_output, analysis_str, summary, invalid_list_str, notice_msg, form_filling_str, token_usage = gemini_analysis(all_images, tanggal_berangkat, tanggal_pulang, api_key)
     rdf = random.randint(5, 10)
     time.sleep(rdf)
 
@@ -277,7 +274,7 @@ def main_process(files, tanggal_berangkat, tanggal_pulang, api_key, dpi):
 
     # Filtering the file
     zip_file_path = process_and_zip_all_images(all_images, api_key, zip_name=f'All_PDF_Docs_{base_name}.zip')
-    return temp_docx_path, form_filling_str, zip_file_path, invalid_list_str, raw_output, summary, notice_msg, total_tokens_in_idr
+    return temp_docx_path, form_filling_str, zip_file_path, invalid_list_str, raw_output, summary, notice_msg, token_usage
 
 
 # Gradio UI update: add ".zip" to accepted file types
@@ -333,12 +330,12 @@ with gr.Blocks() as demo:
     gr.Markdown("## 📝 RAW OUTPUT FROM AI")
     raw_output = gr.Textbox(label="📝 RAW OUTPUT", lines=20)
     gr.Markdown("Token cost in IDR")
-    total_tokens_in_idr = gr.Textbox(label="Token cost in IDR", lines=5)
+    token_usage = gr.Textbox(label="Token cost in IDR", lines=5)
 
     run_btn.click(
         fn=main_process,
         inputs=[file_input, tanggal_berangkat, tanggal_pulang, api_key, dpi_slider],
-        outputs=[download_output_docx, form_filling_output, download_valid_zip, invalid_list_output, raw_output, summary_output, notice_msg, total_tokens_in_idr]
+        outputs=[download_output_docx, form_filling_output, download_valid_zip, invalid_list_output, raw_output, summary_output, notice_msg, token_usage]
     )
 
 demo.launch(debug=True)