src/streamlit_app.py

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")