Upload app.py

app.py

@@ -45,6 +45,7 @@ def home():
         "endpoints": {
             "/": "API information",
             "/health": "Health check",
+            "/wake": "Wake up space (prevent sleep)",
             "/embed": "Generate embeddings (POST)",
             "/similarity": "Compute similarity (POST)",
             "/search": "Search documents (POST)"
@@ -64,6 +65,51 @@ def health():
         "timestamp": time.time()
     })
 
+@app.route('/wake', methods=['GET', 'POST'])
+def wake():
+    """Wake endpoint to prevent space from sleeping"""
+    try:
+        # Ensure model is loaded
+        model = get_model()
+        
+        if model is None:
+            # Try to load model if not loaded
+            logger.info("Wake endpoint triggered - loading model...")
+            model = load_model()
+            
+            if model is None:
+                return jsonify({
+                    "status": "error",
+                    "message": "Failed to load model",
+                    "awake": False,
+                    "timestamp": time.time()
+                }), 500
+        
+        # Perform a quick embedding to ensure model is active
+        try:
+            test_text = ["wake up test"]
+            _ = model.encode(test_text, batch_size=1, max_length=32)
+            logger.info("Wake endpoint: Model is active and responding")
+        except Exception as e:
+            logger.warning(f"Wake test embedding failed: {e}")
+        
+        return jsonify({
+            "status": "awake",
+            "message": "Space is awake and model is loaded",
+            "model_loaded": True,
+            "awake": True,
+            "timestamp": time.time()
+        })
+        
+    except Exception as e:
+        logger.error(f"Error in wake endpoint: {e}")
+        return jsonify({
+            "status": "error",
+            "message": str(e),
+            "awake": False,
+            "timestamp": time.time()
+        }), 500
+
 @app.route('/embed', methods=['POST'])
 def embed_text():
     """Generate embeddings for input text(s)"""