Update app.py

app.py

@@ -1908,10 +1908,161 @@ def burn_model_zero_shot(
 db = ExperimentDatabase(DB_PATH)
 
 # =====================================================
-# Gradio UI (기존 코드와 동일)
+# Gradio UI
 # =====================================================
 
-# (이전과 동일한 Gradio 코드)
+with gr.Blocks(
+    title="🔮 PHOENIX v1.4 - State Dict Direct Loading",
+    theme=gr.themes.Soft(),
+) as demo:
+    
+    gr.Markdown("""
+    # 🔮 PHOENIX Retention Platform v1.4
+    
+    **State Dict Direct Loading + Structure-Aware Burning**
+    
+    ✅ **NEW!** State Dict 직접 로드로 Retention 보존
+    ✅ Model Structure Pre-Analysis
+    ✅ Qwen3 Model Support
+    ✅ Zero-shot Conversion (No Dataset Required)
+    ✅ Optional Fine-tuning
+    ✅ GQA Support
+    ✅ O(n) Complexity
+    ✅ Auto Upload to HuggingFace Hub
+    
+    ---
+    """)
+    
+    with gr.Tabs():
+        with gr.Tab("🔥 Model Burning"):
+            gr.Markdown("""
+            ### 🔥 PHOENIX Model Burning v1.4
+            
+            **모델 구조를 먼저 분석한 후 변환합니다!**
+            **Hub 로드 시 State Dict 직접 로드로 Retention 보존!**
+            """)
+            
+            with gr.Row():
+                with gr.Column(scale=1):
+                    burn_model_url = gr.Textbox(
+                        label="🔗 Model URL",
+                        value=DEFAULT_MODEL,
+                        placeholder="Qwen/Qwen3-0.6B"
+                    )
+                    burn_hierarchical = gr.Checkbox(value=True, label="Hierarchical Retention")
+                    
+                    burn_output_name = gr.Textbox(
+                        label="💾 Output Name",
+                        placeholder="phoenix_my_model"
+                    )
+                    
+                    gr.Markdown("---")
+                    gr.Markdown("### 🌐 HuggingFace Hub Upload")
+                    
+                    burn_upload_hub = gr.Checkbox(value=True, label="📤 Upload to Hub")
+                    burn_hub_repo = gr.Textbox(label="📦 Repo Name (optional)")
+                    burn_hub_private = gr.Checkbox(value=True, label="🔒 Private")
+                    
+                    gr.Markdown("---")
+                    gr.Markdown("### 📊 Dataset (Optional)")
+                    
+                    burn_dataset = gr.Textbox(label="📁 Dataset Path")
+                    burn_use_finetuning = gr.Checkbox(value=False, label="🚀 Enable Fine-tuning")
+                    
+                    with gr.Accordion("⚙️ Fine-tuning Config", open=False):
+                        burn_epochs = gr.Slider(1, 5, 1, step=1, label="Epochs")
+                        burn_batch = gr.Slider(1, 16, 4, step=1, label="Batch Size")
+                        burn_lr = gr.Number(value=5e-5, label="Learning Rate")
+                        burn_max_steps = gr.Slider(10, 500, 100, step=10, label="Max Steps")
+                    
+                    burn_btn = gr.Button("🔥 Burn Model", variant="primary", size="lg")
+                
+                with gr.Column(scale=2):
+                    burn_output = gr.Markdown()
+                    burn_plot = gr.Plot()
+            
+            burn_btn.click(
+                burn_phoenix_model_ui,
+                [
+                    burn_model_url, burn_hierarchical, burn_dataset, burn_output_name,
+                    burn_use_finetuning, burn_epochs, burn_batch, burn_lr, burn_max_steps,
+                    burn_upload_hub, burn_hub_repo, burn_hub_private,
+                ],
+                [burn_output, burn_plot]
+            )
+        
+        with gr.Tab("📊 Burning History"):
+            gr.Markdown("### 📊 Model Burning History")
+            
+            with gr.Row():
+                with gr.Column(scale=1):
+                    hist_btn = gr.Button("📊 Load History", variant="primary")
+                
+                with gr.Column(scale=2):
+                    hist_output = gr.Markdown()
+                    hist_plot = gr.Plot()
+            
+            hist_btn.click(view_burning_history, outputs=[hist_output, hist_plot])
+
+        with gr.Tab("🧪 Model Validation"):
+            gr.Markdown("### 🧪 PHOENIX 모델 검증")
+            
+            with gr.Row():
+                with gr.Column(scale=1):
+                    val_source = gr.Radio(
+                        choices=["hub", "local"],
+                        value="hub",
+                        label="📍 Model Source"
+                    )
+                    
+                    val_path = gr.Textbox(
+                        label="🔗 Model Path/URL",
+                        value="seawolf2357/phoenix-Qwen3-0.6B",
+                        placeholder="seawolf2357/phoenix-model"
+                    )
+                    
+                    val_prompts = gr.Textbox(
+                        label="📝 Test Prompts (one per line)",
+                        lines=5,
+                        value="The future of AI is\nOnce upon a time\nIn machine learning,",
+                    )
+                    
+                    with gr.Row():
+                        val_max_tokens = gr.Slider(16, 256, 64, step=16, label="Max Tokens")
+                        val_temp = gr.Slider(0.1, 2.0, 0.7, step=0.1, label="Temperature")
+                    
+                    val_verify_retention = gr.Checkbox(value=True, label="🔍 Verify Retention")
+                    
+                    val_btn = gr.Button("🧪 Validate Model", variant="primary", size="lg")
+                
+                with gr.Column(scale=2):
+                    val_output = gr.Markdown()
+                    val_plot = gr.Plot()
+            
+            val_btn.click(
+                validate_phoenix_model,
+                [val_source, val_path, val_prompts, val_max_tokens, 
+                 val_temp, val_verify_retention],
+                [val_output, val_plot]
+            )
+    
+    gr.Markdown(f"""
+    ---
+    
+    ## 🔥 PHOENIX Model Burning Platform v1.4
+    
+    ### What's New in v1.4
+    - ✅ **State Dict Direct Loading** - Hub 로드 시 Retention 가중치 보존
+    - ✅ **Fixed Hub Loading** - Custom Code에서 올바른 가중치 로드
+    - ✅ **Model Structure Pre-Analysis** - 변환 전 구조 파악
+    - ✅ **Qwen3 Support** - Qwen3 모델 완벽 지원
+    
+    **HuggingFace Token**: {'✅ Connected' if HF_TOKEN else '❌ Not Found'}
+    **Default Model**: {DEFAULT_MODEL}
+    
+    **VIDraft AI Research Lab** | PHOENIX v1.4
+    """)
 
 if __name__ == "__main__":
-    print("PHOENIX v1.4.1 Ready!")
+    demo.queue(max_size=20)
+    demo.launch(server_name="0.0.0.0", server_port=7860, share=False)