Update app.py

app.py

@@ -161,34 +161,27 @@ def api_restart_space():
 @app.route('/get_live_space_status', methods=['GET'])
 def get_live_space_status():
     """API route to stream live status of a Hugging Face Space."""
-    space_id = 'Pamudu13/web-scraper' 
-
-    if not space_id:
-        return jsonify({'error': 'space_id parameter is required'}), 400
+    space_id = request.args.get('space_id', 'Pamudu13/web-scraper')  # Default to 'Pamudu13/web-scraper' if not provided
 
     def generate():
-        hfapi = HfApi()
-
         while True:
             try:
-                # Get information about the space
-                space_info = hfapi.get_space_info(space_id, token=HF_TOKEN)
+                # Fetch the current runtime status of the Space
+                space_runtime = hf_api.get_space_runtime(repo_id=space_id)
                 
-                # Extract the status of the space (adjust according to the response structure)
-                status = space_info.get('status', 'Unknown')
-                description = space_info.get('description', 'No description available')
+                # Extract relevant details
+                status = space_runtime.stage  # e.g., 'BUILDING', 'RUNNING', etc.
+                hardware = space_runtime.hardware  # e.g., 'cpu-basic', 't4-medium', etc.
                 
                 # Send the status as a Server-Sent Event
                 yield f"data: {status}\n\n"
-
-                # Optionally send more detailed info
-                yield f"data: {description}\n\n"
+                yield f"data: {hardware}\n\n"
 
                 # Delay before checking the status again
-                time.sleep(5)  # Adjust polling interval as needed (e.g., 5 seconds)
+                time.sleep(5)  # Adjust polling interval as needed
 
             except Exception as e:
-                # Handle error, you could send a failure message or stop the stream
+                # Handle errors and send an error message
                 yield f"data: Error: {str(e)}\n\n"
                 break  # Stop the stream in case of an error
 
@@ -196,6 +189,7 @@ def get_live_space_status():
 
 
 
+
 @app.route('/search_images', methods=['GET'])
 def api_search_images():
     try: