Update app.py

app.py

@@ -6,16 +6,16 @@ from huggingface_hub import InferenceClient
 app = Flask(__name__)
 CORS(app)  # Keeps your Acode connection unlocked permanently!
 
-# Pulls your hidden secret token safely
+# Pulls your hidden secret token safely from Space Settings
 HF_TOKEN = os.getenv("HF_TOKEN")
 MODEL_REPO = "Kelvin000010191/Krypton-1"
 
-# Initializes the official native Hugging Face client
-client = InferenceClient(model=MODEL_REPO, token=HF_TOKEN)
+# Initializes the client with the authentication token
+client = InferenceClient(token=HF_TOKEN)
 
 @app.route("/", methods=["GET"])
 def home():
-    return "Krypton-1 Native Proxy Node is Online."
+    return "Krypton-1 Core Node Operational."
 
 @app.route("/api/chat", methods=["POST"])
 def chat():
@@ -23,24 +23,29 @@ def chat():
         data = request.json
         user_prompt = data.get("prompt", "")
         
-        # Uses internal channel loops instead of outbound internet requests
+        # Explicitly targets the model repo and handles text generation
         response = client.text_generation(
-            user_prompt,
+            prompt=user_prompt,
+            model=MODEL_REPO,
             max_new_tokens=250,
-            temperature=0.7,
-            return_full_text=False
+            temperature=0.7
         )
         
         return jsonify({"reply": response})
         
     except Exception as e:
-        error_msg = str(e)
+        raw_error = str(e)
         
-        # Catch if your 15GB model cluster is waking up/loading weights
-        if "loading" in error_msg.lower():
-            return jsonify({"reply": "SYS_BOOT: Krypton-1 is loading weights into memory clusters. Let the cloud cook for a moment, then press RUN again."})
+        # Flags if the newly generated token isn't saved right
+        if "401" in raw_error or "unauthorized" in raw_error.lower():
+            return jsonify({"reply": "SYS_AUTH_ERR: Token unauthorized. Make sure your fresh WRITE token is saved as HF_TOKEN in Space Secrets."})
             
-        return jsonify({"reply": f"SYS_ERR: Native cloud connection broken. {error_msg}"})
+        # Flags if the massive 15GB clusters are waking up
+        if "loading" in raw_error.lower() or "503" in raw_error:
+            return jsonify({"reply": "SYS_BOOT: Krypton-1 engine is loading weights into active memory. Give the cloud 60 seconds, then try again!"})
+            
+        # Prints any other raw issue directly to your phone screen
+        return jsonify({"reply": f"SYS_ALERT: Connected to server, but model errored. Details: {raw_error}"})
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)