Update app.py

app.py

@@ -1,8 +1,9 @@
-from flask import Flask, render_template_string, jsonify
+from flask import Flask, render_template_string, jsonify, Response  # Import Response
 import requests
 import threading
 import time
 from sseclient import SSEClient
+import os  # Import the 'os' module
 
 app = Flask(__name__)
 
@@ -225,7 +226,7 @@ htmlTemplate = f"""
     <script src="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0/js/all.min.js" integrity="sha512-yFjZbTYRCJodnuyGlsKamNE/LlEaEA/3uWCGാരി7eIq7jWqVl3J8jL/kof/tfu9Xqzh/y/VM5sJd/tq5iEew==" crossorigin="anonymous" referrerpolicy="no-referrer"></script>
 
   <script>
-    const username = 'yangtb24';
+    const username = '{{username}}'; // Get username from the template
 
     async function fetchInstances() {{
       try {{
@@ -415,7 +416,8 @@ htmlTemplate = f"""
 </html>
 """
 
-USERNAME = 'yangtb24'  # Replace with your Hugging Face username
+# Get the username from environment variable (Hugging Face Space Secret)
+USERNAME = os.environ.get("USERNAME", "yangtb24")  # Default to "yangtb24" if not set
 
 
 def fetch_instances(username):
@@ -430,7 +432,8 @@ def fetch_instances(username):
 
 @app.route('/')
 def index():
-    return render_template_string(htmlTemplate)
+    # Pass the username to the template
+    return render_template_string(htmlTemplate, username=USERNAME)
 
 @app.route('/instances')
 def get_instances():
@@ -442,7 +445,7 @@ def get_instances():
 def stream_metrics(username, instance_id):
     # Use requests.get with stream=True for SSE
     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"})