Update app.py

app.py

@@ -102,11 +102,15 @@ def initialize_gemini_model():
     return False  # Default case
 
 # --- Helper Functions ---
-# (estimate_token_count, process_zip_file_cached, construct_analysis_prompt,
-#  call_gemini_api, display_results - remain the same as the optimized version)
-
+# Updated estimate_token_count to support integers and strings
 def estimate_token_count(text):
-    """Estimates the number of tokens based on text length."""
+    """
+    Estimates the token count.
+    If a string is provided, it calculates based on its length.
+    If an integer is provided (e.g., total character count), it uses that directly.
+    """
+    if isinstance(text, int):
+        return text // 3
     return len(text) // 3
 
 @st.cache_data(max_entries=5)