Update app.py

app.py

@@ -158,6 +158,39 @@ def api_restart_space():
             'message': f"Error: {str(e)}"
         }), 500    
 
+@app.route('/get_space_status', methods=['GET'])
+def get_space_status():
+    """API route to get the status of a Hugging Face Space."""
+    space_id = request.args.get('space_id')  # Expecting the space_id as a query parameter
+
+    if not space_id:
+        return jsonify({'error': 'space_id parameter is required'}), 400
+
+    try:
+        hfapi = HfApi()
+
+        # Get information about the space
+        space_info = hfapi.get_space_info(space_id, token=HF_TOKEN)
+
+        # Extract relevant details from the space_info (you can modify this based on your needs)
+        status = space_info.get('status', 'Unknown')  # Assuming 'status' field is present
+        description = space_info.get('description', 'No description available')
+
+        return jsonify({
+            'success': True,
+            'space_id': space_id,
+            'status': status,
+            'description': description,
+            'space_info': space_info
+        }), 200
+
+    except Exception as e:
+        return jsonify({
+            'success': False,
+            'message': f"Error: {str(e)}"
+        }), 500
+
+
 @app.route('/search_images', methods=['GET'])
 def api_search_images():
     try: