Update app.py

app.py

@@ -104,7 +104,7 @@ def create_sample_workflow(example_type="basic"):
     """샘플 워크플로우 생성"""
     
     if example_type == "basic":
-        # 기본 예제: 간단한 Q&A
+        # 기본 예제: 간단한 Q&A - VIDraft 사용
         return {
             "nodes": [
                 {
@@ -125,8 +125,8 @@ def create_sample_workflow(example_type="basic"):
                     "data": {
                         "label": "AI Processing",
                         "template": {
-                            "provider": {"value": "OpenAI"},
-                            "model": {"value": "gpt-4.1-mini"},
+                            "provider": {"value": "VIDraft"},  # 기본값을 VIDraft로 변경
+                            "model": {"value": "Gemma-3-r1984-27B"},
                             "temperature": {"value": 0.7},
                             "system_prompt": {"value": "You are a helpful assistant."}
                         }
@@ -242,8 +242,8 @@ def create_sample_workflow(example_type="basic"):
                     "data": {
                         "label": "Generate Learning Plan",
                         "template": {
-                            "provider": {"value": "OpenAI"},
-                            "model": {"value": "gpt-4.1-mini"},
+                            "provider": {"value": "VIDraft"},  # 기본값을 VIDraft로 변경
+                            "model": {"value": "Gemma-3-r1984-27B"},
                             "temperature": {"value": 0.7},
                             "system_prompt": {"value": "You are an expert educational consultant. Create personalized learning plans based on the user's name, topic of interest, and skill level."}
                         }
@@ -396,8 +396,10 @@ def execute_workflow(*input_values):
                 results[node_id] = base_text
                 
         elif node_type in ["llmNode", "OpenAIModel", "ChatModel"]:
-            # Get provider and model
-            provider = template.get("provider", {{}}).get("value", "OpenAI")
+            # Get provider and model - VIDraft as default
+            provider = template.get("provider", {{}}).get("value", "VIDraft")
+            if provider not in ["VIDraft", "OpenAI"]:
+                provider = "VIDraft"  # Default to VIDraft
             temperature = template.get("temperature", {{}}).get("value", 0.7)
             system_prompt = template.get("system_prompt", {{}}).get("value", "")
             
@@ -473,7 +475,7 @@ def execute_workflow(*input_values):
                 elif provider == "VIDraft":
                     results[node_id] = "[VIDraft API key not found. Please set FRIENDLI_TOKEN in Space secrets]"
                 else:
-                    results[node_id] = f"[Simulated Response: {{input_text[:50]}}...]"
+                    results[node_id] = f"[No API key found for {provider}. Using simulated response: {input_text[:50]}...]"
                 
         elif node_type in ["ChatOutput", "textOutput", "Output"]:
             # Get connected result
@@ -497,14 +499,29 @@ with gr.Blocks(title="{app_name}", theme=gr.themes.Soft()) as demo:
     vidraft_token = os.getenv("FRIENDLI_TOKEN")
     openai_key = os.getenv("OPENAI_API_KEY")
     
-    if not vidraft_token and not openai_key:
-        gr.Markdown("""
-        ⚠️ **API Keys Required**
+    with gr.Accordion("🔑 API Status", open=False):
+        if vidraft_token:
+            gr.Markdown("✅ **VIDraft API**: Connected (Gemma-3-r1984-27B)")
+        else:
+            gr.Markdown("❌ **VIDraft API**: Not configured")
+            
+        if openai_key:
+            gr.Markdown("✅ **OpenAI API**: Connected (gpt-4.1-mini)")
+        else:
+            gr.Markdown("⚠️ **OpenAI API**: Not configured (optional)")
         
-        Please set the following environment variables in Space settings → Secrets:
-        - `FRIENDLI_TOKEN` for VIDraft (Gemma-3-r1984-27B)
-        - `OPENAI_API_KEY` for OpenAI (gpt-4.1-mini)
-        """)
+        if not vidraft_token:
+            gr.Markdown("""
+            **⚠️ Important**: Please add FRIENDLI_TOKEN to Space secrets for the app to work properly.
+            
+            Go to: Space settings → Repository secrets → Add secret
+            """)
+        elif not openai_key:
+            gr.Markdown("""
+            **💡 Tip**: The app will work with VIDraft alone. Add OPENAI_API_KEY if you need OpenAI features.
+            """)
+        else:
+            gr.Markdown("**✨ All APIs configured! Your app is fully functional.**")
     
     # Extract nodes
     nodes = WORKFLOW_DATA.get("nodes", [])
@@ -559,12 +576,15 @@ requests
 
 def deploy_to_huggingface(workflow_data: dict, app_name: str, app_description: str, 
                          hf_token: str, space_name: str, is_private: bool = False,
-                         include_friendli_token: bool = False, friendli_token: str = None) -> dict:
-    """Deploy workflow to Hugging Face Space"""
+                         api_keys: dict = None) -> dict:
+    """Deploy workflow to Hugging Face Space with API keys"""
     
     if not HF_HUB_AVAILABLE:
         return {"success": False, "error": "huggingface-hub library not installed"}
     
+    if api_keys is None:
+        api_keys = {}
+    
     try:
         # Initialize HF API
         api = HfApi(token=hf_token)
@@ -578,11 +598,32 @@ def deploy_to_huggingface(workflow_data: dict, app_name: str, app_description: s
             exist_ok=True
         )
         
+        # Detect which providers are used in the workflow
+        providers_used = set()
+        nodes = workflow_data.get("nodes", [])
+        for node in nodes:
+            if node.get("type") in ["llmNode", "OpenAIModel", "ChatModel"]:
+                template = node.get("data", {}).get("template", {})
+                provider = template.get("provider", {}).get("value", "")
+                if provider:
+                    providers_used.add(provider)
+        
         # Generate files
         app_code = generate_standalone_app(workflow_data, app_name, app_description)
         requirements = generate_requirements_txt()
         
         # README with API setup instructions
+        api_status = []
+        if "FRIENDLI_TOKEN" in api_keys and api_keys["FRIENDLI_TOKEN"]:
+            api_status.append("- **FRIENDLI_TOKEN**: ✅ Will be configured automatically")
+        else:
+            api_status.append("- **FRIENDLI_TOKEN**: ⚠️ Not provided (VIDraft won't work)")
+            
+        if "OPENAI_API_KEY" in api_keys and api_keys["OPENAI_API_KEY"]:
+            api_status.append("- **OPENAI_API_KEY**: ✅ Will be configured automatically")
+        elif "OpenAI" in providers_used:
+            api_status.append("- **OPENAI_API_KEY**: ❌ Required but not provided")
+        
         readme = f"""---
 title: {app_name}
 emoji: 🐭
@@ -598,25 +639,17 @@ pinned: false
 
 {app_description}
 
-## 🔑 API Configuration
+## 🔑 API Configuration Status
 
-This app requires API keys to function properly. Please add the following secrets in your Space settings:
+{chr(10).join(api_status)}
 
-### For VIDraft (Gemma-3-r1984-27B):
-- **Secret Name**: `FRIENDLI_TOKEN`
-- **Secret Value**: Your VIDraft API token (starts with `flp_`)
-{f"- **Status**: ✅ Already configured" if include_friendli_token and friendli_token else "- **Status**: ❌ Needs to be added manually"}
+## 📝 Providers Used in This Workflow
 
-### For OpenAI (gpt-4.1-mini):
-- **Secret Name**: `OPENAI_API_KEY`
-- **Secret Value**: Your OpenAI API key (starts with `sk-`)
-- **Status**: ❌ Needs to be added manually
+{', '.join(providers_used) if providers_used else 'No LLM providers detected'}
 
-## 📝 How to add secrets:
-1. Go to your Space settings
-2. Click on "Repository secrets" 
-3. Add the required secrets
-4. Restart your Space
+## 🚀 Default Configuration
+
+This app is configured to use **VIDraft (Gemma-3-r1984-27B)** as the default LLM provider for optimal performance.
 
 ---
 Generated by MOUSE Workflow
@@ -644,17 +677,22 @@ Generated by MOUSE Workflow
             repo_type="space"
         )
         
-        # Add secrets if requested
-        if include_friendli_token and friendli_token:
-            try:
-                # Add FRIENDLI_TOKEN as a secret
-                api.add_space_secret(
-                    repo_id=repo_id.repo_id,
-                    key="FRIENDLI_TOKEN",
-                    value=friendli_token
-                )
-            except Exception as e:
-                print(f"Warning: Could not add FRIENDLI_TOKEN secret: {e}")
+        # Add all provided API keys as secrets
+        added_secrets = []
+        failed_secrets = []
+        
+        for key_name, key_value in api_keys.items():
+            if key_value:  # Only add non-empty keys
+                try:
+                    api.add_space_secret(
+                        repo_id=repo_id.repo_id,
+                        key=key_name,
+                        value=key_value
+                    )
+                    added_secrets.append(key_name)
+                except Exception as e:
+                    failed_secrets.append(f"{key_name}: {str(e)}")
+                    print(f"Warning: Could not add {key_name} secret: {e}")
         
         space_url = f"https://huggingface.co/spaces/{repo_id.repo_id}"
         
@@ -662,7 +700,9 @@ Generated by MOUSE Workflow
             "success": True,
             "space_url": space_url,
             "message": f"Successfully deployed to {space_url}",
-            "friendli_token_added": include_friendli_token and friendli_token
+            "added_secrets": added_secrets,
+            "failed_secrets": failed_secrets,
+            "providers_used": list(providers_used)
         }
         
     except Exception as e:
@@ -745,11 +785,11 @@ def execute_workflow_simple(workflow_data: dict, input_values: dict) -> dict:
                 
                 # 프로바이더 정보 추출 - VIDraft 또는 OpenAI만 허용
                 provider_info = template.get("provider", {})
-                provider = provider_info.get("value", "OpenAI") if isinstance(provider_info, dict) else "OpenAI"
+                provider = provider_info.get("value", "VIDraft") if isinstance(provider_info, dict) else "VIDraft"  # 기본값 VIDraft
                 
-                # provider가 VIDraft 또는 OpenAI가 아닌 경우 OpenAI로 기본 설정
+                # provider가 VIDraft 또는 OpenAI가 아닌 경우 VIDraft로 기본 설정
                 if provider not in ["VIDraft", "OpenAI"]:
-                    provider = "OpenAI"
+                    provider = "VIDraft"
                 
                 # 모델 정보 추출
                 if provider == "OpenAI":
@@ -759,7 +799,7 @@ def execute_workflow_simple(workflow_data: dict, input_values: dict) -> dict:
                     # VIDraft는 Gemma-3-r1984-27B로 고정
                     model = "Gemma-3-r1984-27B"
                 else:
-                    model = "gpt-4.1-mini"  # 기본값
+                    model = "Gemma-3-r1984-27B"  # 기본값 VIDraft 모델
                 
                 # 온도 정보 추출
                 temp_info = template.get("temperature", {})
@@ -913,6 +953,25 @@ CSS = """
     padding:24px;border-radius:12px;margin:24px 0;
     border:1px solid #fbbf24;
 }
+.save-indicator{
+    text-align:right;
+    font-size:14px;
+    color:#16a34a;
+    padding:8px 16px;
+    background:#f0fdf4;
+    border-radius:20px;
+    display:inline-block;
+    margin-left:auto;
+}
+.workflow-info{
+    font-size:14px;
+    color:#475569;
+    background:#f8fafc;
+    padding:8px 16px;
+    border-radius:8px;
+    display:inline-block;
+    margin-bottom:16px;
+}
 """
 
 # -------------------------------------------------------------------
@@ -925,10 +984,11 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
         gr.Markdown("**Visual Workflow Builder with Interactive UI Execution**")
         gr.HTML('<p class="powered-by">@Powered by VIDraft & Huggingface gradio</p>')
         
-        gr.HTML(
+                    gr.HTML(
             """
             <div class="component-description">
-              <p style="font-size:16px;margin:0;">Build sophisticated workflows visually • Import/Export JSON • Generate interactive UI for end-users</p>
+              <p style="font-size:16px;margin:0;">Build sophisticated workflows visually • Import/Export JSON • Generate interactive UI for end-users • Default LLM: VIDraft (Gemma-3-r1984-27B)</p>
+              <p style="font-size:14px;margin-top:8px;color:#64748b;">💡 Tip: Your workflow is automatically saved as you make changes. The JSON preview updates in real-time!</p>
             </div>
             """
         )
@@ -950,34 +1010,56 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
             - VIDraft: Gemma-3-r1984-27B (model ID: dep89a2fld32mcm)
             
             **Sample Workflows:**
-            - Basic Q&A: Simple question-answer flow
+            - Basic Q&A: Simple question-answer flow (VIDraft)
             - VIDraft: Korean language example with Gemma model
-            - Multi-Input: Combine multiple inputs for personalized output
-            - Chain: Sequential processing with multiple outputs
+            - Multi-Input: Combine multiple inputs for personalized output (VIDraft)
+            - Chain: Sequential processing with multiple outputs (VIDraft + OpenAI)
             
-            *Note: Without API keys, the UI will simulate AI responses.*
+            **Note**: All examples prioritize VIDraft for optimal performance. Friendli API token will be automatically configured during deployment.
             """)
         
         # State for storing workflow data
         loaded_data = gr.State(None)
         trigger_update = gr.State(False)
+        save_status = gr.State("Ready")
         
         # ─── Dynamic Workflow Container ───
         with gr.Column(elem_classes=["workflow-container"]):
+            # Auto-save status indicator
+            with gr.Row():
+                gr.Markdown("### 🎨 Visual Workflow Designer")
+                save_indicator = gr.Markdown("💾 Auto-save: Ready", elem_classes=["save-indicator"])
+            
             @gr.render(inputs=[loaded_data, trigger_update])
             def render_workflow(data, trigger):
                 """동적으로 WorkflowBuilder 렌더링"""
                 workflow_value = data if data else {"nodes": [], "edges": []}
                 
-                return WorkflowBuilder(
-                    label="🎨 Visual Workflow Designer",
-                    info="Drag from sidebar → Connect nodes → Edit properties",
+                wb = WorkflowBuilder(
+                    label="",
+                    info="Drag nodes → Connect edges → Edit properties → Changes auto-save!",
                     value=workflow_value,
                     elem_id="main_workflow"
                 )
+                
+                # WorkflowBuilder 변경사항을 자동으로 loaded_data에 저장
+                def update_workflow_data(workflow_data):
+                    """워크플���우 데이터 업데이트 및 상태 표시"""
+                    import time
+                    # 즉시 저장 상태 표시
+                    return workflow_data, f"💾 Auto-save: Saved ✓ ({time.strftime('%H:%M:%S')})"
+                
+                wb.change(
+                    fn=update_workflow_data,
+                    inputs=wb,
+                    outputs=[loaded_data, save_indicator]
+                )
+                
+                return wb
         
         # ─── Import Section ───
         with gr.Accordion("📥 Import Workflow", open=True):
+            gr.Markdown("*Load an existing workflow from JSON or start with a sample template*")
             with gr.Row():
                 with gr.Column(scale=2):
                     import_json_text = gr.Code(
@@ -1012,30 +1094,39 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
                     )
         
         # ─── Export Section ───
-        gr.Markdown("## 💾 Export")
+        gr.Markdown("## 💾 Export / Live Preview")
+        gr.Markdown("*Your workflow is automatically saved. The JSON below shows your current workflow in real-time.*")
+        
+        # Workflow info display
+        workflow_info = gr.Markdown("📊 Empty workflow", elem_classes=["workflow-info"])
         
         with gr.Row():
             with gr.Column(scale=3):
                 export_preview = gr.Code(
                     language="json", 
-                    label="Current Workflow JSON", 
-                    lines=8
+                    label="Current Workflow JSON (Live Preview)", 
+                    lines=8,
+                    interactive=False
                 )
+                gr.Markdown("*💡 This JSON updates automatically as you modify the workflow above*")
             with gr.Column(scale=1):
-                btn_preview = gr.Button("👁️ Preview JSON", size="lg")
+                btn_preview = gr.Button("🔄 Force Refresh", size="lg", variant="secondary")
                 btn_download = gr.DownloadButton(
                     "💾 Download JSON", 
                     size="lg",
+                    variant="primary",
                     visible=True
                 )
         
         # ─── Deploy Section ───
         with gr.Accordion("🚀 Deploy to Hugging Face Space", open=False, elem_classes=["deploy-section"]):
             gr.Markdown("""
-            Deploy your workflow as an independent Hugging Face Space app. 
-            This will create a standalone application that anyone can use without the workflow builder.
+            Deploy your **current workflow** as an independent Hugging Face Space app. 
+            The workflow shown in the JSON preview above will be deployed exactly as is.
             """)
             
+            gr.Markdown("*⚠️ Make sure to save/finalize your workflow design before deploying!*")
+            
             with gr.Row():
                 with gr.Column(scale=2):
                     deploy_name = gr.Textbox(
@@ -1063,22 +1154,26 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
                         info="Get your token from huggingface.co/settings/tokens"
                     )
                     
+                    # API Keys 설정 섹션
+                    gr.Markdown("### 🔑 API Keys Configuration")
+                    
                     # FRIENDLI_TOKEN 설정
-                    with gr.Group():
-                        include_friendli = gr.Checkbox(
-                            label="Include FRIENDLI_TOKEN in Space Secrets",
-                            value=True,
-                            info="Automatically add VIDraft API token to your Space"
-                        )
-                        
-                        friendli_token_input = gr.Textbox(
-                            label="FRIENDLI_TOKEN (optional)",
-                            type="password",
-                            placeholder="flp_...",
-                            value=os.getenv("FRIENDLI_TOKEN", ""),
-                            info="Leave empty to use current token, or enter a new one",
-                            visible=True
-                        )
+                    friendli_token_input = gr.Textbox(
+                        label="FRIENDLI_TOKEN (VIDraft/Gemma)",
+                        type="password",
+                        placeholder="flp_...",
+                        value=os.getenv("FRIENDLI_TOKEN", ""),
+                        info="Required for VIDraft. Will be added as secret."
+                    )
+                    
+                    # OpenAI API Key 설정
+                    openai_token_input = gr.Textbox(
+                        label="OPENAI_API_KEY (Optional)",
+                        type="password",
+                        placeholder="sk-...",
+                        value=os.getenv("OPENAI_API_KEY", ""),
+                        info="Optional. Leave empty if not using OpenAI."
+                    )
                     
                     deploy_private = gr.Checkbox(
                         label="Make Space Private",
@@ -1098,19 +1193,12 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
                     lines=20
                 )
         
-        # Toggle FRIENDLI token input visibility
-        include_friendli.change(
-            fn=lambda x: gr.update(visible=x),
-            inputs=include_friendli,
-            outputs=friendli_token_input
-        )
-        
         # ─── UI Execution Section ───
         with gr.Column(elem_classes=["ui-execution-section"]):
             gr.Markdown("## 🚀 UI Execution")
-            gr.Markdown("Generate an interactive UI from your workflow for end-users")
+            gr.Markdown("Test your workflow instantly! Click below to generate and run the UI from your current workflow design.")
             
-            btn_execute_ui = gr.Button("▶️ Generate & Run UI", variant="primary", size="lg")
+            btn_execute_ui = gr.Button("▶️ Generate & Run UI from Current Workflow", variant="primary", size="lg")
             
             # UI execution state
             ui_workflow_data = gr.State(None)
@@ -1215,6 +1303,10 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
                     if not vidraft_token and not openai_key:
                         log += "\n⚠️ No API keys found. Results will be simulated.\n"
                         log += "To get real AI responses, set API keys in environment variables.\n"
+                        log += "Minimum requirement: FRIENDLI_TOKEN for VIDraft\n"
+                    elif vidraft_token and not openai_key:
+                        log += "\n✅ VIDraft API connected - Basic functionality available\n"
+                        log += "💡 Add OPENAI_API_KEY for full functionality\n"
                     
                     log += "\n--- Processing Nodes ---\n"
                     
@@ -1258,14 +1350,15 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
         def load_workflow(json_text, file_obj):
             data, status = load_json_from_text_or_file(json_text, file_obj)
             if data:
-                return data, status, json_text if not file_obj else export_pretty(data)
+                # 로드 성공시 자동으로 미리보기 업데이트
+                return data, status, json_text if not file_obj else export_pretty(data), "💾 Auto-save: Loaded ✓"
             else:
-                return None, status, gr.update()
+                return None, status, gr.update(), gr.update()
         
         btn_load.click(
             fn=load_workflow,
             inputs=[import_json_text, file_upload],
-            outputs=[loaded_data, status_text, import_json_text]
+            outputs=[loaded_data, status_text, import_json_text, save_indicator]
         ).then(
             fn=lambda current_trigger: not current_trigger,
             inputs=trigger_update,
@@ -1276,7 +1369,7 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
         file_upload.change(
             fn=load_workflow,
             inputs=[import_json_text, file_upload],
-            outputs=[loaded_data, status_text, import_json_text]
+            outputs=[loaded_data, status_text, import_json_text, save_indicator]
         ).then(
             fn=lambda current_trigger: not current_trigger,
             inputs=trigger_update,
@@ -1285,8 +1378,8 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
         
         # Load samples
         btn_sample_basic.click(
-            fn=lambda: (create_sample_workflow("basic"), "✅ Basic Q&A sample loaded", export_pretty(create_sample_workflow("basic"))),
-            outputs=[loaded_data, status_text, import_json_text]
+            fn=lambda: (create_sample_workflow("basic"), "✅ Basic Q&A sample loaded", export_pretty(create_sample_workflow("basic")), "💾 Auto-save: Sample loaded ✓"),
+            outputs=[loaded_data, status_text, import_json_text, save_indicator]
         ).then(
             fn=lambda current_trigger: not current_trigger,
             inputs=trigger_update,
@@ -1294,8 +1387,8 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
         )
         
         btn_sample_vidraft.click(
-            fn=lambda: (create_sample_workflow("vidraft"), "✅ VIDraft sample loaded", export_pretty(create_sample_workflow("vidraft"))),
-            outputs=[loaded_data, status_text, import_json_text]
+            fn=lambda: (create_sample_workflow("vidraft"), "✅ VIDraft sample loaded", export_pretty(create_sample_workflow("vidraft")), "💾 Auto-save: Sample loaded ✓"),
+            outputs=[loaded_data, status_text, import_json_text, save_indicator]
         ).then(
             fn=lambda current_trigger: not current_trigger,
             inputs=trigger_update,
@@ -1303,8 +1396,8 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
         )
         
         btn_sample_multi.click(
-            fn=lambda: (create_sample_workflow("multi_input"), "✅ Multi-input sample loaded", export_pretty(create_sample_workflow("multi_input"))),
-            outputs=[loaded_data, status_text, import_json_text]
+            fn=lambda: (create_sample_workflow("multi_input"), "✅ Multi-input sample loaded", export_pretty(create_sample_workflow("multi_input")), "💾 Auto-save: Sample loaded ✓"),
+            outputs=[loaded_data, status_text, import_json_text, save_indicator]
         ).then(
             fn=lambda current_trigger: not current_trigger,
             inputs=trigger_update,
@@ -1312,47 +1405,65 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
         )
         
         btn_sample_chain.click(
-            fn=lambda: (create_sample_workflow("chain"), "✅ Chain processing sample loaded", export_pretty(create_sample_workflow("chain"))),
-            outputs=[loaded_data, status_text, import_json_text]
+            fn=lambda: (create_sample_workflow("chain"), "✅ Chain processing sample loaded", export_pretty(create_sample_workflow("chain")), "💾 Auto-save: Sample loaded ✓"),
+            outputs=[loaded_data, status_text, import_json_text, save_indicator]
         ).then(
             fn=lambda current_trigger: not current_trigger,
             inputs=trigger_update,
             outputs=trigger_update
         )
         
-        # Preview current workflow
+        # Preview current workflow - 강제 새로고침
+        def force_refresh_preview(current_data):
+            """현재 워크플로우 데이터를 강제로 새로고침"""
+            if current_data:
+                node_count = len(current_data.get("nodes", []))
+                edge_count = len(current_data.get("edges", []))
+                info = f"📊 Workflow contains {node_count} nodes and {edge_count} edges"
+                return export_pretty(current_data), "💾 Auto-save: Refreshed ✓", info
+            return "No workflow data available", "💾 Auto-save: No data", "📊 Empty workflow"
+        
         btn_preview.click(
-            fn=export_pretty,
+            fn=force_refresh_preview,
             inputs=loaded_data,
-            outputs=export_preview
+            outputs=[export_preview, save_indicator, workflow_info]
         )
         
-        # Download workflow
-        def prepare_download(data):
-            """다운로드를 위한 파일 준비"""
-            if not data:
-                return None
-            return export_file(data)
+        # Download workflow는 이미 loaded_data.change에서 처리됨
         
-        # Download 버튼에 파일 연결
-        loaded_data.change(
-            fn=prepare_download,
-            inputs=loaded_data,
-            outputs=btn_download
-        )
+        # Generate UI execution - 현재 워크플로우 사용
+        def prepare_ui_execution(current_data):
+            """현재 워크플로우를 UI 실행용으로 준비"""
+            if not current_data or not current_data.get("nodes"):
+                gr.Warning("Please create a workflow first!")
+                return None
+            return current_data
         
-        # Generate UI execution
         btn_execute_ui.click(
-            fn=lambda data: data,
+            fn=prepare_ui_execution,
             inputs=loaded_data,
             outputs=ui_workflow_data
         )
         
         # Auto-update export preview when workflow changes
+        def update_preview_and_download(data):
+            """워크플로우 변경시 미리보기와 다운로드 업데이트"""
+            if data:
+                preview = export_pretty(data)
+                download_file = export_file(data)
+                node_count = len(data.get("nodes", []))
+                edge_count = len(data.get("edges", []))
+                status = f"📊 Workflow contains {node_count} nodes and {edge_count} edges"
+                return preview, download_file, status
+            return "No workflow data", None, "📊 Empty workflow"
+        
+        # Status display for workflow info
+        workflow_info = gr.Markdown("📊 Empty workflow", elem_classes=["workflow-info"])
+        
         loaded_data.change(
-            fn=export_pretty,
+            fn=update_preview_and_download,
             inputs=loaded_data,
-            outputs=export_preview
+            outputs=[export_preview, btn_download, workflow_info]
         )
         
         # ─── Deploy Event Handlers ───
@@ -1360,13 +1471,16 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
         # Preview generated code
         def preview_generated_code(workflow_data, app_name, app_description):
             if not workflow_data:
-                return "No workflow loaded"
+                return "# No workflow loaded\n# Create or load a workflow first"
+            
+            if not workflow_data.get("nodes"):
+                return "# Empty workflow\n# Add some nodes to see the generated code"
             
             try:
                 code = generate_standalone_app(workflow_data, app_name, app_description)
                 return code
             except Exception as e:
-                return f"Error generating code: {str(e)}"
+                return f"# Error generating code\n# {str(e)}"
         
         # Update preview when inputs change
         deploy_name.change(
@@ -1381,6 +1495,7 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
             outputs=generated_code_preview
         )
         
+        # Update preview when workflow changes too
         loaded_data.change(
             fn=preview_generated_code,
             inputs=[loaded_data, deploy_name, deploy_description],
@@ -1389,9 +1504,12 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
         
         # Deploy handler
         def handle_deploy(workflow_data, app_name, app_description, hf_token, space_name, 
-                         include_friendli, friendli_token, is_private):
+                         friendli_token, openai_token, is_private):
             if not workflow_data:
-                return "❌ No workflow loaded. Please load a workflow first."
+                return "❌ No workflow loaded. Please create or load a workflow first."
+            
+            if not workflow_data.get("nodes"):
+                return "❌ Empty workflow. Please add some nodes to your workflow."
             
             if not hf_token:
                 return "❌ Hugging Face token is required. Get yours at huggingface.co/settings/tokens"
@@ -1410,9 +1528,20 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
             # Show deploying status
             yield "🔄 Deploying to Hugging Face Space..."
             
-            # Use current FRIENDLI_TOKEN if not provided
-            if include_friendli and not friendli_token:
+            # Prepare API keys
+            api_keys = {}
+            
+            # Always include FRIENDLI_TOKEN (even if empty)
+            if not friendli_token:
                 friendli_token = os.getenv("FRIENDLI_TOKEN", "")
+            if friendli_token:
+                api_keys["FRIENDLI_TOKEN"] = friendli_token
+            
+            # Include OpenAI key if provided
+            if not openai_token:
+                openai_token = os.getenv("OPENAI_API_KEY", "")
+            if openai_token:
+                api_keys["OPENAI_API_KEY"] = openai_token
             
             # Deploy
             result = deploy_to_huggingface(
@@ -1422,16 +1551,27 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
                 hf_token=hf_token,
                 space_name=space_name,
                 is_private=is_private,
-                include_friendli_token=include_friendli,
-                friendli_token=friendli_token
+                api_keys=api_keys
             )
             
             if result["success"]:
-                secret_msg = ""
-                if result.get("friendli_token_added"):
-                    secret_msg = "\n\n✅ **FRIENDLI_TOKEN has been automatically added to Space secrets!**"
-                elif include_friendli and not friendli_token:
-                    secret_msg = "\n\n⚠️ **No FRIENDLI_TOKEN provided. Please add it manually in Space settings.**"
+                # Build secrets status message
+                secrets_msg = "\n\n**🔑 API Keys Status:**"
+                
+                if result.get("added_secrets"):
+                    for secret in result["added_secrets"]:
+                        secrets_msg += f"\n- {secret}: ✅ Successfully added"
+                
+                if result.get("failed_secrets"):
+                    for failure in result["failed_secrets"]:
+                        secrets_msg += f"\n- {failure}: ❌ Failed to add"
+                
+                # Check for missing required keys
+                providers = result.get("providers_used", [])
+                if "VIDraft" in providers and "FRIENDLI_TOKEN" not in result.get("added_secrets", []):
+                    secrets_msg += "\n- FRIENDLI_TOKEN: ⚠️ Required for VIDraft but not provided"
+                if "OpenAI" in providers and "OPENAI_API_KEY" not in result.get("added_secrets", []):
+                    secrets_msg += "\n- OPENAI_API_KEY: ⚠️ Required for OpenAI but not provided"
                 
                 yield f"""✅ **Deployment Successful!**
                 
@@ -1439,17 +1579,14 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
 [{result['space_url']}]({result['space_url']})
 
 ⏱️ The Space will be ready in a few minutes. Building usually takes 2-5 minutes.
-{secret_msg}
 
-📝 **Next Steps:**
-1. Visit your Space URL
-2. Wait for the build to complete
-3. Share the URL with others
-4. You can edit the code directly on Hugging Face if needed
+{secrets_msg}
+
+📝 **Providers Detected in Workflow:**
+{', '.join(result.get('providers_used', [])) if result.get('providers_used') else 'No LLM providers detected'}
 
-💡 **API Keys Configuration:**
-- FRIENDLI_TOKEN (for VIDraft): {"✅ Automatically added" if result.get("friendli_token_added") else "❌ Add manually in Space settings"}
-- OPENAI_API_KEY (for OpenAI): ❌ Add manually in Space settings if needed
+🚀 **Default Configuration:**
+The app is configured to prioritize VIDraft (Gemma-3-r1984-27B) for optimal performance.
 
 📚 **Space Management:**
 - To update secrets: Go to Space settings → Repository secrets
@@ -1462,7 +1599,7 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
         btn_deploy.click(
             fn=handle_deploy,
             inputs=[loaded_data, deploy_name, deploy_description, deploy_token, deploy_space_name, 
-                   include_friendli, friendli_token_input, deploy_private],
+                   friendli_token_input, openai_token_input, deploy_private],
             outputs=deploy_status
         )