Update app.py

app.py

@@ -332,7 +332,6 @@ def create_sample_workflow(example_type="basic"):
     # 기본값은 basic
     return create_sample_workflow("basic")
 
-# 배포를 위한 독립 앱 생성 함수
 def generate_standalone_app(workflow_data: dict, app_name: str, app_description: str) -> str:
     """워크플로우를 독립적인 Gradio 앱으로 변환"""
     
@@ -468,7 +467,13 @@ def execute_workflow(*input_values):
                 except Exception as e:
                     results[node_id] = f"[VIDraft Error: {{str(e)}}]"
             else:
-                results[node_id] = f"[Simulated Response: {{input_text[:50]}}...]"
+                # Show which API key is missing
+                if provider == "OpenAI":
+                    results[node_id] = "[OpenAI API key not found. Please set OPENAI_API_KEY in Space secrets]"
+                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]}}...]"
                 
         elif node_type in ["ChatOutput", "textOutput", "Output"]:
             # Get connected result
@@ -488,6 +493,19 @@ with gr.Blocks(title="{app_name}", theme=gr.themes.Soft()) as demo:
     gr.Markdown("# {app_name}")
     gr.Markdown("{app_description}")
     
+    # API Status Check
+    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**
+        
+        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)
+        """)
+    
     # Extract nodes
     nodes = WORKFLOW_DATA.get("nodes", [])
     input_nodes = [n for n in nodes if n.get("type") in ["ChatInput", "textInput", "Input", "numberInput"]]
@@ -540,7 +558,8 @@ requests
 """
 
 def deploy_to_huggingface(workflow_data: dict, app_name: str, app_description: str, 
-                         hf_token: str, space_name: str, is_private: bool = False) -> dict:
+                         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"""
     
     if not HF_HUB_AVAILABLE:
@@ -562,6 +581,8 @@ def deploy_to_huggingface(workflow_data: dict, app_name: str, app_description: s
         # Generate files
         app_code = generate_standalone_app(workflow_data, app_name, app_description)
         requirements = generate_requirements_txt()
+        
+        # README with API setup instructions
         readme = f"""---
 title: {app_name}
 emoji: 🐭
@@ -577,6 +598,27 @@ pinned: false
 
 {app_description}
 
+## 🔑 API Configuration
+
+This app requires API keys to function properly. Please add the following secrets in your Space settings:
+
+### 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"}
+
+### 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
+
+## 📝 How to add secrets:
+1. Go to your Space settings
+2. Click on "Repository secrets" 
+3. Add the required secrets
+4. Restart your Space
+
+---
 Generated by MOUSE Workflow
 """
         
@@ -602,12 +644,25 @@ 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}")
+        
         space_url = f"https://huggingface.co/spaces/{repo_id.repo_id}"
         
         return {
             "success": True,
             "space_url": space_url,
-            "message": f"Successfully deployed to {space_url}"
+            "message": f"Successfully deployed to {space_url}",
+            "friendli_token_added": include_friendli_token and friendli_token
         }
         
     except Exception as e:
@@ -1007,6 +1062,24 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
                         placeholder="hf_...",
                         info="Get your token from huggingface.co/settings/tokens"
                     )
+                    
+                    # 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
+                        )
+                    
                     deploy_private = gr.Checkbox(
                         label="Make Space Private",
                         value=False
@@ -1025,6 +1098,13 @@ 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")
@@ -1308,7 +1388,8 @@ 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, is_private):
+        def handle_deploy(workflow_data, app_name, app_description, hf_token, space_name, 
+                         include_friendli, friendli_token, is_private):
             if not workflow_data:
                 return "❌ No workflow loaded. Please load a workflow first."
             
@@ -1329,6 +1410,10 @@ 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:
+                friendli_token = os.getenv("FRIENDLI_TOKEN", "")
+            
             # Deploy
             result = deploy_to_huggingface(
                 workflow_data=workflow_data,
@@ -1336,16 +1421,25 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
                 app_description=app_description,
                 hf_token=hf_token,
                 space_name=space_name,
-                is_private=is_private
+                is_private=is_private,
+                include_friendli_token=include_friendli,
+                friendli_token=friendli_token
             )
             
             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.**"
+                
                 yield f"""✅ **Deployment Successful!**
                 
 🎉 Your workflow has been deployed to:
 [{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
@@ -1353,16 +1447,22 @@ with gr.Blocks(title="🐭 MOUSE Workflow", theme=gr.themes.Soft(), css=CSS) as
 3. Share the URL with others
 4. You can edit the code directly on Hugging Face if needed
 
-💡 **Tip:** Set your API keys as secrets in the Space settings:
-- FRIENDLI_TOKEN (for VIDraft)
-- OPENAI_API_KEY (for OpenAI)
+💡 **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
+
+📚 **Space Management:**
+- To update secrets: Go to Space settings → Repository secrets
+- To restart Space: Go to Space settings → Factory reboot
+- To make changes: Edit files directly in the Space repository
 """
             else:
                 yield f"❌ **Deployment Failed**\n\nError: {result['error']}"
         
         btn_deploy.click(
             fn=handle_deploy,
-            inputs=[loaded_data, deploy_name, deploy_description, deploy_token, deploy_space_name, deploy_private],
+            inputs=[loaded_data, deploy_name, deploy_description, deploy_token, deploy_space_name, 
+                   include_friendli, friendli_token_input, deploy_private],
             outputs=deploy_status
         )