Update app.py

app.py

@@ -6,7 +6,6 @@ import os
 # ============================================
 # MiniMax-M2.1 Streaming Chat
 # Dual Provider: Novita (HF) + MiniMax Official API
-# Gradio 6.0 Compatible
 # ============================================
 
 # Provider 상태 관리
@@ -29,9 +28,9 @@ class ProviderManager:
     
     def get_status(self):
         if self.current_provider == "novita":
-            return "🟢 Novita (HuggingFace)"
+            return "🟢 Novita"
         else:
-            return "🟡 MiniMax Official API"
+            return "🟡 MiniMax"
     
     def switch_to_minimax(self):
         self.current_provider = "minimax"
@@ -76,94 +75,64 @@ def chat_with_minimax(messages):
         if chunk.choices and chunk.choices[0].delta.content:
             yield chunk.choices[0].delta.content
 
-def chat_stream(message, history):
+def respond(message, history):
     """
     Streaming chat with automatic fallback
-    Gradio 6.0 messages format compatible
     """
     if not message.strip():
-        yield [], provider.get_status()
+        yield ""
         return
     
-    # Build API messages from Gradio history
+    # Build API messages
     api_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."
     }]
     
-    # Convert Gradio messages format to API format
-    for msg in history:
-        if isinstance(msg, dict):
-            api_messages.append({
-                "role": msg.get("role", "user"),
-                "content": msg.get("content", "")
-            })
+    # Add history
+    for h in history:
+        if isinstance(h, dict):
+            api_messages.append({"role": h.get("role", "user"), "content": h.get("content", "")})
+        elif isinstance(h, (list, tuple)) and len(h) == 2:
+            if h[0]:
+                api_messages.append({"role": "user", "content": h[0]})
+            if h[1]:
+                api_messages.append({"role": "assistant", "content": h[1]})
     
     api_messages.append({"role": "user", "content": message})
     
     response_text = ""
     
-    # 1차 시도: Novita
+    # 1차: Novita
     if provider.novita_available and provider.current_provider == "novita":
         try:
             for chunk in chat_with_novita(api_messages):
                 response_text += chunk
-                # Gradio 6.0 messages format
-                new_history = history + [
-                    {"role": "user", "content": message},
-                    {"role": "assistant", "content": response_text}
-                ]
-                yield new_history, "🟢 Novita (HuggingFace)"
+                yield response_text
             return
         except Exception as e:
             error_msg = str(e).lower()
-            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}")
+            if any(kw in error_msg for kw in ["rate limit", "quota", "exceeded", "insufficient", "credit", "balance", "limit", "429", "402", "payment"]):
+                print(f"⚠️ Novita error: {e}")
                 provider.switch_to_minimax()
             else:
-                print(f"⚠️ Novita temporary error: {e}")
+                print(f"⚠️ Novita error: {e}")
     
-    # 2차 시도: MiniMax Official API
+    # 2차: MiniMax
     try:
         if not os.environ.get("MINIMAX_API_KEY"):
-            error_history = history + [
-                {"role": "user", "content": message},
-                {"role": "assistant", "content": "❌ Error: MINIMAX_API_KEY not configured. Please add it to Secrets."}
-            ]
-            yield error_history, "🔴 No Provider"
+            yield "❌ Error: MINIMAX_API_KEY not configured."
             return
         
         for chunk in chat_with_minimax(api_messages):
             response_text += chunk
-            new_history = history + [
-                {"role": "user", "content": message},
-                {"role": "assistant", "content": response_text}
-            ]
-            yield new_history, "🟡 MiniMax Official API"
+            yield response_text
         return
         
     except Exception as e:
-        error_msg = str(e)
-        if "rate limit" not in error_msg.lower():
+        if "rate limit" not in str(e).lower():
             provider.reset_to_novita()
-        
-        error_history = history + [
-            {"role": "user", "content": message},
-            {"role": "assistant", "content": f"❌ Error: {error_msg}"}
-        ]
-        yield error_history, "🔴 Error"
-
-def reset_provider():
-    """수동으로 Novita로 리셋"""
-    provider.reset_to_novita()
-    return "✅ Reset to Novita!"
-
-def clear_chat():
-    """채팅 초기화"""
-    return [], "", provider.get_status()
+        yield f"❌ Error: {str(e)}"
 
 # ============================================
 # 🎨 Comic Classic Theme CSS
@@ -187,24 +156,13 @@ footer, .footer, .gradio-footer, .built-with-gradio {
     display: none !important;
 }
 
