Update main.py

main.py

@@ -8,7 +8,7 @@ import json
 import tiktoken
 
 
-# --- Core Helper Functions (Preserved from original script) ---
+# --- Core Helper Functions ---
 
 def is_file_type(file_path: str, file_extension: str) -> bool:
     """Check if the file has the specified file extension."""
@@ -65,7 +65,7 @@ def is_test_file(file_content: str, lang: str) -> bool:
     return False
 
 
-# --- New Token Calculation Function ---
+# --- Token Calculation Function ---
 def get_token_count(text: str) -> int:
     """Calculates the number of tokens in a string using the cl100k_base encoding."""
     try:
@@ -77,7 +77,7 @@ def get_token_count(text: str) -> int:
         return 0
 
 
-# --- Core Processing Logic (Adapted for UI Integration) ---
+# --- Core Processing Logic ---
 
 def download_and_process_repo(repo_url: str, branch_or_tag: str, log: ui.log) -> str | None:
     """
@@ -225,4 +225,4 @@ def main_page():
 
 
 # NiceGUI application on port 8080
-ui.run(host='0.0.0.0', port=8080, storage_secret='MAGIC')
+ui.run()