Update app.py

Add token cost in IDR

app.py

@@ -147,6 +147,7 @@ def gemini_analysis(images, tanggal_berangkat, tanggal_pulang, api_key):
     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
     raw_output = response.text
     # ✅ Inisialisasi variabel default
     analysis = {}
@@ -171,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, total_tokens
+    return raw_output, analysis_str, summary, invalid_list_str, notice_msg, form_filling_str, total_tokens_in_idr
 
 def process_and_zip_all_images(images, api_key, zip_name="All_PDF_Docs.zip"):
     # Inisialisasi Gemini client
@@ -252,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, total_tokens = 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_in_idr = gemini_analysis(all_images, tanggal_berangkat, tanggal_pulang, api_key)
     rdf = random.randint(5, 10)
     time.sleep(rdf)
 
@@ -276,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, total_tokens
+    return temp_docx_path, form_filling_str, zip_file_path, invalid_list_str, raw_output, summary, notice_msg, total_tokens_in_idr
 
 
 # Gradio UI update: add ".zip" to accepted file types
@@ -331,12 +332,13 @@ 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)
+    gr.Markdown("Token cost in IDR")
+    total_tokens_in_idr = 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, token_counter]
+        outputs=[download_output_docx, form_filling_output, download_valid_zip, invalid_list_output, raw_output, summary_output, notice_msg, total_tokens_in_idr]
     )
 
 demo.launch(debug=True)