Update app.py

app.py

@@ -6,13 +6,13 @@ import os
 # ============================================
 # MiniMax-M2.1 Streaming Chat
 # Dual Provider: Novita (HF) + MiniMax Official API
-# Comic Classic Theme
+# Gradio 6.0 Compatible
 # ============================================
 
 # Provider 상태 관리
 class ProviderManager:
     def __init__(self):
-        self.current_provider = "novita"  # 기본값: Novita
+        self.current_provider = "novita"
         self.novita_available = True
         
         # Novita Client (via HuggingFace)
@@ -79,39 +79,44 @@ def chat_with_minimax(messages):
 def chat_stream(message, history):
     """
     Streaming chat with automatic fallback
-    1차: Novita (HuggingFace) 
-    2차: MiniMax Official API (fallback)
+    Gradio 6.0 messages format compatible
     """
     if not message.strip():
-        yield "", provider.get_status()
+        yield [], provider.get_status()
         return
     
-    # Build messages
-    messages = [{
+    # Build API messages from Gradio history
+    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."
     }]
     
-    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})
+    # 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", "")
+            })
     
-    messages.append({"role": "user", "content": message})
+    api_messages.append({"role": "user", "content": message})
     
     response_text = ""
     
-    # 1차 시도: Novita (if available)
+    # 1차 시도: Novita
     if provider.novita_available and provider.current_provider == "novita":
         try:
-            for chunk in chat_with_novita(messages):
+            for chunk in chat_with_novita(api_messages):
                 response_text += chunk
-                yield response_text, "🟢 Novita (HuggingFace)"
+                # Gradio 6.0 messages format
+                new_history = history + [
+                    {"role": "user", "content": message},
+                    {"role": "assistant", "content": response_text}
+                ]
+                yield new_history, "🟢 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"
@@ -119,43 +124,53 @@ def chat_stream(message, history):
                 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"
+            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"
             return
         
-        for chunk in chat_with_minimax(messages):
+        for chunk in chat_with_minimax(api_messages):
             response_text += chunk
-            yield response_text, "🟡 MiniMax Official API"
+            new_history = history + [
+                {"role": "user", "content": message},
+                {"role": "assistant", "content": response_text}
+            ]
+            yield new_history, "🟡 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"
+        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 "✅ Provider reset to Novita. Next message will try Novita first."
+    return "✅ Reset to Novita!"
+
+def clear_chat():
+    """채팅 초기화"""
+    return [], "", provider.get_status()
 
 # ============================================
 # 🎨 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;
@@ -164,34 +179,25 @@ css = """
     font-family: 'Comic Neue', cursive, sans-serif !important;
 }
 
-/* ===== 허깅페이스 헤더 숨김 ===== */
-.huggingface-space-header,
-#space-header,
-.space-header,
-[class*="space-header"] {
+.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;
+    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;
@@ -201,7 +207,6 @@ footer, .footer, .gradio-footer, .built-with-gradio {
     font-weight: 700 !important;
 }
 
-/* ===== 🎨 채팅 컨테이너 ===== */
 .chatbot {
     background: #FFFFFF !important;
     border: 3px solid #1F2937 !important;
@@ -209,7 +214,6 @@ footer, .footer, .gradio-footer, .built-with-gradio {
     box-shadow: 6px 6px 0px #1F2937 !important;
 }
 
-/* ===== 🎨 입력 필드 ===== */
 textarea, input[type="text"] {
     background: #FFFFFF !important;
     border: 3px solid #1F2937 !important;
@@ -226,7 +230,6 @@ textarea:focus, input[type="text"]:focus {
     outline: none !important;
 }
 
-/* ===== 🎨 Primary 버튼 ===== */
 .gr-button-primary, button.primary {
     background: #3B82F6 !important;
     border: 3px solid #1F2937 !important;
@@ -251,7 +254,6 @@ textarea:focus, input[type="text"]:focus {
     box-shadow: 2px 2px 0px #1F2937 !important;
 }
 
-/* ===== 🎨 Secondary 버튼 ===== */
 .gr-button-secondary, button.secondary {
     background: #EF4444 !important;
     border: 3px solid #1F2937 !important;
@@ -268,7 +270,6 @@ textarea:focus, input[type="text"]:focus {
     transform: translate(-2px, -2px) !important;
 }
 
-/* ===== 🎨 카드/패널 ===== */
 .gr-panel, .gr-box, .block, .gr-group {
     background: #FFFFFF !important;
     border: 3px solid #1F2937 !important;
@@ -276,7 +277,6 @@ textarea:focus, input[type="text"]:focus {
     box-shadow: 6px 6px 0px #1F2937 !important;
 }
 
-/* ===== 🎨 아코디언 ===== */
 .gr-accordion {
     background: #FACC15 !important;
     border: 3px solid #1F2937 !important;
@@ -284,7 +284,6 @@ textarea:focus, input[type="text"]:focus {
     box-shadow: 4px 4px 0px #1F2937 !important;
 }
 
-/* ===== 🎨 모델 정보 박스 ===== */
 .model-info {
     background: linear-gradient(135deg, #3B82F6 0%, #8B5CF6 100%) !important;
     border: 3px solid #1F2937 !important;
@@ -295,7 +294,6 @@ textarea:focus, input[type="text"]:focus {
     margin-bottom: 20px !important;
 }
 
-/* ===== 🎨 Provider 상태 박스 ===== */
 .provider-status {
     background: #1F2937 !important;
     border: 3px solid #10B981 !important;
@@ -308,7 +306,6 @@ textarea:focus, input[type="text"]:focus {
     box-shadow: 4px 4px 0px #10B981 !important;
 }
 
-/* ===== 🎨 코드 블록 ===== */
 pre, code {
     background: #1F2937 !important;
     color: #10B981 !important;
@@ -317,37 +314,23 @@ pre, code {
     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 { 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; }
 
-::-webkit-scrollbar-thumb:hover {
-    background: #EF4444;
-}
-
-/* ===== 반응형 ===== */
 @media (max-width: 768px) {
-    .header-text h1 {
-        font-size: 2.2rem !important;
-    }
+    .header-text h1 { font-size: 2.2rem !important; }
 }
 """
 
 # ============================================
