Update app.py

app.py

@@ -1,205 +1,93 @@
 import os
 import gradio as gr
-from authlib.integrations.starlette_client import OAuth, OAuthError
-from starlette.config import Config
-from starlette.middleware.sessions import SessionMiddleware
-import secrets
-import json
 
-# 환경 변수에서 Google OAuth 정보 가져오기
+# 환경 변수에서 설정 가져오기
 GOOGLE_CLIENT_ID = os.environ.get("GOOGLE_CLIENT_ID", "your-client-id")
 GOOGLE_CLIENT_SECRET = os.environ.get("GOOGLE_CLIENT_SECRET", "your-secret")
 
-# Hugging Face Space URL (환경변수로 설정 가능)
-SPACE_HOST = os.environ.get("SPACE_HOST", "http://localhost:7860")
-
-# 세션을 위한 비밀 키
-SECRET_KEY = os.environ.get("SECRET_KEY", secrets.token_urlsafe(32))
-
-# OAuth 설정
-config = Config(environ={
-    'GOOGLE_CLIENT_ID': GOOGLE_CLIENT_ID,
-    'GOOGLE_CLIENT_SECRET': GOOGLE_CLIENT_SECRET
-})
-
-oauth = OAuth(config)
-oauth.register(
-    name='google',
-    server_metadata_url='https://accounts.google.com/.well-known/openid-configuration',
-    client_kwargs={'scope': 'openid email profile'}
-)
-
-# 전역 변수로 사용자 정보 저장 (실제 프로덕션에서는 세션 사용 권장)
-current_user = None
-
-def get_login_url():
-    """Google 로그인 URL 생성"""
-    redirect_uri = f"{SPACE_HOST}/auth/callback"
-    return oauth.google.authorize_redirect(redirect_uri).headers['Location']
-
-def handle_user_info(info):
-    """사용자 정보 처리"""
-    global current_user
-    current_user = info
-    return info
-
-def logout():
-    """로그아웃 처리"""
-    global current_user
-    current_user = None
-    return None, gr.update(visible=True), gr.update(visible=False)
-
-def create_oauth_app():
-    with gr.Blocks(title="Google OAuth Gradio 앱") as demo:
-        gr.Markdown("# Google OAuth 통합 Gradio 앱")
-        
-        # 상태 관리를 위한 State
-        user_info_state = gr.State(value=None)
-        
-        # 로그인 전 화면
-        with gr.Column(visible=True) as login_section:
-            gr.Markdown("### 환경 설정 상태")
-            gr.Markdown(f"✅ Client ID 설정됨" if GOOGLE_CLIENT_ID != "your-client-id" else "❌ Client ID 미설정")
-            gr.Markdown(f"✅ Client Secret 설정됨" if GOOGLE_CLIENT_SECRET != "your-secret" else "❌ Client Secret 미설정")
-            
-            gr.Markdown("---")
-            gr.Markdown("### Google 로그인이 필요합니다")
-            
-            if GOOGLE_CLIENT_ID != "your-client-id" and GOOGLE_CLIENT_SECRET != "your-secret":
-                login_btn = gr.Button("🔐 Google로 로그인", variant="primary", size="lg")
-                gr.Markdown("""
-                **참고:** Hugging Face Spaces에서는 OAuth 리다이렉트가 제한적일 수 있습니다.
-                
-                OAuth 설정 방법:
-                1. [Google Cloud Console](https://console.cloud.google.com/)에서 OAuth 2.0 클라이언트 생성
-                2. 승인된 리다이렉트 URI에 추가:
-                   - `https://[space-name].hf.space/auth/callback`
-                   - `http://localhost:7860/auth/callback` (로컬 테스트용)
-                3. Hugging Face Space Settings에서 환경변수 설정
-                """)
-            else:
-                gr.Markdown("""
-                ⚠️ **OAuth 설정이 필요합니다**
-                
-                1. 환경변수 `GOOGLE_CLIENT_ID`와 `GOOGLE_CLIENT_SECRET` 설정
-                2. Hugging Face Space Settings > Variables and secrets에서 추가
-                """)
-        
-        # 로그인 후 화면
-        with gr.Column(visible=False) as user_section:
-            gr.Markdown("### 👤 로그인된 사용자 정보")
-            
-            with gr.Row():
-                user_name = gr.Textbox(label="이름", interactive=False)
-                user_email = gr.Textbox(label="이메일", interactive=False)
-            
-            user_picture = gr.Image(label="프로필 사진", type="filepath", interactive=False, height=100)
-            
-            logout_btn = gr.Button("로그아웃", variant="secondary")
-            
-            gr.Markdown("---")
-            gr.Markdown("### 🎉 환영합니다! 이제 앱을 사용할 수 있습니다.")
-            
-            # 로그인 후 사용 가능한 기능들
-            with gr.Row():
-                name_input = gr.Textbox(label="인사할 이름", placeholder="친구 이름")
-                greeting_output = gr.Textbox(label="개인화된 인사말")
-            
-            greet_btn = gr.Button("인사하기")
-            
-            def personalized_greet(name, user_info):
-                if user_info and name:
-                    user_name = user_info.get('name', '사용자')
-                    return f"안녕하세요 {name}님! {user_name}님이 인사드립니다! 😊"
-                return "이름을 입력해주세요."
-            
-            greet_btn.click(
-                fn=personalized_greet,
-                inputs=[name_input, user_info_state],
-                outputs=greeting_output
-            )
+# Gradio의 기본 인증 사용 (OAuth 대신)
+USERS = os.environ.get("USERS", "user1:pass1,user2:pass2").split(",")
+auth_dict = {}
+for user in USERS:
+    if ":" in user:
+        username, password = user.split(":", 1)
+        auth_dict[username] = password
+
+def create_app():
+    with gr.Blocks(title="Google 로그인 스타일 앱") as demo:
+        gr.Markdown("# 🔐 로그인이 필요한 Gradio 앱")
         
