Update app.py

app.py

@@ -721,18 +721,25 @@ with gr.Blocks(title="AI PROMPT ENHANCER") as demo:
     gr.HTML(f"""
     <style>{CUSTOM_CSS}</style>
     <script>
-        // HuggingFace Spaces URL이면 커스텀 도메인으로 리다이렉트
-        (function() {{
-            const currentHost = window.location.hostname;
-            const customDomain = '1street.ai';
-            
-            // huggingface.co 도메인이면 커스텀 도메인으로 리다이렉트
-            if (currentHost.includes('huggingface.co') || currentHost.includes('hf.space')) {{
-                const newUrl = 'https://' + customDomain + window.location.pathname + window.location.search;
-                window.location.replace(newUrl);
+        // 페이지 로드 즉시 실행
+        const currentHost = window.location.hostname;
+        const customDomain = '1street.ai';
+        
+        // huggingface.co 또는 hf.space 도메인이면 커스텀 도메인으로 리다이렉트
+        if (currentHost.includes('huggingface.co') || currentHost.includes('hf.space')) {{
+            window.location.href = 'https://' + customDomain;
+        }}
+        
+        // DOM 로드 후에도 체크
+        document.addEventListener('DOMContentLoaded', function() {{
+            if (window.location.hostname.includes('huggingface.co') || window.location.hostname.includes('hf.space')) {{
+                window.location.href = 'https://' + customDomain;
             }}
-        }})();
+        }});
     </script>
+    <noscript>
+        <meta http-equiv="refresh" content="0;url=https://1street.ai">
+    </noscript>
     """)
     
     # 헤더 영역
@@ -745,7 +752,16 @@ with gr.Blocks(title="AI PROMPT ENHANCER") as demo:
                 </div>
             ''')
         with gr.Column(scale=1, min_width=200):
-            login_btn = gr.LoginButton(value="🤗 로그인", size="sm")
+            gr.HTML('''
+                <a href="/login/huggingface?_target_url=https://1street.ai" 
+                   style="display: inline-flex; align-items: center; gap: 8px;
+                          background: linear-gradient(135deg, #ff9d00 0%, #ffb347 100%);
+                          color: #000; font-weight: 600; font-size: 0.9rem;
+                          padding: 10px 20px; border-radius: 25px; text-decoration: none;
+                          box-shadow: 0 4px 12px rgba(255, 157, 0, 0.3);">
+                    🤗 로그인
+                </a>
+            ''')
     
     # 유저 정보 표시
     user_status = gr.Markdown("")