Update app.py

app.py

@@ -158,37 +158,42 @@ 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 = 'Pamudu13/web-scraper'    # Expecting the space_id as a query parameter
+@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
 
-    try:
+    def generate():
         hfapi = HfApi()
 
-        # Get information about the space
-        space_info = hfapi.get_space_info(space_id, token=HF_TOKEN)
+        while True:
+            try:
+                # Get information about the space
+                space_info = hfapi.get_space_info(space_id, token=HF_TOKEN)
+                
+                # 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')
+                
+                # 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"
+
+                # Delay before checking the status again
+                time.sleep(5)  # Adjust polling interval as needed (e.g., 5 seconds)
 
-        # 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')
+            except Exception as e:
+                # Handle error, you could send a failure message or stop the stream
+                yield f"data: Error: {str(e)}\n\n"
+                break  # Stop the stream in case of an error
 
-        return jsonify({
-            'success': True,
-            'space_id': space_id,
-            'status': status,
-            'description': description,
-            'space_info': space_info
-        }), 200
+    return Response(stream_with_context(generate()), mimetype='text/event-stream')
 
-    except Exception as e:
-        return jsonify({
-            'success': False,
-            'message': f"Error: {str(e)}"
-        }), 500
 
 
 @app.route('/search_images', methods=['GET'])