Update app.py

app.py

@@ -474,5 +474,65 @@ htmlTemplate = f"""
 </html>
 """
 
+USERNAME = os.environ.get("USERNAME", "yangtb24")  # Default username
+
+def fetch_instances(username):
+    try:
+        response = requests.get(f"https://huggingface.co/api/spaces?author={username}")
+        response.raise_for_status()
+        user_instances = response.json()
+        return [{"id": instance["id"].split('/')[1], "owner": username} for instance in user_instances]
+    except requests.exceptions.RequestException as e:
+        print(f"Error fetching instances: {e}")
+        return []
+
+@app.route('/')
+def index():
+    return render_template_string(htmlTemplate, username=USERNAME)
+
+@app.route('/instances')
+def get_instances():
+    instances = fetch_instances(USERNAME)
+    return jsonify(instances)
+
+@app.route('/metrics/<username>/<instance_id>')
+def stream_metrics(username, instance_id):
+    url = f"https://api.hf.space/v1/{username}/{instance_id}/live-metrics/sse"
+
+    def generate():
+        try:
+            response = requests.get(url, stream=True, headers={"Accept": "text/event-stream"}, timeout=15)
+            response.raise_for_status()
+
+            buffer = ""
+            for chunk in response.iter_content(chunk_size=1024, decode_unicode=True):
+                if chunk:
+                    buffer += chunk
+                    while "\n\n" in buffer:
+                        event_data, buffer = buffer.split("\n\n", 1)
+                        lines = event_data.split("\n")
+                        event_type = "message"
+                        data_lines = []
+                        for line in lines:
+                            if line.startswith("event:"):
+                                event_type = line.split(":", 1)[1].strip()
+                            elif line.startswith("data:"):
+                                data_lines.append(line.split(":", 1)[1].strip())
+
+                        if event_type == "metric":
+                            # Correctly yield the JSON data:
+                            yield f"event: {event_type}\ndata: {json.dumps(json.loads(''.join(data_lines)))}\n\n"
+
+
+        except requests.exceptions.RequestException as e:
+            print(f"Request Exception: {e}")
+            yield f"event: error\ndata: Connection error: {e}\\n\\n"
+        except Exception as e:
+            print(f"An error occurred: {e}")
+            yield f"event: error\ndata: An error occurred: {e}\\n\\n"
+
+    return Response(generate(), mimetype='text/event-stream')
+
 if __name__ == '__main__':
     app.run(debug=True, host='0.0.0.0', port=7860)
+