-        # 임시 로그인 시뮬레이션 (실제로는 OAuth 콜백 처리 필요)
-        def simulate_login():
-            # 실제로는 OAuth 콜백에서 받은 정보를 사용
-            fake_user_info = {
-                'name': '테스트 사용자',
-                'email': 'test@example.com',
-                'picture': 'https://via.placeholder.com/100'
-            }
-            return (
-                fake_user_info,  # user_info_state
-                gr.update(visible=False),  # login_section
-                gr.update(visible=True),   # user_section
-                fake_user_info.get('name', ''),  # user_name
-                fake_user_info.get('email', ''), # user_email
-                fake_user_info.get('picture', '') # user_picture
-            )
+        # 환경 설정 상태
+        gr.Markdown("### 환경 설정 상태")
+        gr.Markdown(f"✅ Client ID 설정됨" if GOOGLE_CLIENT_ID != "your-client-id" else "❌ Client ID 미설정")
+        gr.Markdown(f"✅ Client Secret 설정됨" if GOOGLE_CLIENT_SECRET != "your-secret" else "❌ Client Secret 미설정")
         
-        def handle_logout():
-            return (
-                None,  # user_info_state
-                gr.update(visible=True),   # login_section
-                gr.update(visible=False),  # user_section
-                '',  # user_name
-                '',  # user_email
-                None # user_picture
-            )
+        gr.Markdown("---")
         
-        if GOOGLE_CLIENT_ID != "your-client-id" and GOOGLE_CLIENT_SECRET != "your-secret":
-            login_btn.click(
-                fn=simulate_login,
-                inputs=[],
-                outputs=[user_info_state, login_section, user_section, user_name, user_email, user_picture]
+        # 메인 기능
+        gr.Markdown("### 👋 환영합니다!")
+        gr.Markdown("이 앱은 로그인된 사용자만 사용할 수 있습니다.")
+        
+        with gr.Row():
+            name_input = gr.Textbox(
+                label="이름을 입력하세요", 
+                placeholder="홍길동",
+                info="인사받을 이름을 입력하세요"
             )
+            output = gr.Textbox(label="인사말")
+        
+        greet_btn = gr.Button("인사하기", variant="primary")
+        
+        def greet(name):
+            if name:
+                return f"안녕하세요, {name}님! 로그인해주셔서 감사합니다! 😊"
+            return "이름을 입력해주세요."
         
-        logout_btn.click(
-            fn=handle_logout,
-            inputs=[],
-            outputs=[user_info_state, login_section, user_section, user_name, user_email, user_picture]
-        )
+        greet_btn.click(fn=greet, inputs=name_input, outputs=output)
+        
+        gr.Markdown("---")
+        
+        # Google OAuth 안내
+        gr.Markdown("""
+        ### 📝 Google OAuth 통합 안내
+        
+        Hugging Face Spaces는 서버사이드 OAuth를 완전히 지원하지 않습니다.
+        
+        **대안:**
+        1. **Gradio 기본 인증 사용** (현재 적용됨)
+           - Space Settings에서 `USERS` 환경변수 설정
+           - 예: `USERS=admin:password123,user:pass456`
+        
+        2. **Google OAuth 버튼 (외부 링크)**
+           - 아래 버튼을 클릭하면 Google 로그인 페이지로 이동합니다
+        """)
+        
+        # Google OAuth URL 생성 (클라이언트 사이드)
+        if GOOGLE_CLIENT_ID != "your-client-id":
+            oauth_url = f"https://accounts.google.com/o/oauth2/v2/auth?client_id={GOOGLE_CLIENT_ID}&redirect_uri=https://your-space.hf.space&response_type=token&scope=email%20profile"
+            gr.Markdown(f"[🔗 Google로 로그인]({oauth_url})")
         
-        # OAuth 콜백 처리를 위한 숨겨진 컴포넌트
         gr.Markdown("""
-        <script>
-        // OAuth 콜백 처리를 위한 JavaScript
-        if (window.location.search.includes('code=')) {
-            console.log('OAuth callback detected');
-            // 실제 구현시 여기서 토큰 교환 및 사용자 정보 가져오기
-        }
-        </script>
-        """, visible=False)
+        ### 🔧 설정 방법
+        
+        1. **Hugging Face Space Settings > Variables**에서:
+           - `USERS`: 사용자명:비밀번호 형식으로 설정
+           - 예: `admin:mypassword,john:doe123`
+        
+        2. **앱 재시작 후 로그인 창이 표시됩니다**
+        """)
     
     return demo
 
-# Gradio 앱 ��성
-demo = create_oauth_app()
-
-# 미들웨어 추가 (세션 관리)
-demo.app.add_middleware(SessionMiddleware, secret_key=SECRET_KEY)
-
-# OAuth 라우트 추가 (Gradio 앱에 커스텀 라우트 추가)
-@demo.app.get("/auth/google")
-async def auth_google(request):
-    redirect_uri = f"{SPACE_HOST}/auth/callback"
-    return await oauth.google.authorize_redirect(request, redirect_uri)
-
-@demo.app.get("/auth/callback")
-async def auth_callback(request):
-    try:
-        token = await oauth.google.authorize_access_token(request)
-        user = token.get('userinfo')
-        if user:
-            request.session['user'] = user
-        return {"status": "success", "user": user}
-    except OAuthError as e:
-        return {"status": "error", "message": str(e)}
-
-# requirements.txt에 추가 필요:
-# gradio
-# authlib
-# httpx
-# starlette
+# 인증이 있는 앱 생성
+if auth_dict:
+    # 사용자 인증이 설정된 경우
+    demo = create_app()
+    demo.auth = list(auth_dict.items())
+    demo.auth_message = "Google 계정 대신 설정된 사용자명/비밀번호를 사용하세요"
+else:
+    # 인증이 설정되지 않은 경우
+    demo = create_app()
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.launch(server_name="0.0.0.0", server_port=786