Update main.py

main.py

@@ -7,7 +7,6 @@ from huggingface_hub import hf_hub_download
 from llama_cpp import Llama
 
 # --- CONFIGURATION ---
-# We use the FAST model (Llama 3.2 3B) for Telegram speed
 REPO_ID = "hugging-quants/Llama-3.2-3B-Instruct-Q4_K_M-GGUF"
 FILENAME = "llama-3.2-3b-instruct-q4_k_m.gguf"
 
@@ -29,7 +28,7 @@ def load_brain():
         print("🧠 LOADING AXIOM INTO RAM...")
         axiom_model = Llama(
             model_path=model_path,
-            n_ctx=2048, # 2k Context is fine for 3B model
+            n_ctx=2048, 
             n_threads=2,
             verbose=False
         )
@@ -53,7 +52,7 @@ async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
     # Notify user we are thinking (Typing status)
     await context.bot.send_chat_action(chat_id=update.effective_chat.id, action="typing")
 
-    # 1. Format Prompt with System Identity
+    # 1. Format Prompt
     prompt = f"""<|start_header_id|>system<|end_header_id|>
 
 You are Axiom 3.1, the Sovereign AI of Cool Shot Systems, created by Professor Heritage.
@@ -62,8 +61,7 @@ You are helpful, strategic, and concise.<|eot_id|><|start_header_id|>user<|end_h
 {user_text}<|eot_id|><|start_header_id|>assistant<|end_header_id|>
 
 """
-    # 2. Generate
-    # We run this in a thread so we don't block the bot
+    # 2. Generate in a thread
     output = await asyncio.to_thread(
         axiom_model,
         prompt,
@@ -83,12 +81,8 @@ async def startup_event():
     # 1. Load Brain
     load_brain()
     
-    # 2. Start Telegram Bot (Background Task)
-    # HARDCODED TOKEN FOR IMMEDIATE LAUNCH
-token = "8229846436:AAGa3MierT7lZV3WOdE1ebi2FaOoqVE4nLI"
-    if not token:
-        print("❌ NO TELEGRAM TOKEN FOUND!")
-        return
+    # 2. Hardcoded Token
+    token = "8229846436:AAGa3MierT7lZV3WOdE1ebi2FaOoqVE4nLI"
 
     print("🤖 STARTING TELEGRAM BOT...")
     global bot_app
@@ -102,11 +96,11 @@ token = "8229846436:AAGa3MierT7lZV3WOdE1ebi2FaOoqVE4nLI"
     await bot_app.initialize()
     await bot_app.start()
     
-    # Run polling in a separate non-blocking task
+    # Run polling in a separate task
     asyncio.create_task(bot_app.updater.start_polling())
     print("✅ TELEGRAM BOT LISTENING")
 
-# --- DUMMY SERVER (Keeps Space Alive) ---
+# --- DUMMY SERVER ---
 @app.get("/")
 def home():
     return {"status": "Axiom Telegram Bot Running"}