Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,220 @@
-import altair as alt
-import numpy as np
-import pandas as pd
 import streamlit as st
+import json
+from huggingface_hub import InferenceClient
+import time
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+# 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")