Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, render_template, Response
+from flask import Flask, render_template, Response, render_template_string
 import requests
 import json
 from threading import Lock, Thread
@@ -349,29 +349,35 @@ def connect_and_listen(instance):
     url = f"https://api.hf.space/v1/{username}/{space_id}/live-metrics/sse"
     print(f"Connecting to SSE for space: {space_id}")
 
-    while True: # 持续重连循环
+    while True:  # 持续重连循环
         try:
-            response = requests.get(url, stream=True)
-            response.raise_for_status()  # 检查请求是否成功
+            # Add the Accept header for SSE
+            headers = {'Accept': 'text/event-stream'}
+            response = requests.get(url, stream=True, headers=headers)
+            response.raise_for_status()  # 检查请求是否成功, 4xx or 5xx errors
             event_sources[space_id] = response
 
-            for line in response.iter_lines():
-                if line:
-                    line = line.decode("utf-8").strip()
-                    # print(f"Received line from SSE ({space_id}): {line}") # 调试
-                    if line.startswith("event: metric"):
-                        data_line = next(response.iter_lines()).decode("utf-8").strip()
-                        if data_line.startswith("data: "):
-                            try:
-                                data = json.loads(data_line[6:])
-                                update_server_data(data, space_id, username)
-                            except json.JSONDecodeError:
-                                print(f"Error decoding JSON for {space_id}: {data_line}")
+            # Check if the response is valid before accessing .raw
+            if response and hasattr(response, 'raw'):
+                for line in response.iter_lines():
+                    if line:
+                        line = line.decode("utf-8").strip()
+                        # print(f"Received line from SSE ({space_id}): {line}") # 调试
+                        if line.startswith("event: metric"):
+                            data_line = next(response.iter_lines()).decode("utf-8").strip()
+                            if data_line.startswith("data: "):
+                                try:
+                                    data = json.loads(data_line[6:])
+                                    update_server_data(data, space_id, username)
+                                except json.JSONDecodeError:
+                                    print(f"Error decoding JSON for {space_id}: {data_line}")
+            else:
+                print(f"Invalid response from SSE for {space_id}")
 
         except requests.exceptions.RequestException as e:
             print(f"Connection error for {space_id}: {e}, retrying in 5 seconds...")
-            time.sleep(5) # 重连间隔
-        except Exception as e:
+            time.sleep(5)  # 重连间隔
+        except Exception as e:  # Catch other potential errors
             print(f"Unexpected error for {space_id}: {e}, retrying in 5 seconds...")
             time.sleep(5)
 
@@ -430,7 +436,7 @@ def home():
                         // 1. 移除不再存在的卡片
                         let currentCards = currentServers.querySelectorAll('.server-card');
                         currentCards.forEach(card => {{
-                            if (!newServers.querySelector(`#${card.id}`)) {{
+                            if (!newServers.querySelector(`#${{card.id}}`)) {{ // Corrected string interpolation
                                 card.remove();
                             }}
                         }});
@@ -438,7 +444,7 @@ def home():
                         // 2. 添加/更新卡片
                         let newCards = newServers.querySelectorAll('.server-card');
                         newCards.forEach(newCard => {{
-                            let currentCard = currentServers.querySelector(`#${newCard.id}`);
+                            let currentCard = currentServers.querySelector(`#${{newCard.id}}`);
                             if (currentCard) {{
                                 // 更新现有卡片
                                 currentCard.innerHTML = newCard.innerHTML;
@@ -485,4 +491,4 @@ if __name__ == "__main__":
     # 启动后台任务
     thread = Thread(target=background_task, daemon=True)
     thread.start()
-    app.run(debug=True, host="0.0.0.0", port=7860)
+    app.run(debug=True, host="0.0.0.0", port=7860)