-.header-text h1 {
+h1 {
     font-family: 'Bangers', cursive !important;
     color: #1F2937 !important;
-    font-size: 3.5rem !important;
+    font-size: 3rem !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 {
@@ -230,44 +188,35 @@ textarea:focus, input[type="text"]:focus {
     outline: none !important;
 }
 
-.gr-button-primary, button.primary {
+button {
     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;
+    font-size: 1.1rem !important;
     letter-spacing: 2px !important;
-    padding: 14px 28px !important;
-    box-shadow: 5px 5px 0px #1F2937 !important;
+    box-shadow: 4px 4px 0px #1F2937 !important;
     text-shadow: 1px 1px 0px #1F2937 !important;
 }
 
-.gr-button-primary:hover, button.primary:hover {
+button:hover {
     background: #2563EB !important;
     transform: translate(-2px, -2px) !important;
-    box-shadow: 7px 7px 0px #1F2937 !important;
+    box-shadow: 6px 6px 0px #1F2937 !important;
 }
 
-.gr-button-primary:active, button.primary:active {
-    transform: translate(3px, 3px) !important;
+button:active {
+    transform: translate(2px, 2px) !important;
     box-shadow: 2px 2px 0px #1F2937 !important;
 }
 
-.gr-button-secondary, button.secondary {
+.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 {
+.secondary:hover {
     background: #DC2626 !important;
-    transform: translate(-2px, -2px) !important;
 }
 
 .gr-panel, .gr-box, .block, .gr-group {
@@ -291,19 +240,7 @@ textarea:focus, input[type="text"]:focus {
     padding: 15px !important;
     color: white !important;
     box-shadow: 5px 5px 0px #1F2937 !important;
-    margin-bottom: 20px !important;
-}
-
-.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;
+    margin: 20px 0 !important;
 }
 
 pre, code {
@@ -320,128 +257,47 @@ pre, code {
 ::-webkit-scrollbar-thumb:hover { background: #EF4444; }
 
 @media (max-width: 768px) {
-    .header-text h1 { font-size: 2.2rem !important; }
+    h1 { font-size: 2rem !important; }
 }
 """
 
 # ============================================
-# Gradio Interface (6.0 Compatible)
+# Gradio ChatInterface (Simple & Compatible)
 # ============================================
-with gr.Blocks(title="MiniMax-M2.1 Chat") as demo:
-    
-    # Inject CSS
-    gr.HTML(f"<style>{css}</style>")
-    
-    # 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>
 
-        </div>
-    """)
-    
-    # Header
-    gr.Markdown("""# 🤖 MINIMAX-M2.1 CHAT 💬""", elem_classes="header-text")
-    gr.Markdown("""<p class="subtitle">⚡ Claude Sonnet 4.5 수준의 코딩 & 에이전트 성능! 230B 파라미터 오픈소스 모델 🚀</p>""")
-    
-    # Model Info
-    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", variant="secondary", scale=0)
-    
-    # Chat Interface - Gradio 6.0 messages format
-    chatbot = gr.Chatbot(
-        label="💬 Chat",
-        height=450,
-        show_label=False,
-        type="messages",  # Gradio 6.0 format
-        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)** - HF PRO 크레딧 사용
-
-**2차: MiniMax Official API** - 크레딧 소진 시 자동 전환
-
-### 필요한 Secrets
-- `HF_TOKEN`: HuggingFace 토큰
-- `MINIMAX_API_KEY`: MiniMax API 키
+# Header HTML
+header_html = """
+<div style="text-align: center; margin: 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>
+
+</div>
+<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>
+"""
 
-**MiniMax API:** [platform.minimax.io](https://platform.minimax.io)
-        """)
-    
-    # Model capabilities
-    with gr.Accordion("🎯 Model Capabilities", open=False):
-        gr.Markdown("""
-- **🏆 SOTA 코딩** — SWE-bench에서 Claude Sonnet 4.5 능가
-- **🌍 다국어** — Python, Rust, Java, Go, C++, TypeScript
-- **🤖 에이전트** — 복잡한 멀티스텝 태스크
-- **💡 추론** — Interleaved Thinking
-- **🔧 도구** — 함수 호출 및 API 연동
-        """)
-    
-    # Event handlers
-    def respond(message, history):
-        if not message.strip():
-            yield history, "", provider.get_status()
-            return
-        
-        for new_history, status in chat_stream(message, history):
-            yield new_history, "", status
-    
-    # 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(reset_provider, outputs=[provider_status])
+# Create ChatInterface
+demo = gr.ChatInterface(
+    fn=respond,
+    title="🤖 MINIMAX-M2.1 CHAT 💬",
+    description=header_html,
+    examples=[
+        "Python으로 퀵소트 알고리즘을 구현해줘",
+        "React로 Todo 앱 컴포넌트를 만들어줘",
+        "Docker와 Kubernetes의 차이점을 설명해줘",
+        "FastAPI로 JWT 인증 구현해줘",
+        "머신러닝 모델 배포 파이프라인을 설계해줘",
+    ],
+    css=css,
+    theme=gr.themes.Soft(),
+)
 
 if __name__ == "__main__":
-    demo.launch(ssr_mode=False)
+    demo.launch()