Update app.py

app.py

@@ -1,6 +1,489 @@
 import gradio as gr
+from huggingface_hub import InferenceClient
+from openai import OpenAI
+import os
 
-gr.load(
-   "models/MiniMaxAI/MiniMax-M2.1",
-   provider="novita",
-).launch()
+# ============================================
+# MiniMax-M2.1 Streaming Chat
+# Dual Provider: Novita (HF) + MiniMax Official API
+# Comic Classic Theme
+# ============================================
+
+# Provider 상태 관리
+class ProviderManager:
+    def __init__(self):
+        self.current_provider = "novita"  # 기본값: Novita
+        self.novita_available = True
+        
+        # Novita Client (via HuggingFace)
+        self.novita_client = InferenceClient(
+            provider="novita",
+            api_key=os.environ.get("HF_TOKEN"),
+        )
+        
+        # MiniMax Official API Client
+        self.minimax_client = OpenAI(
+            api_key=os.environ.get("MINIMAX_API_KEY", ""),
+            base_url="https://api.minimax.chat/v1"
+        )
+    
+    def get_status(self):
+        if self.current_provider == "novita":
+            return "🟢 Novita (HuggingFace)"
+        else:
+            return "🟡 MiniMax Official API"
+    
+    def switch_to_minimax(self):
+        self.current_provider = "minimax"
+        self.novita_available = False
+        print("⚠️ Switched to MiniMax Official API")
+    
+    def reset_to_novita(self):
+        self.current_provider = "novita"
+        self.novita_available = True
+        print("✅ Reset to Novita provider")
+
+# Global provider manager
+provider = ProviderManager()
+
+def chat_with_novita(messages):
+    """Novita provider를 통한 스트리밍"""
+    stream = provider.novita_client.chat.completions.create(
+        model="MiniMaxAI/MiniMax-M2.1",
+        messages=messages,
+        max_tokens=4096,
+        temperature=1.0,
+        top_p=0.95,
+        stream=True
+    )
+    
+    for chunk in stream:
+        if chunk.choices[0].delta.content:
+            yield chunk.choices[0].delta.content
+
+def chat_with_minimax(messages):
+    """MiniMax 공식 API를 통한 스트리밍"""
+    stream = provider.minimax_client.chat.completions.create(
+        model="MiniMax-M2.1",
+        messages=messages,
+        max_tokens=4096,
+        temperature=1.0,
+        top_p=0.95,
+        stream=True
+    )
+    
+    for chunk in stream:
+        if chunk.choices and chunk.choices[0].delta.content:
+            yield chunk.choices[0].delta.content
+
+def chat_stream(message, history):
+    """
+    Streaming chat with automatic fallback
+    1차: Novita (HuggingFace) 
+    2차: MiniMax Official API (fallback)
+    """
+    if not message.strip():
+        yield "", provider.get_status()
+        return
+    
+    # Build messages
+    messages = [{
+        "role": "system",
+        "content": "You are MiniMax-M2.1, a helpful AI assistant built by MiniMax. You excel at coding, tool use, and complex reasoning tasks. Respond in the same language as the user."
+    }]
+    
+    for user_msg, assistant_msg in history:
+        if user_msg:
+            messages.append({"role": "user", "content": user_msg})
+        if assistant_msg:
+            messages.append({"role": "assistant", "content": assistant_msg})
+    
+    messages.append({"role": "user", "content": message})
+    
+    response_text = ""
+    
+    # 1차 시도: Novita (if available)
+    if provider.novita_available and provider.current_provider == "novita":
+        try:
+            for chunk in chat_with_novita(messages):
+                response_text += chunk
+                yield response_text, "🟢 Novita (HuggingFace)"
+            return
+        except Exception as e:
+            error_msg = str(e).lower()
+            # 크레딧 소진 또는 rate limit 관련 에러 감지
+            if any(keyword in error_msg for keyword in [
+                "rate limit", "quota", "exceeded", "insufficient", 
+                "credit", "balance", "limit", "429", "402", "payment"
+            ]):
+                print(f"⚠️ Novita error (switching to MiniMax): {e}")
+                provider.switch_to_minimax()
+            else:
+                # 다른 에러는 일시적일 수 있으므로 한 번만 시도
+                print(f"⚠️ Novita temporary error: {e}")
+    
+    # 2차 시도: MiniMax Official API
+    try:
+        # MiniMax API 키 확인
+        if not os.environ.get("MINIMAX_API_KEY"):
+            yield "❌ Error: MINIMAX_API_KEY not configured. Please add it to Secrets.", "🔴 No Provider"
+            return
+        
+        for chunk in chat_with_minimax(messages):
+            response_text += chunk
+            yield response_text, "🟡 MiniMax Official API"
+        return
+        
+    except Exception as e:
+        error_msg = str(e)
+        
+        # MiniMax도 실패하면 Novita 다시 시도 가능하도록
+        if "rate limit" not in error_msg.lower():
+            provider.reset_to_novita()
+        
+        yield f"❌ Error: {error_msg}", "🔴 Error"
+
+def reset_provider():
+    """수동으로 Novita로 리셋"""
+    provider.reset_to_novita()
+    return "✅ Provider reset to Novita. Next message will try Novita first."
+
+# ============================================
+# 🎨 Comic Classic Theme CSS
+# ============================================
+css = """
+/* ===== 🎨 Google Fonts Import ===== */
+@import url('https://fonts.googleapis.com/css2?family=Bangers&family=Comic+Neue:wght@400;700&display=swap');
+
+/* ===== 🎨 Comic Classic 배경 ===== */
+.gradio-container {
+    background-color: #FEF9C3 !important;
+    background-image: radial-gradient(#1F2937 1px, transparent 1px) !important;
+    background-size: 20px 20px !important;
+    min-height: 100vh !important;
+    font-family: 'Comic Neue', cursive, sans-serif !important;
+}
+
+/* ===== 허깅페이스 헤더 숨김 ===== */
+.huggingface-space-header,
+#space-header,
+.space-header,
+[class*="space-header"] {
+    display: none !important;
+}
+
+/* ===== Footer 숨김 ===== */
+footer, .footer, .gradio-footer, .built-with-gradio {
+    display: none !important;
+}
+
+/* ===== 🎨 헤더 타이틀 ===== */
+.header-text h1 {
+    font-family: 'Bangers', cursive !important;
+    color: #1F2937 !important;
+    font-size: 3.5rem !important;
+    text-align: center !important;
+    margin-bottom: 0.5rem !important;
+    text-shadow: 
+        4px 4px 0px #FACC15,
+        6px 6px 0px #1F2937 !important;
+    letter-spacing: 3px !important;
+    -webkit-text-stroke: 2px #1F2937 !important;
+}
+
+/* ===== 🎨 서브타이틀 ===== */
+.subtitle {
+    text-align: center !important;
+    font-family: 'Comic Neue', cursive !important;
+    font-size: 1.2rem !important;
+    color: #1F2937 !important;
+    margin-bottom: 1.5rem !important;
+    font-weight: 700 !important;
+}
+
+/* ===== 🎨 채팅 컨테이너 ===== */
+.chatbot {
+    background: #FFFFFF !important;
+    border: 3px solid #1F2937 !important;
+    border-radius: 12px !important;
+    box-shadow: 6px 6px 0px #1F2937 !important;
+}
+
+/* ===== 🎨 입력 필드 ===== */
+textarea, input[type="text"] {
+    background: #FFFFFF !important;
+    border: 3px solid #1F2937 !important;
+    border-radius: 8px !important;
+    color: #1F2937 !important;
+    font-family: 'Comic Neue', cursive !important;
+    font-size: 1rem !important;
+    font-weight: 700 !important;
+}
+
+textarea:focus, input[type="text"]:focus {
+    border-color: #3B82F6 !important;
+    box-shadow: 4px 4px 0px #3B82F6 !important;
+    outline: none !important;
+}
+
+/* ===== 🎨 Primary 버튼 ===== */
+.gr-button-primary, button.primary {
+    background: #3B82F6 !important;
+    border: 3px solid #1F2937 !important;
+    border-radius: 8px !important;
+    color: #FFFFFF !important;
+    font-family: 'Bangers', cursive !important;
+    font-size: 1.3rem !important;
+    letter-spacing: 2px !important;
+    padding: 14px 28px !important;
+    box-shadow: 5px 5px 0px #1F2937 !important;
+    text-shadow: 1px 1px 0px #1F2937 !important;
+}
+
+.gr-button-primary:hover, button.primary:hover {
+    background: #2563EB !important;
+    transform: translate(-2px, -2px) !important;
+    box-shadow: 7px 7px 0px #1F2937 !important;
+}
+
+.gr-button-primary:active, button.primary:active {
+    transform: translate(3px, 3px) !important;
+    box-shadow: 2px 2px 0px #1F2937 !important;
+}
+
+/* ===== 🎨 Secondary 버튼 ===== */
+.gr-button-secondary, button.secondary {
+    background: #EF4444 !important;
+    border: 3px solid #1F2937 !important;
+    border-radius: 8px !important;
+    color: #FFFFFF !important;
+    font-family: 'Bangers', cursive !important;
+    font-size: 1.1rem !important;
+    box-shadow: 4px 4px 0px #1F2937 !important;
+    text-shadow: 1px 1px 0px #1F2937 !important;
+}
+
+.gr-button-secondary:hover, button.secondary:hover {
+    background: #DC2626 !important;
+    transform: translate(-2px, -2px) !important;
+}
+
+/* ===== 🎨 카드/패널 ===== */
+.gr-panel, .gr-box, .block, .gr-group {
+    background: #FFFFFF !important;
+    border: 3px solid #1F2937 !important;
+    border-radius: 8px !important;
+    box-shadow: 6px 6px 0px #1F2937 !important;
+}
+
+/* ===== 🎨 아코디언 ===== */
+.gr-accordion {
+    background: #FACC15 !important;
+    border: 3px solid #1F2937 !important;
+    border-radius: 8px !important;
+    box-shadow: 4px 4px 0px #1F2937 !important;
+}
+
+/* ===== 🎨 모델 정보 박스 ===== */
+.model-info {
+    background: linear-gradient(135deg, #3B82F6 0%, #8B5CF6 100%) !important;
+    border: 3px solid #1F2937 !important;
+    border-radius: 12px !important;
+    padding: 15px !important;
+    color: white !important;
+    box-shadow: 5px 5px 0px #1F2937 !important;
+    margin-bottom: 20px !important;
+}
+
+/* ===== 🎨 Provider 상태 박스 ===== */
+.provider-status {
+    background: #1F2937 !important;
+    border: 3px solid #10B981 !important;
+    border-radius: 8px !important;
+    padding: 10px 15px !important;
+    color: #10B981 !important;
+    font-family: 'Courier New', monospace !important;
+    font-weight: bold !important;
+    text-align: center !important;
+    box-shadow: 4px 4px 0px #10B981 !important;
+}
+
+/* ===== 🎨 코드 블록 ===== */
+pre, code {
+    background: #1F2937 !important;
+    color: #10B981 !important;
+    border: 2px solid #10B981 !important;
+    border-radius: 6px !important;
+    font-family: 'Courier New', monospace !important;
+}
+
+/* ===== 🎨 스크롤바 ===== */
+::-webkit-scrollbar {
+    width: 12px;
+}
+
+::-webkit-scrollbar-track {
+    background: #FEF9C3;
+    border: 2px solid #1F2937;
+}
+
+::-webkit-scrollbar-thumb {
+    background: #3B82F6;
+    border: 2px solid #1F2937;
+}
+
+::-webkit-scrollbar-thumb:hover {
+    background: #EF4444;
+}
+
+/* ===== 반응형 ===== */
+@media (max-width: 768px) {
+    .header-text h1 {
+        font-size: 2.2rem !important;
+    }
+}
+"""
+
+# ============================================
+# Gradio Interface
+# ============================================
+with gr.Blocks(fill_height=True, css=css, title="MiniMax-M2.1 Chat") as demo:
+    
+    # HOME Badge
+    gr.HTML("""
+        <div style="text-align: center; margin: 20px 0 10px 0;">
+            <a href="https://www.humangen.ai" target="_blank" style="text-decoration: none;">
+                <img src="https://img.shields.io/static/v1?label=🏠 HOME&message=HUMANGEN.AI&color=0000ff&labelColor=ffcc00&style=for-the-badge" alt="HOME">
+            </a>
+            <a href="https://huggingface.co/MiniMaxAI/MiniMax-M2.1" target="_blank" style="text-decoration: none; margin-left: 10px;">
+                <img src="https://img.shields.io/static/v1?label=🤗&message=MiniMax-M2.1&color=ff6f00&labelColor=1F2937&style=for-the-badge" alt="Model">
+            </a>
+        </div>
+    """)
+    
+    # Header Title
+    gr.Markdown(
+        """# 🤖 MINIMAX-M2.1 CHAT 💬""",
+        elem_classes="header-text"
+    )
+    
+    gr.Markdown(
+        """<p class="subtitle">⚡ Claude Sonnet 4.5 수준의 코딩 & 에이전트 성능! 230B 파라미터 오픈소스 모델 🚀</p>""",
+    )
+    
+    # Model Info Box
+    gr.HTML("""
+        <div class="model-info">
+            <div style="display: flex; justify-content: space-around; flex-wrap: wrap; text-align: center;">
+                <div><strong style="font-size: 1.5rem;">230B</strong><br><span style="font-size: 0.9rem;">Total Params</span></div>
+                <div><strong style="font-size: 1.5rem;">10B</strong><br><span style="font-size: 0.9rem;">Active Params</span></div>
+                <div><strong style="font-size: 1.5rem;">88.6</strong><br><span style="font-size: 0.9rem;">VIBE Score</span></div>
+                <div><strong style="font-size: 1.5rem;">#1</strong><br><span style="font-size: 0.9rem;">Open Source</span></div>
+            </div>
+        </div>
+    """)
+    
+    # Provider Status
+    with gr.Row():
+        provider_status = gr.Textbox(
+            value="🟢 Novita (HuggingFace)",
+            label="🔌 Current Provider",
+            interactive=False,
+            elem_classes="provider-status"
+        )
+        reset_btn = gr.Button("🔄 Reset to Novita", variant="secondary", scale=0)
+    
+    # Chat Interface
+    chatbot = gr.Chatbot(
+        label="💬 Chat",
+        height=450,
+        show_label=False,
+        elem_classes="chatbot"
+    )
+    
+    with gr.Row():
+        msg = gr.Textbox(
+            label="",
+            placeholder="메시지를 입력하세요... (코딩, 분석, 창작 등 무엇이든!)",
+            scale=9,
+            container=False
+        )
+        submit_btn = gr.Button("📤 SEND", variant="primary", scale=1)
+    
+    with gr.Row():
+        clear_btn = gr.Button("🗑️ CLEAR CHAT", variant="secondary")
+    
+    # Examples
+    with gr.Accordion("💡 Example Prompts", open=False):
+        gr.Examples(
+            examples=[
+                "Python으로 퀵소트 알고리즘을 구현해줘",
+                "React로 Todo 앱 컴포넌트를 만들어줘",
+                "Docker와 Kubernetes의 차이점을 설명해줘",
+                "REST API 설계 베스트 프랙티스를 알려줘",
+                "FastAPI로 JWT 인증 구현해줘",
+            ],
+            inputs=msg
+        )
+    
+    # Provider Info
+    with gr.Accordion("🔌 Provider Information", open=False):
+        gr.Markdown("""
+        ### 듀얼 프로바이더 시스템
+        
+        이 앱은 **자동 폴백** 기능이 있습니다:
+        
+        **1차: Novita (HuggingFace)**
+        - HuggingFace PRO 크레딧 사용
+        - 빠른 응답 속도
+        
+        **2차: MiniMax Official API** (폴백)
+        - Novita 크레딧 소진 시 자동 전환
+        - MiniMax 공식 API (현재 한시적 무료)
+        
+        ### 필요한 Secrets
+        ```
+        HF_TOKEN: HuggingFace 토큰 (PRO 계정 권장)
+        MINIMAX_API_KEY: MiniMax 공식 API 키 (폴백용)
+        ```
+        
+        **MiniMax API 키 발급:** [platform.minimax.io](https://platform.minimax.io)
+        """)
+    
+    # Model capabilities
+    with gr.Accordion("🎯 Model Capabilities", open=False):
+        gr.Markdown("""
+        ### MiniMax-M2.1 특징
+        
+        - **🏆 SOTA 코딩 성능** — SWE-bench에서 Claude Sonnet 4.5 능가
+        - **🌍 다국어 프로그래밍** — Python, Rust, Java, Go, C++, TypeScript 등
+        - **🤖 에이전트 능력** — 복잡한 멀티스텝 태스크 수행
+        - **💡 추론 능력** — Interleaved Thinking으로 체계적 문제 해결
+        - **🔧 도구 사용** — 함수 호출 및 API 연동
+        """)
+    
+    # Event handlers
+    def respond(message, history):
+        history = history or []
+        bot_response = ""
+        current_status = ""
+        
+        for response, status in chat_stream(message, history):
+            bot_response = response
+            current_status = status
+            yield history + [(message, bot_response)], "", current_status
+    
+    def clear_chat():
+        return [], "", provider.get_status()
+    
+    def do_reset():
+        return reset_provider()
+    
+    # Connect events
+    msg.submit(respond, [msg, chatbot], [chatbot, msg, provider_status])
+    submit_btn.click(respond, [msg, chatbot], [chatbot, msg, provider_status])
+    clear_btn.click(clear_chat, outputs=[chatbot, msg, provider_status])
+    reset_btn.click(do_reset, outputs=[provider_status])
+
+if __name__ == "__main__":
+    demo.launch()