Update app.py

app.py

@@ -122,6 +122,38 @@ def search_images(query, num_images=5):
         logger.error(f"An error occurred: {str(e)}")
         return []
 
+
+
+HF_TOKEN = os.getenv("HF_TOKEN")  # Make sure you set the HF_TOKEN in your environment
+
+def restart_space(space_id):
+    """Restart a Hugging Face Space using the API."""
+    api_url = f"https://api.huggingface.co/spaces/{space_id}/restart"  # Corrected API endpoint for restarting Spaces
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+    try:
+        response = requests.post(api_url, headers=headers)
+        if response.status_code == 200:
+            logger.info(f"Successfully restarted Space: {space_id}")
+            return jsonify({"success": True, "message": f"Successfully restarted Space: {space_id}"}), 200
+        else:
+            logger.error(f"Failed to restart Space: {space_id} - {response.status_code} - {response.text}")
+            return jsonify({"success": False, "message": f"Failed to restart Space: {space_id} - {response.status_code} - {response.text}"}), 500
+    except Exception as e:
+        logger.error(f"Error restarting space: {str(e)}")
+        return jsonify({"success": False, "message": f"Error: {str(e)}"}), 500
+
+@app.route('/restart_space', methods=['POST'])
+def api_restart_space():
+    """API route to restart a Hugging Face Space."""
+    space_id = 'Pamudu13/web-scraper'  
+    
+    # Expecting the space_id in the JSON body
+
+    if not space_id:
+        return jsonify({'error': 'space_id parameter is required'}), 400
+
+    return restart_space(space_id)
+
 @app.route('/search_images', methods=['GET'])
 def api_search_images():
     try: