Update app.py

app.py

@@ -1,27 +1,158 @@
 import gradio as gr
 import os
-
 from smolagents import InferenceClientModel, CodeAgent, MCPClient
+from huggingface_hub import login, whoami
+
+def check_and_login():
+    """허깅페이스 로그인 상태 확인 및 로그인"""
+    try:
+        user_info = whoami()
+        print(f"✓ 이미 로그인됨: {user_info['name']}")
+        return True
+    except:
+        token = os.getenv("HUGGINGFACE_API_TOKEN") or os.getenv("HF_TOKEN")
+        if token:
+            try:
+                login(token=token)
+                user_info = whoami()
+                print(f"✓ 토큰으로 로그인 성공: {user_info['name']}")
+                return True
+            except Exception as e:
+                print(f"✗ 토큰 로그인 실패: {e}")
+                return False
+        else:
+            print("✗ 토큰이 없습니다.")
+            return False
+
+def safe_mcp_connection():
+    """안전한 MCP 연결 시도"""
+    mcp_configs = [
+        # 1. 올바른 transport 명시한 설정
+        {
+            "url": "https://abidlabs-mcp-tool-http.hf.space/gradio_api/mcp/sse",
+            "transport": "sse"
+        },
+        # 2. 다른 MCP 서버 시도
+        {
+            "url": "https://0208suin-mcp-client.hf.space/gradio_api/mcp/sse", 
+            "transport": "sse"
+        }
+    ]
+    
+    for i, config in enumerate(mcp_configs, 1):
+        try:
+            print(f"MCP 서버 {i} 연결 시도...")
+            mcp_client = MCPClient(config)
+            tools = mcp_client.get_tools()
+            print(f"✓ MCP 서버 {i} 연결 성공: {len(tools)}개 도구")
+            return mcp_client, tools
+        except Exception as e:
+            print(f"✗ MCP 서버 {i} 연결 실패: {str(e)[:100]}...")
+            continue
+    
+    print("모든 MCP 서버 연결 실패, 기본 도구로 실행")
+    return None, []
+
+def main():
+    mcp_client = None
+    
+    try:
+        # 1. 허깅페이스 로그인 확인
+        if not check_and_login():
+            def login_error(message, history):
+                return """🔑 로그인이 필요합니다:
+
+1. 터미널에서: `huggingface-cli login`
+2. 또는 스페이스 Settings에서 HUGGINGFACE_API_TOKEN 설정
+3. 토큰 발급: https://huggingface.co/settings/tokens"""
+            
+            return gr.ChatInterface(
+                fn=login_error,
+                type="messages",
+                title="🔑 로그인 필요",
+                description="Hugging Face 로그인이 필요합니다.",
+            )
+        
+        # 2. MCP 연결 시도
+        mcp_client, tools = safe_mcp_connection()
+        
+        # 3. 모델 초기화
+        try:
+            model = InferenceClientModel()
+            print("✓ 모델 초기화 완료")
+        except Exception as e:
+            print(f"✗ 모델 초기화 실패: {e}")
+            # 대안: HfApiModel 사용
+            from smolagents import HfApiModel
+            token = os.getenv("HUGGINGFACE_API_TOKEN") or os.getenv("HF_TOKEN")
+            model = HfApiModel(model_id="microsoft/DialoGPT-medium", token=token)
+            print("✓ 대안 모델(HfApiModel) 초기화 완료")
+        
+        # 4. 에이전트 초기화
+        agent = CodeAgent(
+            tools=tools, 
+            model=model, 
+            additional_authorized_imports=["json", "ast", "urllib", "base64"]
+        )
+        print("✓ 에이전트 초기화 완료")
+
+        # 5. 채팅 함수
+        def agent_chat(message, history):
+            try:
+                result = agent.run(message)
+                return str(result)
+            except Exception as e:
+                error_msg = str(e)
+                if "nebius" in error_msg.lower():
+                    return "❌ 모델 접근 오류입니다. 허깅페이스 로그인을 확인해주세요."
+                elif "mcp" in error_msg.lower():
+                    return f"⚠️ MCP 도구 오류: {error_msg[:200]}..."
+                else:
+                    return f"❌ 오류가 발생했습니다: {error_msg[:200]}..."
 
+        # 6. 인터페이스 생성
+        demo = gr.ChatInterface(
+            fn=agent_chat,
+            type="messages",
+            examples=[
+                "안녕하세요! 어떻게 도와드릴까요?",
+                "Analyze the sentiment of 'This is awesome!'",
+                "사용 가능한 도구가 무엇인가요?"
+            ],
+            title="🤖 Agent with MCP Tools",
+            description=f"MCP 도구를 사용하는 스마트 에이전트 (도구 {len(tools)}개)",
+        )
 
+        return demo
 
-try:
-    mcp_client = MCPClient(
-        {"url": "https://0208suin/mcp-client/gradio_api/mcp/sse"}
-    )
-    tools = mcp_client.get_tools()
+    except Exception as e:
+        print(f"초기화 오류: {e}")
+        
+        def error_message(message, history):
+            return f"""❌ 서비스 초기화 실패: 
 
-    model = InferenceClientModel(token=os.getenv("HUGGINGFACE_API_TOKEN"))
-    agent = CodeAgent(tools=[*tools], model=model, additional_authorized_imports=["json", "ast", "urllib", "base64"])
+{str(e)[:300]}...
 
-    demo = gr.ChatInterface(
-        fn=lambda message, history: str(agent.run(message)),
-        type="messages",
-        examples=["Analyze the sentiment of the following text 'This is awesome'"],
-        title="Agent with MCP Tools",
-        description="This is a simple agent that uses MCP tools to answer questions.",
-    )
+해결 방법:
+1. 스페이스 재시작
+2. 환경 변수 확인 
+3. 허깅페이스 토큰 확인"""
+        
+        return gr.ChatInterface(
+            fn=error_message,
+            type="messages",
+            title="❌ 초기화 오류",
+            description="서비스 초기화에 실패했습니다.",
+        )
+    
+    finally:
+        if mcp_client:
+            try:
+                mcp_client.disconnect()
+                print("✓ MCP 연결 종료")
+            except:
+                pass
 
-    demo.launch()
-finally:
-    mcp_client.disconnect()
+if __name__ == "__main__":
+    demo = main()
+    demo.launch()