Update app.py

app.py

@@ -1,158 +1,28 @@
 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("✓ 에이전트 초기화 완료")
+from smolagents import InferenceClientModel, CodeAgent, MCPClient
 
-        # 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.hf.space/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="서비스 초기화에 실패했습니다.",
-        )
+    demo.launch(mcp_server=True)
+finally:
+    mcp_client.disconnect()
     
-    finally:
-        if mcp_client:
-            try:
-                mcp_client.disconnect()
-                print("✓ MCP 연결 종료")
-            except:
-                pass
-
-if __name__ == "__main__":
-    demo = main()
-    demo.launch()