Update app.py

Add total token counter

app.py

@@ -145,6 +145,8 @@ def gemini_analysis(images, tanggal_berangkat, tanggal_pulang, api_key):
               response_mime_type="application/json"
             )
       )
+    total_tokens = client.models.count_tokens(
+    model="gemini-2.0-flash-lite", contents=[prompt_with_date] + images)
     raw_output = response.text
     # ✅ Inisialisasi variabel default
     analysis = {}
@@ -170,7 +172,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
+    return raw_output, analysis_str, summary, invalid_list_str, notice_msg, form_filling_str, total_tokens
 
 def process_and_zip_all_images(images, api_key, zip_name="All_PDF_Docs.zip"):
     # Inisialisasi Gemini client
@@ -251,7 +253,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 = gemini_analysis(all_images, tanggal_berangkat, tanggal_pulang, api_key)
+    raw_output, analysis_str, summary, invalid_list_str, notice_msg, form_filling_str, total_tokens = gemini_analysis(all_images, tanggal_berangkat, tanggal_pulang, api_key)
     rdf = random.randint(5, 10)
     time.sleep(rdf)
 
@@ -275,7 +277,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
+    return temp_docx_path, form_filling_str, zip_file_path, invalid_list_str, raw_output, summary, notice_msg, total_tokens
 
 
 # Gradio UI update: add ".zip" to accepted file types
@@ -330,11 +332,12 @@ with gr.Blocks() as demo:
     notice_msg = gr.Textbox(label="📝 NOTIFICATION MSG", lines=10)
     gr.Markdown("## 📝 RAW OUTPUT FROM AI")
     raw_output = gr.Textbox(label="📝 RAW OUTPUT", lines=20)
+    token_counter = gr.Textbox(label="Token Usage Count", 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]
+        outputs=[download_output_docx, form_filling_output, download_valid_zip, invalid_list_output, raw_output, summary_output, notice_msg, token_counter]
     )
 
 demo.launch(debug=True)