-# Gradio Interface
+# Gradio Interface (6.0 Compatible)
 # ============================================
-with gr.Blocks(fill_height=True, css=css, title="MiniMax-M2.1 Chat") as demo:
+with gr.Blocks(title="MiniMax-M2.1 Chat") as demo:
+    
+    # Inject CSS
+    gr.HTML(f"<style>{css}</style>")
     
     # HOME Badge
     gr.HTML("""
@@ -355,23 +338,15 @@ with gr.Blocks(fill_height=True, css=css, title="MiniMax-M2.1 Chat") as demo:
             <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"
-    )
+    # Header
+    gr.Markdown("""# 🤖 MINIMAX-M2.1 CHAT 💬""", elem_classes="header-text")
+    gr.Markdown("""<p class="subtitle">⚡ Claude Sonnet 4.5 수준의 코딩 & 에이전트 성능! 230B 파라미터 오픈소스 모델 🚀</p>""")
     
-    gr.Markdown(
-        """<p class="subtitle">⚡ Claude Sonnet 4.5 수준의 코딩 & 에이전트 성능! 230B 파라미터 오픈소스 모델 🚀</p>""",
-    )
-    
-    # Model Info Box
+    # Model Info
     gr.HTML("""
         <div class="model-info">
             <div style="display: flex; justify-content: space-around; flex-wrap: wrap; text-align: center;">
@@ -391,13 +366,14 @@ with gr.Blocks(fill_height=True, css=css, title="MiniMax-M2.1 Chat") as demo:
             interactive=False,
             elem_classes="provider-status"
         )
-        reset_btn = gr.Button("🔄 Reset to Novita", variant="secondary", scale=0)
+        reset_btn = gr.Button("🔄 Reset", variant="secondary", scale=0)
     
-    # Chat Interface
+    # 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"
     )
     
@@ -429,61 +405,43 @@ with gr.Blocks(fill_height=True, css=css, title="MiniMax-M2.1 Chat") as demo:
     # 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)
+### 듀얼 프로바이더 시스템
+
+**1차: Novita (HuggingFace)** - HF PRO 크레딧 사용
+
+**2차: MiniMax Official API** - 크레딧 소진 시 자동 전환
+
+### 필요한 Secrets
+- `HF_TOKEN`: HuggingFace 토큰
+- `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 연동
+- **🏆 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 = ""
+        if not message.strip():
+            yield history, "", provider.get_status()
+            return
         
-        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()
+        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(do_reset, outputs=[provider_status])
+    reset_btn.click(reset_provider, outputs=[provider_status])
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(ssr_mode=False)