Spaces:
Sleeping
Sleeping
File size: 8,378 Bytes
baa968a 4d14e9a baa968a 4d14e9a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 |
import streamlit as st
import json
from huggingface_hub import InferenceClient
import time
# Initialize the client (no API key needed for public models)
@st.cache_resource
def get_client():
return InferenceClient()
client = get_client()
st.set_page_config(page_title="n8n Workflow Generator", page_icon="🤖", layout="wide")
st.title("🤖 n8n Workflow Generator")
st.markdown("Generate n8n workflows from your ideas using AI - completely free!")
# Sidebar for settings
with st.sidebar:
st.header("⚙️ Settings")
model_choice = st.selectbox(
"Select Model",
["Qwen/Qwen2.5-Coder-32B-Instruct", "meta-llama/Llama-3.2-3B-Instruct", "mistralai/Mistral-7B-Instruct-v0.3"],
index=0
)
temperature = st.slider("Creativity", 0.1, 1.0, 0.7)
st.markdown("---")
st.markdown("### 📚 How to Use")
st.markdown("1. Describe your workflow idea\n2. Get enhanced suggestions\n3. Generate n8n JSON\n4. Import to n8n")
# Main content
col1, col2 = st.columns([1, 1])
with col1:
st.header("💡 Your Workflow Idea")
user_idea = st.text_area(
"Describe your workflow",
placeholder="Example: Create a workflow that monitors GitHub issues, sends them to Slack, and logs them in Google Sheets",
height=150
)
if st.button("🔍 Enhance My Idea", type="primary", use_container_width=True):
if user_idea:
with st.spinner("Brainstorming enhancements..."):
try:
prompt = f"""You are an n8n workflow expert. A user wants to create this workflow:
{user_idea}
Provide 3-5 enhanced versions of this idea with:
1. Additional useful features
2. Error handling suggestions
3. Optimization tips
4. Integration possibilities
Keep each suggestion concise but actionable. Format as a numbered list."""
response = client.text_generation(
prompt,
model=model_choice,
max_new_tokens=800,
temperature=temperature,
return_full_text=False
)
st.session_state['enhanced_ideas'] = response
st.session_state['original_idea'] = user_idea
except Exception as e:
st.error(f"Error: {str(e)}")
else:
st.warning("Please enter a workflow idea first!")
with col2:
st.header("✨ Enhanced Suggestions")
if 'enhanced_ideas' in st.session_state:
st.markdown(st.session_state['enhanced_ideas'])
else:
st.info("Click 'Enhance My Idea' to get AI-powered suggestions")
st.markdown("---")
# Generation section
st.header("🚀 Generate n8n Workflow")
final_idea = st.text_area(
"Final workflow description (edit if needed)",
value=st.session_state.get('original_idea', ''),
height=100,
key="final_idea"
)
col_gen1, col_gen2, col_gen3 = st.columns([1, 1, 1])
with col_gen1:
include_error_handling = st.checkbox("Include error handling", value=True)
with col_gen2:
include_schedule = st.checkbox("Add schedule trigger", value=False)
with col_gen3:
workflow_complexity = st.select_slider("Complexity", ["Simple", "Medium", "Advanced"], value="Medium")
if st.button("⚡ Generate n8n JSON", type="primary", use_container_width=True):
if final_idea:
with st.spinner("Generating your workflow..."):
try:
prompt = f"""You are an expert n8n workflow developer. Create a complete, working n8n workflow JSON based on this description:
{final_idea}
Requirements:
- Complexity level: {workflow_complexity}
- Error handling: {"Yes" if include_error_handling else "No"}
- Schedule trigger: {"Yes" if include_schedule else "No"}
Generate ONLY valid n8n workflow JSON. The JSON must include:
1. "nodes" array with proper node types
2. "connections" object linking nodes
3. Realistic node configurations
4. Valid n8n node types (HTTP Request, Set, IF, Schedule Trigger, etc.)
Return ONLY the JSON, no explanations. Make it production-ready."""
response = client.text_generation(
prompt,
model=model_choice,
max_new_tokens=2000,
temperature=0.3, # Lower temperature for more precise JSON
return_full_text=False
)
# Try to extract JSON from response
json_str = response.strip()
# Remove markdown code blocks if present
if json_str.startswith("```"):
json_str = json_str.split("```")[1]
if json_str.startswith("json"):
json_str = json_str[4:]
json_str = json_str.strip()
# Validate JSON
try:
workflow_json = json.loads(json_str)
# Add metadata if missing
if "name" not in workflow_json:
workflow_json["name"] = "AI Generated Workflow"
if "nodes" not in workflow_json:
workflow_json["nodes"] = []
if "connections" not in workflow_json:
workflow_json["connections"] = {}
st.session_state['generated_workflow'] = workflow_json
st.success("✅ Workflow generated successfully!")
except json.JSONDecodeError as je:
st.error("Generated response wasn't valid JSON. Trying to fix...")
# Create a basic fallback workflow
workflow_json = {
"name": "AI Generated Workflow",
"nodes": [
{
"parameters": {},
"name": "Start",
"type": "n8n-nodes-base.start",
"typeVersion": 1,
"position": [250, 300]
},
{
"parameters": {
"url": "https://api.example.com",
"options": {}
},
"name": "HTTP Request",
"type": "n8n-nodes-base.httpRequest",
"typeVersion": 1,
"position": [450, 300]
}
],
"connections": {
"Start": {
"main": [[{"node": "HTTP Request", "type": "main", "index": 0}]]
}
}
}
st.session_state['generated_workflow'] = workflow_json
st.info("Created a basic template. Please customize it in n8n.")
except Exception as e:
st.error(f"Generation error: {str(e)}")
else:
st.warning("Please enter a workflow description!")
# Display and download section
if 'generated_workflow' in st.session_state:
st.markdown("---")
st.header("📋 Your n8n Workflow")
workflow = st.session_state['generated_workflow']
# Display stats
col_stat1, col_stat2, col_stat3 = st.columns(3)
with col_stat1:
st.metric("Nodes", len(workflow.get('nodes', [])))
with col_stat2:
st.metric("Connections", len(workflow.get('connections', {})))
with col_stat3:
st.metric("Ready to Import", "✓")
# JSON display
st.json(workflow)
# Download button
json_str = json.dumps(workflow, indent=2)
st.download_button(
label="📥 Download Workflow JSON",
data=json_str,
file_name="n8n_workflow.json",
mime="application/json",
use_container_width=True
)
st.info("💡 **How to import:** Open n8n → Workflows → Import from File → Select the downloaded JSON")
# Footer
st.markdown("---")
st.markdown("Made by Muhammad Shaheer") |