Update app.py

app.py

@@ -444,26 +444,36 @@ def main():
             <meta name="viewport" content="width=device-width, initial-scale=1.0">
             <meta name="apple-mobile-web-app-capable" content="yes">
             <meta name="apple-mobile-web-app-title" content="PawMatch AI">
+        """)
+        
+        # 2. PWA 資源載入（使用相對路徑）
+        gr.HTML("""
             <link rel="manifest" href="manifest.json">
             <link rel="apple-touch-icon" href="assets/icon-192.png">
         """)
-
-        # 使用獨立的 JavaScript 區塊，避免格式化問題
+        
+        # 3. 安全的 Service Worker 註冊
         gr.HTML("""
             <script>
-                // 確保代碼在頁面載入後執行
-                document.addEventListener('DOMContentLoaded', function() {
-                    // 使用標準的 JavaScript 語法
-                    if ('serviceWorker' in navigator) {
-                        navigator.serviceWorker.register('service-worker.js')
-                        .then(function(registration) {
-                            console.log('Service Worker 註冊成功');
-                        })
-                        .catch(function(error) {
-                            console.log('Service Worker 註冊失敗:', error);
-                        });
-                    }
-                });
+            (function() {
+                // 確保在正確的網域上運行
+                var currentDomain = window.location.hostname;
+                if (currentDomain.includes('hf.space')) {
+                    // 等待頁面完全載入
+                    window.addEventListener('load', function() {
+                        if ('serviceWorker' in navigator) {
+                            // 使用相對路徑
+                            navigator.serviceWorker.register('service-worker.js')
+                            .then(function(registration) {
+                                console.log('Service Worker 註冊成功');
+                            })
+                            .catch(function(err) {
+                                console.log('Service Worker 註冊失敗:', err);
+                            });
+                        }
+                    });
+                }
+            })();
             </script>
         """)
         # Header HTML