Update app.py

app.py

@@ -5,6 +5,7 @@ import gc
 import spaces
 import xml.etree.ElementTree as ET
 import re
+import os
 
 # Clear GPU memory
 torch.cuda.empty_cache()
@@ -135,6 +136,27 @@ def generate_svg(prompt):
     
     return svg_code, svg_display
 
+# Authentication function using HF Space secrets
+def authenticate(username, password):
+    """
+    Authentication function for Gradio using HF Space secrets
+    Returns True if credentials are valid, False otherwise
+    """
+    # Get credentials from HF Space secrets
+    valid_username = os.getenv("user")  # This matches your secret name "user"
+    valid_password = os.getenv("password")  # This matches your secret name "password"
+    
+    # Fallback credentials if secrets are not available (for local testing)
+    if valid_username is None:
+        valid_username = "user"
+        print("Warning: 'user' secret not found, using fallback")
+    
+    if valid_password is None:
+        valid_password = "password"
+        print("Warning: 'password' secret not found, using fallback")
+    
+    return username == valid_username and password == valid_password
+
 # Minimal CSS for slightly larger HTML preview only
 custom_css = """
 div[data-testid="HTML"] {
@@ -158,6 +180,4 @@ gradio_app = gr.Interface(
 )
 
 if __name__ == "__main__":
-    gradio_app.launch()
-
-    
+    gradio_app.launch(auth=authenticate)